From fb586f8f556a83231af2a9b00df272e63b2433f7 Mon Sep 17 00:00:00 2001
From: Borja Penuelas <bmpenuelas@gmail.com>
Date: Sat, 12 Dec 2020 19:17:43 +0100
Subject: [PATCH] Reference the User model with get_user_model() for
 compatibility with custom models

---
 drf_social_oauth2/management/commands/createapp.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/drf_social_oauth2/management/commands/createapp.py b/drf_social_oauth2/management/commands/createapp.py
index d882ba1..d00f0f4 100644
--- a/drf_social_oauth2/management/commands/createapp.py
+++ b/drf_social_oauth2/management/commands/createapp.py
@@ -1,10 +1,13 @@
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from django.core.management.base import BaseCommand
 
 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)"