-
jan.bednarik authoredjan.bednarik authored
wagtail_hooks.py 640 B
from wagtail import hooks
from district.models import DistrictCenterPage, DistrictHomePage
@hooks.register("after_copy_page")
def handle_copy_calendar(request, origin_page, copied_page):
if (
isinstance(copied_page, DistrictHomePage)
or isinstance(copied_page, DistrictCenterPage)
) and origin_page.specific.calendar_id:
# get current calendar from origin page
calendar = origin_page.specific.calendar
# create copy of calendar
calendar.pk = None
calendar.save()
# set new copy of calendar
copied_page.calendar_id = calendar.id
copied_page.save()