Add error handling to migration

This commit is contained in:
Ramon Wenger 2023-03-16 12:07:11 +01:00
parent ec71c66f3c
commit 981758a99c
1 changed files with 15 additions and 8 deletions

View File

@ -4,16 +4,23 @@ from django.contrib.auth.models import Group
from django.db import migrations from django.db import migrations
from wagtail.models import GroupPagePermission, Site from wagtail.models import GroupPagePermission, Site
from core.logger import get_logger
logger = get_logger(__name__)
def add_group_page_permissions(apps, schema_editor): def add_group_page_permissions(apps, schema_editor):
types = ["lock", "bulk_delete", "edit", "publish", "unlock", "add"] try:
group = Group.objects.get(name="CMS-Editors") types = ["lock", "bulk_delete", "edit", "pgroup.ublish", "unlock", "add"]
site = Site.objects.get(is_default_site=True) group = Group.objects.get(name="CMS-Editors")
page = site.root_page site = Site.objects.get(is_default_site=True)
for tp in types: page = site.root_page
GroupPagePermission.objects.get_or_create( for tp in types:
group=group, page=page, permission_type=tp GroupPagePermission.objects.get_or_create(
) group=group, page=page, permission_type=tp
)
except Exception as e:
logger.error(e)
class Migration(migrations.Migration): class Migration(migrations.Migration):