207 lines
6.9 KiB
Python
207 lines
6.9 KiB
Python
from django.conf import settings
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.shortcuts import get_object_or_404, redirect
|
|
from django.template.loader import render_to_string
|
|
from django.template.response import TemplateResponse
|
|
from django.urls import reverse
|
|
from django.utils.decorators import method_decorator
|
|
from django.utils.safestring import mark_safe
|
|
from django.utils.translation import gettext as _
|
|
from django.utils.translation import gettext_lazy
|
|
|
|
from wagtail.admin import messages
|
|
from wagtail.admin.action_menu import PageActionMenu
|
|
from wagtail.admin.auth import user_has_any_page_permission, user_passes_test
|
|
from wagtail.admin.ui.components import MediaContainer
|
|
from wagtail.admin.ui.side_panels import (
|
|
ChecksSidePanel,
|
|
CommentsSidePanel,
|
|
PageStatusSidePanel,
|
|
PreviewSidePanel,
|
|
)
|
|
from wagtail.admin.views.generic.models import (
|
|
RevisionsCompareView,
|
|
RevisionsUnscheduleView,
|
|
)
|
|
from wagtail.admin.views.generic.preview import PreviewRevision
|
|
from wagtail.models import Page
|
|
from wagtail.utils.timestamps import render_timestamp
|
|
|
|
|
|
def revisions_index(request, page_id):
|
|
return redirect("wagtailadmin_pages:history", page_id)
|
|
|
|
|
|
def revisions_revert(request, page_id, revision_id):
|
|
# TODO: refactor this into a class-based view that extends the EditView
|
|
page = get_object_or_404(Page, id=page_id).specific
|
|
page_perms = page.permissions_for_user(request.user)
|
|
if not page_perms.can_edit():
|
|
raise PermissionDenied
|
|
|
|
revision = get_object_or_404(page.revisions, id=revision_id)
|
|
revision_page = revision.as_object()
|
|
|
|
scheduled_page = page.get_scheduled_revision_as_object()
|
|
|
|
content_type = ContentType.objects.get_for_model(page)
|
|
page_class = content_type.model_class()
|
|
|
|
if getattr(settings, "WAGTAIL_I18N_ENABLED", False):
|
|
locale = page.locale
|
|
translations = [
|
|
{
|
|
"locale": translation.locale,
|
|
"url": reverse("wagtailadmin_pages:edit", args=[translation.id]),
|
|
}
|
|
for translation in page.get_translations()
|
|
.only("id", "locale", "depth")
|
|
.select_related("locale")
|
|
if translation.permissions_for_user(request.user).can_edit()
|
|
]
|
|
else:
|
|
locale = None
|
|
translations = []
|
|
|
|
edit_handler = page_class.get_edit_handler()
|
|
form_class = edit_handler.get_form_class()
|
|
|
|
form = form_class(instance=revision_page, for_user=request.user)
|
|
edit_handler = edit_handler.get_bound_panel(
|
|
instance=revision_page, request=request, form=form
|
|
)
|
|
|
|
preview_url = reverse("wagtailadmin_pages:preview_on_edit", args=[page.id])
|
|
lock = page.get_lock()
|
|
|
|
action_menu = PageActionMenu(
|
|
request,
|
|
view="revisions_revert",
|
|
is_revision=True,
|
|
page=page,
|
|
lock=lock,
|
|
locked_for_user=lock is not None and lock.for_user(request.user),
|
|
)
|
|
side_panels = [
|
|
PageStatusSidePanel(
|
|
revision_page,
|
|
request,
|
|
show_schedule_publishing_toggle=form.show_schedule_publishing_toggle,
|
|
live_object=page,
|
|
scheduled_object=scheduled_page,
|
|
locale=locale,
|
|
translations=translations,
|
|
),
|
|
]
|
|
if page.is_previewable():
|
|
side_panels.append(PreviewSidePanel(page, request, preview_url=preview_url))
|
|
side_panels.append(ChecksSidePanel(page, request))
|
|
if form.show_comments_toggle:
|
|
side_panels.append(CommentsSidePanel(page, request))
|
|
side_panels = MediaContainer(side_panels)
|
|
|
|
media = MediaContainer([edit_handler, form, action_menu, side_panels]).media
|
|
|
|
user_avatar = render_to_string(
|
|
"wagtailadmin/shared/user_avatar.html", {"user": revision.user}
|
|
)
|
|
|
|
messages.warning(
|
|
request,
|
|
mark_safe(
|
|
_(
|
|
"You are viewing a previous version of this page from <b>%(created_at)s</b> by %(user)s"
|
|
)
|
|
% {
|
|
"created_at": render_timestamp(revision.created_at),
|
|
"user": user_avatar,
|
|
}
|
|
),
|
|
)
|
|
|
|
page_title = _("Editing %(page_type)s") % {
|
|
"page_type": page_class.get_verbose_name()
|
|
}
|
|
page_subtitle = page.get_admin_display_title()
|
|
header_title = f"{page_title}: {page_subtitle}"
|
|
|
|
return TemplateResponse(
|
|
request,
|
|
"wagtailadmin/pages/edit.html",
|
|
{
|
|
"page": page,
|
|
"revision": revision,
|
|
"is_revision": True,
|
|
"content_type": content_type,
|
|
"edit_handler": edit_handler,
|
|
"errors_debug": None,
|
|
"action_menu": action_menu,
|
|
"side_panels": side_panels,
|
|
"header_title": header_title,
|
|
"form": form, # Used in unit tests
|
|
"media": media,
|
|
},
|
|
)
|
|
|
|
|
|
@method_decorator(user_passes_test(user_has_any_page_permission), name="dispatch")
|
|
class RevisionsView(PreviewRevision):
|
|
model = Page
|
|
|
|
def setup(self, request, page_id, revision_id, *args, **kwargs):
|
|
# Rename path kwargs from pk to page_id
|
|
return super().setup(request, page_id, revision_id, *args, **kwargs)
|
|
|
|
def get_object(self):
|
|
page = get_object_or_404(Page, id=self.pk).specific
|
|
|
|
perms = page.permissions_for_user(self.request.user)
|
|
if not (perms.can_publish() or perms.can_edit()):
|
|
raise PermissionDenied
|
|
|
|
return page
|
|
|
|
|
|
class RevisionsCompare(RevisionsCompareView):
|
|
history_label = gettext_lazy("Page history")
|
|
edit_label = gettext_lazy("Edit this page")
|
|
history_url_name = "wagtailadmin_pages:history"
|
|
edit_url_name = "wagtailadmin_pages:edit"
|
|
header_icon = "doc-empty-inverse"
|
|
|
|
@method_decorator(user_passes_test(user_has_any_page_permission))
|
|
def dispatch(self, request, *args, **kwargs):
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
def get_object(self, queryset=None):
|
|
return get_object_or_404(Page, id=self.pk).specific
|
|
|
|
def get_edit_handler(self):
|
|
return self.object.get_edit_handler()
|
|
|
|
def get_page_subtitle(self):
|
|
return self.object.get_admin_display_title()
|
|
|
|
|
|
class RevisionsUnschedule(RevisionsUnscheduleView):
|
|
model = Page
|
|
edit_url_name = "wagtailadmin_pages:edit"
|
|
history_url_name = "wagtailadmin_pages:history"
|
|
revisions_unschedule_url_name = "wagtailadmin_pages:revisions_unschedule"
|
|
header_icon = "doc-empty-inverse"
|
|
|
|
def setup(self, request, page_id, revision_id, *args, **kwargs):
|
|
# Rename path kwargs from pk to page_id
|
|
return super().setup(request, page_id, revision_id, *args, **kwargs)
|
|
|
|
def get_object(self, queryset=None):
|
|
page = get_object_or_404(Page, id=self.pk).specific
|
|
|
|
if not page.permissions_for_user(self.request.user).can_unschedule():
|
|
raise PermissionDenied
|
|
return page
|
|
|
|
def get_object_display_title(self):
|
|
return self.object.get_admin_display_title()
|