Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# Generated by Django 2.0.9 on 2018-12-19 13:21
from __future__ import unicode_literals
from django.core.exceptions import ObjectDoesNotExist
from django.core.management.sql import emit_post_migrate_signal
from django.db import migrations
from opentech.apply.users.groups import GROUPS, COMMUNITY_REVIEWER_GROUP_NAME
def add_groups(apps, schema_editor):
# Workaround for https://code.djangoproject.com/ticket/23422
db_alias = schema_editor.connection.alias
emit_post_migrate_signal(2, False, db_alias)
Group = apps.get_model('auth.Group')
Permission = apps.get_model('auth.Permission')
for group_data in GROUPS:
group, created = Group.objects.get_or_create(name=group_data['name'])
for permission in group_data['permissions']:
try:
group.permissions.add(Permission.objects.get(codename=permission))
except ObjectDoesNotExist:
print("Could not find the '%s' permission" % permission)
def remove_groups(apps, schema_editor):
Group = apps.get_model('auth.Group')
Group.objects.filter(name__in=[COMMUNITY_REVIEWER_GROUP_NAME]).delete()
class Migration(migrations.Migration):
dependencies = [
('users', '0009_add_partner_group'),
]
operations = [
migrations.RunPython(add_groups, remove_groups)
]