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 04fec1c
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion fancypages/assets/models.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,12 @@
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 +15,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 04fec1c

Please sign in to comment.