convert Organization to django migration

This commit is contained in:
Wayne Witzel III
2016-02-08 15:53:48 -05:00
parent b8a7ad17ea
commit 1ed18e4561
4 changed files with 42 additions and 14 deletions
@@ -0,0 +1,16 @@
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from awx.main.migrations import _rbac as rbac
from django.db import migrations
class Migration(migrations.Migration):
dependencies = [
('main', '0003_rbac_changes'),
]
operations = [
migrations.RunPython(rbac.migrate_organization, rbac.unmigrate_organization),
]
+17
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