Skip to content

Commit

Permalink
Add support for custom user model (Django 1.5+)
Browse files Browse the repository at this point in the history
  • Loading branch information
Sebastian Vetter committed Jul 16, 2013
1 parent 1c7951f commit 352d225
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion fancypages/assets/models.py
Original file line number Diff line number Diff line change
@@ -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)
Expand All @@ -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
Expand Down

0 comments on commit 352d225

Please sign in to comment.