79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
|
from django.forms import Media
|
||
|
|
|
||
|
|
from wagtail import hooks
|
||
|
|
from wagtail.admin.auth import user_has_any_page_permission
|
||
|
|
from wagtail.admin.navigation import get_site_for_user
|
||
|
|
from wagtail.admin.ui.components import Component
|
||
|
|
from wagtail.models import Page, Site
|
||
|
|
|
||
|
|
|
||
|
|
class SummaryItem(Component):
|
||
|
|
order = 100
|
||
|
|
|
||
|
|
def __init__(self, request):
|
||
|
|
self.request = request
|
||
|
|
|
||
|
|
def is_shown(self):
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
class PagesSummaryItem(SummaryItem):
|
||
|
|
order = 100
|
||
|
|
template_name = "wagtailadmin/home/site_summary_pages.html"
|
||
|
|
|
||
|
|
def get_context_data(self, parent_context):
|
||
|
|
site_details = get_site_for_user(self.request.user)
|
||
|
|
root_page = site_details["root_page"]
|
||
|
|
site_name = site_details["site_name"]
|
||
|
|
|
||
|
|
if root_page:
|
||
|
|
page_count = Page.objects.descendant_of(root_page, inclusive=True).count()
|
||
|
|
|
||
|
|
if root_page.is_root():
|
||
|
|
# If the root page the user has access to is the Wagtail root,
|
||
|
|
# subtract one from this count because the root is not a real page.
|
||
|
|
page_count -= 1
|
||
|
|
|
||
|
|
# If precisely one site exists, link to its homepage rather than the
|
||
|
|
# tree root, to discourage people from trying to create pages as siblings
|
||
|
|
# of the homepage (#1883)
|
||
|
|
try:
|
||
|
|
root_page = Site.objects.get().root_page
|
||
|
|
except (Site.DoesNotExist, Site.MultipleObjectsReturned):
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
page_count = 0
|
||
|
|
|
||
|
|
return {
|
||
|
|
"root_page": root_page,
|
||
|
|
"total_pages": page_count,
|
||
|
|
"site_name": site_name,
|
||
|
|
}
|
||
|
|
|
||
|
|
def is_shown(self):
|
||
|
|
return user_has_any_page_permission(self.request.user)
|
||
|
|
|
||
|
|
|
||
|
|
class SiteSummaryPanel(Component):
|
||
|
|
template_name = "wagtailadmin/home/site_summary.html"
|
||
|
|
|
||
|
|
def __init__(self, request):
|
||
|
|
self.request = request
|
||
|
|
summary_items = []
|
||
|
|
for fn in hooks.get_hooks("construct_homepage_summary_items"):
|
||
|
|
fn(request, summary_items)
|
||
|
|
self.summary_items = [s for s in summary_items if s.is_shown()]
|
||
|
|
self.summary_items.sort(key=lambda p: p.order)
|
||
|
|
|
||
|
|
def get_context_data(self, parent_context):
|
||
|
|
context = super().get_context_data(parent_context)
|
||
|
|
context["summary_items"] = self.summary_items
|
||
|
|
return context
|
||
|
|
|
||
|
|
@property
|
||
|
|
def media(self):
|
||
|
|
media = Media()
|
||
|
|
for item in self.summary_items:
|
||
|
|
media += item.media
|
||
|
|
return media
|