use get_user_model

django_app/redbox_app/redbox_core/forms.py

@@ -2,8 +2,9 @@
 from typing import ClassVar
 
 from django import forms
+from django.contrib.auth import get_user_model
 
-from redbox_app.redbox_core.models import User
+User = get_user_model()
 
 
 class SignInForm(forms.Form):

django_app/redbox_app/redbox_core/management/commands/show_magiclink_url.py

@@ -1,11 +1,12 @@
 import logging
 from argparse import ArgumentParser
 
+from django.contrib.auth import get_user_model
 from django.core.management import BaseCommand, CommandError
 from django.db.models import Max
 from magic_link.models import MagicLink
 
-from redbox_app.redbox_core.models import User
+User = get_user_model()
 
 logger = logging.getLogger(__name__)
 

django_app/redbox_app/redbox_core/views/demographics_views.py

@@ -1,5 +1,6 @@
 import logging
 
+from django.contrib.auth import get_user_model
 from django.contrib.auth.decorators import login_required
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.http import HttpRequest, HttpResponse
@@ -9,7 +10,8 @@
 from django.views.generic import UpdateView
 
 from redbox_app.redbox_core.forms import DemographicsForm
-from redbox_app.redbox_core.models import User
+
+User = get_user_model()
 
 logger = logging.getLogger(__name__)
 

django_app/tests/views/test_demographic_views.py

@@ -3,11 +3,11 @@
 
 import pytest
 from bs4 import BeautifulSoup
+from django.contrib.auth import get_user_model
 from django.test import Client
 from pytest_django.asserts import assertRedirects
 
-from redbox_app.redbox_core.models import User
-
+User = get_user_model()
 logger = logging.getLogger(__name__)
 
 

django_app/tests/views/test_misc_views.py

@@ -4,10 +4,11 @@
 
 import pytest
 from django.conf import Settings, settings
+from django.contrib.auth import get_user_model
 from django.test import Client
 from yarl import URL
 
-from redbox_app.redbox_core.models import User
+User = get_user_model()
 
 logger = logging.getLogger(__name__)