wagtail/wagtail/actions/convert_alias.py

70 wiersze
1.9 KiB
Python

from django.core.exceptions import PermissionDenied
from wagtail.log_actions import log
class ConvertAliasPageError(RuntimeError):
"""
Raised when the page to convert is not an alias.
"""
pass
class ConvertAliasPagePermissionError(PermissionDenied):
"""
Raised when the alias page conversion cannot be performed due to insufficient permissions.
"""
pass
class ConvertAliasPageAction:
def __init__(self, page, *, log_action="wagtail.convert_alias", user=None):
self.page = page
self.log_action = log_action
self.user = user
def check(self, skip_permission_checks=False):
if not self.page.alias_of_id:
raise ConvertAliasPageError("Page must be an alias to be converted.")
if (
not skip_permission_checks
and self.user
and not self.page.permissions_for_user(self.user).can_edit()
):
raise ConvertAliasPagePermissionError(
"You do not have permission to edit this page."
)
def _convert_alias(self, page, log_action, user):
page.alias_of_id = None
page.save(update_fields=["alias_of_id"], clean=False)
# Create an initial revision
revision = page.save_revision(user=user, changed=False, clean=False)
if page.live:
page.live_revision = revision
page.save(update_fields=["live_revision"], clean=False)
# Log
if log_action:
log(
instance=page,
action=log_action,
revision=revision,
user=user,
data={
"page": {"id": page.id, "title": page.get_admin_display_title()},
},
)
return page
def execute(self, skip_permission_checks=False):
self.check(skip_permission_checks=skip_permission_checks)
return self._convert_alias(self.page, self.log_action, self.user)