mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00
Code Issues Releases Wiki Activity

Make wagtaildocs migration 0002 reversible

Browse Source
This commit is contained in:
Benjamin Bach 2015-12-28 02:28:27 +01:00
parent 09282c9f6b
commit a830179bae
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
wagtail/wagtaildocs/migrations/0002_initial_data.py
View File

@ -38,6 +38,20 @@ 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 +62,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),
]