From 352d22517d37a1cbc1396b28e57f39caa0da333e Mon Sep 17 00:00:00 2001 From: Sebastian Vetter Date: Tue, 16 Jul 2013 21:21:48 +1000 Subject: [PATCH] Add support for custom user model (Django 1.5+) --- fancypages/assets/models.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/fancypages/assets/models.py b/fancypages/assets/models.py index 56b2bfe..5351446 100644 --- a/fancypages/assets/models.py +++ b/fancypages/assets/models.py @@ -1,6 +1,15 @@ + + + from django.db import models from django.utils.translation import ugettext as _ +try: + from django.contrib.auth import get_user_model + User = get_user_model() +except ImportError: + from django.contrib.auth.models import User + class AbstractAsset(models.Model): name = models.CharField(_("Name"), max_length=255) @@ -9,7 +18,7 @@ class AbstractAsset(models.Model): date_modified = models.DateTimeField(_("Date modified"), auto_now=True) description = models.TextField(_("Description"), default="") - creator = models.ForeignKey('auth.User', verbose_name=_("Creator")) + creator = models.ForeignKey(User, verbose_name=_("Creator")) def __unicode__(self): return self.name