Migration and tests for super users

This commit is contained in:
Akita Noek
2016-02-08 22:45:57 -05:00
parent 5008e3faf5
commit d51447e158
2 changed files with 30 additions and 0 deletions

View File

@@ -1,5 +1,15 @@
from collections import defaultdict
def migrate_users(apps, schema_editor):
migrations = list()
User = apps.get_model('auth', "User")
Role = apps.get_model('main', "Role")
for user in User.objects.all():
if user.is_superuser:
Role.singleton('System Administrator').members.add(user)
migrations.append(user)
return migrations
def migrate_organization(apps, schema_editor):
migrations = defaultdict(list)
organization = apps.get_model('main', "Organization")