make arctic role models swappable.

dpgmediamagazines · Oct 10, 2016 · 4b585a6 · 4b585a6
1 parent a34389e
commit 4b585a6
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/arctic/migrations/0001_initial.py b/arctic/migrations/0001_initial.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-# Generated by Django 1.9.6 on 2016-07-30 20:16
+# Generated by Django 1.9.6 on 2016-10-10 13:54
 from __future__ import unicode_literals
 
 from django.conf import settings
@@ -23,13 +23,19 @@ class Migration(migrations.Migration):
                 ('name', models.CharField(max_length=100, unique=True, verbose_name='Role')),
                 ('is_active', models.BooleanField(default=True)),
             ],
+            options={
+                'swappable': 'ARCTIC_ROLE_MODEL',
+            },
         ),
         migrations.CreateModel(
             name='UserRole',
             fields=[
                 ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('role', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='arctic.Role')),
+                ('role', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=getattr(settings, 'ARCTIC_ROLE_MODEL', 'arctic.Role'))),
                 ('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='urole', to=settings.AUTH_USER_MODEL)),
             ],
+            options={
+                'swappable': 'ARCTIC_USER_ROLE_MODEL',
+            },
         ),
     ]
diff --git a/arctic/models.py b/arctic/models.py
@@ -4,7 +4,10 @@
 
 class UserRole(models.Model):
     user = models.OneToOneField(settings.AUTH_USER_MODEL, related_name='urole')
-    role = models.ForeignKey('Role')
+    role = models.ForeignKey(settings.ARCTIC_ROLE_MODEL)
+
+    class Meta:
+        swappable = 'ARCTIC_USER_ROLE_MODEL'
 
 
 class Role(models.Model):
@@ -13,3 +16,6 @@ class Role(models.Model):
 
     def __str__(self):
         return self.name
+
+    class Meta:
+        swappable = 'ARCTIC_ROLE_MODEL'
diff --git a/example/example/arctic.py b/example/example/arctic.py
@@ -22,6 +22,9 @@
     )),
 )
 
+ARCTIC_USER_ROLE_MODEL = 'arctic.UserRole'
+ARCTIC_ROLE_MODEL = 'arctic.Role'
+
 ARCTIC_ROLES = {
     'editor': (
         'articles_view',