Initial setup to use a custom User model, I still need to figure out how to use the built in Django permissions though

1 files changed, 3 insertions, 1 deletions

diff --git a/app/dispatch/views.py b/app/dispatch/views.py
index 918cf0d..5a94775 100644
--- a/app/dispatch/views.py
+++ b/app/dispatch/views.py
@@ -15,7 +15,7 @@ from django.core.urlresolvers import reverse_lazy
 from dispatch.models import Customer, Load, Paperwork, \
         Invoice, UserInvoiceNumber, Identity
 from dispatch.forms import AddPaperworkForm
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from django.contrib.auth.mixins import UserPassesTestMixin
 # from django.http import HttpResponseRedirect
 from .misc import get_week_dates, split_loads_by_day
@@ -23,6 +23,8 @@ import datetime
 import re
 import os
 
+User = get_user_model()
+
 
 def home(request):
     return redirect(reverse('load_list'))