From e16fa08b915562c6ab77ce7bb79a9d766b5a4036 Mon Sep 17 00:00:00 2001
From: Mitch Riedstra <Mitch@riedstra.us>
Date: Thu, 2 Nov 2017 16:03:14 -0400
Subject: Initial setup to use a custom User model, I still need to figure out
 how to use the built in Django permissions though

---
 app/dispatch/views.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'app/dispatch/views.py')

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'))
-- 
cgit v1.2.3