mirror of https://gitlab.federez.net/re2o/re2o
1 changed files with 47 additions and 0 deletions
@ -0,0 +1,47 @@ |
|||||
|
# -*- coding: utf-8 -*- |
||||
|
from __future__ import unicode_literals |
||||
|
|
||||
|
from django.db import migrations |
||||
|
|
||||
|
def remove_permission_alias(apps, schema_editor): |
||||
|
Permission = apps.get_model('auth', 'Permission') |
||||
|
for codename in ['add_alias', 'change_alias', 'delete_alias']: |
||||
|
# Retrieve the wrong permission |
||||
|
try: |
||||
|
to_remove = Permission.objects.get( |
||||
|
codename=codename, |
||||
|
content_type__model='domain' |
||||
|
) |
||||
|
except Permission.DoesNotExist: |
||||
|
# The permission is missing so no problem |
||||
|
pass |
||||
|
else: |
||||
|
to_remove.delete() |
||||
|
|
||||
|
|
||||
|
def remove_permission_text(apps, schema_editor): |
||||
|
Permission = apps.get_model('auth', 'Permission') |
||||
|
for codename in ['add_text', 'change_text', 'delete_text']: |
||||
|
# Retrieve the wrong permission |
||||
|
try: |
||||
|
to_remove = Permission.objects.get( |
||||
|
codename=codename, |
||||
|
content_type__model='txt' |
||||
|
) |
||||
|
except Permission.DoesNotExist: |
||||
|
# The permission is missing so no problem |
||||
|
pass |
||||
|
else: |
||||
|
to_remove.delete() |
||||
|
|
||||
|
|
||||
|
class Migration(migrations.Migration): |
||||
|
|
||||
|
dependencies = [ |
||||
|
('machines', '0082_auto_20180525_2209'), |
||||
|
] |
||||
|
|
||||
|
operations = [ |
||||
|
migrations.RunPython(remove_permission_text), |
||||
|
migrations.RunPython(remove_permission_alias), |
||||
|
] |
||||
Loading…
Reference in new issue