From 1c4300000d1f51bdbbfd1f2df63cf41e54cb21dd Mon Sep 17 00:00:00 2001 From: Drew Hubl Date: Tue, 6 May 2014 07:55:30 -0600 Subject: [PATCH] Properly update user foreign key to Django 1.6+ --- tos/models.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tos/models.py b/tos/models.py index 0468008..0ecc636 100644 --- a/tos/models.py +++ b/tos/models.py @@ -1,9 +1,7 @@ from django.core.exceptions import ValidationError from django.db import models -from django.contrib.auth import get_user_model from django.utils.translation import ugettext_lazy as _ -User = get_user_model() class NoActiveTermsOfService(ValidationError): pass @@ -60,7 +58,7 @@ class TermsOfService(BaseModel): class UserAgreement(BaseModel): terms_of_service = models.ForeignKey(TermsOfService, related_name='terms') - user = models.ForeignKey(User, related_name='user_agreement') + user = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='user_agreement') def __unicode__(self): return u'%s agreed to TOS: %s' % (self.user.username,