Merge pull request #236 from bmpenuelas/master

Reference the User model with get_user_model()

Author: islam-kamel

rest_framework_social_oauth2/management/commands/createapp.py

@@ -1,9 +1,13 @@
 from django.contrib.auth.models import User
 from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth import get_user_model
 from oauth2_provider.models import Application
 from oauth2_provider.generators import generate_client_id, generate_client_secret
 
 
+User = get_user_model()
+
+
 class Command(BaseCommand):
     help = "Create a Django OAuth Toolkit application (an existing admin is required)"
 
@@ -23,7 +27,7 @@ def add_arguments(self, parser):
 
     def handle(self, *args, **options):
         new_application = Application(
-            user= User.objects.filter(is_superuser=True)[0],
+            user=User.objects.filter(is_superuser=True)[0],
             client_type="confidential",
             authorization_grant_type="password",
             name=options["name"] or "socialauth_application",