From 6a58f06cf5c8e73101e087839e1562052b7cf3c9 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Wed, 7 May 2014 17:45:03 +0000 Subject: [PATCH] Make initial migration compatible with custom user models --- notifications/migrations/0001_initial.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/notifications/migrations/0001_initial.py b/notifications/migrations/0001_initial.py index a2dc05d..b7ede3d 100644 --- a/notifications/migrations/0001_initial.py +++ b/notifications/migrations/0001_initial.py @@ -4,6 +4,12 @@ from south.db import db from south.v2 import SchemaMigration from django.db import models +try: + from django.contrib.auth import get_user_model +except ImportError: # django < 1.5 + from django.contrib.auth.models import User +else: + User = get_user_model() class Migration(SchemaMigration): @@ -11,7 +17,7 @@ class Migration(SchemaMigration): # Adding model 'Notification' db.create_table('notifications_notification', ( ('id', self.gf('django.db.models.fields.AutoField')(primary_key=True)), - ('recipient', self.gf('django.db.models.fields.related.ForeignKey')(related_name='notifications', to=orm['auth.User'])), + ('recipient', self.gf('django.db.models.fields.related.ForeignKey')(related_name='notifications', to=User)), ('readed', self.gf('django.db.models.fields.BooleanField')(default=False)), ('actor_content_type', self.gf('django.db.models.fields.related.ForeignKey')(related_name='notify_actor', to=orm['contenttypes.ContentType'])), ('actor_object_id', self.gf('django.db.models.fields.CharField')(max_length=255)), @@ -87,4 +93,4 @@ class Migration(SchemaMigration): } } - complete_apps = ['notifications'] \ No newline at end of file + complete_apps = ['notifications']