From f9dc7c485a44899d58a89050fc395c9510b51f21 Mon Sep 17 00:00:00 2001 From: Benjamin Bach Date: Mon, 28 Dec 2015 02:28:27 +0100 Subject: [PATCH] Make wagtaildocs migration 0002 reversible --- .../wagtaildocs/migrations/0002_initial_data.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/wagtail/wagtaildocs/migrations/0002_initial_data.py b/wagtail/wagtaildocs/migrations/0002_initial_data.py index 41b8bec540..eb137f12bd 100644 --- a/wagtail/wagtaildocs/migrations/0002_initial_data.py +++ b/wagtail/wagtaildocs/migrations/0002_initial_data.py @@ -38,6 +38,21 @@ def add_document_permissions_to_admin_groups(apps, schema_editor): group.permissions.add(add_document_permission, change_document_permission, delete_document_permission) +def remove_document_permissions(apps, schema_editor): + """Reverse the above additions of permissions.""" + ContentType = apps.get_model('contenttypes.ContentType') + Permission = apps.get_model('auth.Permission') + document_content_type = ContentType.objects.get( + model='document', + app_label='wagtaildocs', + ) + # This cascades to Group + Permission.objects.filter( + content_type=document_content_type, + codename__in=('add_document', 'change_document', 'delete_document'), + ).delete() + + class Migration(migrations.Migration): dependencies = [ @@ -48,5 +63,5 @@ class Migration(migrations.Migration): ] operations = [ - migrations.RunPython(add_document_permissions_to_admin_groups), + migrations.RunPython(add_document_permissions_to_admin_groups, remove_document_permissions), ]