tcms.bugs.migrations.0002_add_permissions

Complexity

Total Complexity

2

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	forwards_add_perms()	0	12	1
A	backwards()	0	6	1

from django.db import migrations


def forwards_add_perms(apps, schema_editor):
    """
        Adds permissions for this app to the group 'Tester'.
        This is useful in case that is an existing installation
        upgrading post 7.0.
    """
    Group = apps.get_model('auth', 'Group')
    Permission = apps.get_model('auth', 'Permission')

    tester = Group.objects.get(name='Tester')
    app_perms = Permission.objects.filter(content_type__app_label__contains='bugs')
    tester.permissions.add(*app_perms)


def backwards(apps, schema_editor):
    Group = apps.get_model('auth', 'Group')
    Permission = apps.get_model('auth', 'Permission')
    tester = Group.objects.get(name='Tester')
    app_perms = Permission.objects.filter(content_type__app_label__contains='bugs')
    tester.permissions.remove(*app_perms)


class Migration(migrations.Migration):
    dependencies = [
        ('bugs', '0001_initial'),
        ('core', '0001_squashed'),
    ]

    operations = [
        migrations.RunPython(forwards_add_perms, backwards),
    ]