convert Organization to django migration

This commit is contained in:
Wayne Witzel III
2016-02-08 15:35:39 -05:00
parent b8a7ad17ea
commit 1ed18e4561
4 changed files with 42 additions and 14 deletions

View File

@@ -0,0 +1,17 @@
from collections import defaultdict
def migrate_organization(apps, schema_editor):
migrations = defaultdict(list)
organization = apps.get_model('main', "Organization")
for org in organization.objects.all():
for admin in org.admins.all():
org.admin_role.members.add(admin)
migrations[org.name].append(admin)
for user in org.users.all():
org.auditor_role.members.add(user)
migrations[org.name].append(user)
return migrations
def unmigrate_organization(apps, schema_editor):
pass