8
0
Fork 0
mirror of https://gitlab2.federez.net/re2o/re2o synced 2024-12-28 09:53:47 +00:00
re2o/machines/migrations/0083_remove_duplicate_rights.py
Hugo Levy-Falk c4a104b3b6 I like my black.
Just ran black on the whoe repository. Fix #210.
2019-11-04 22:47:24 +01:00

44 lines
1.3 KiB
Python

# -*- 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),
]