Source code for wagtailapproval.wagtail_hooks

from django.conf.urls import include, url
from wagtail.wagtailcore import hooks
from wagtail.wagtailcore.models import Page

from . import urls
from .menu import ApprovalMenuItem
from .models import ApprovalStep


@hooks.register('register_admin_urls')
def register_admin_urls():
    return [
        url(r'^approval/', include(urls)),
        ]

@hooks.register('register_admin_menu_item')
def register_admin_menu_item():
    return ApprovalMenuItem()

@hooks.register('after_create_page')
def take_ownership_if_necessary(request, page):
    '''Checks the request user and takes ownership of the page if it is created
    by an owned user'''
    user = request.user
    # Do not take control of our pipeline or step types
    if not isinstance(page.specific, (ApprovalPipeline, ApprovalStep)):
        for step in ApprovalStep.objects.filter(group__in=user.groups.all()):
            # We do this to enforce that ApprovalTickets only grab the base
            # Page object, not the subclass
            step.take_ownership(Page.objects.get(pk=page.pk))
            step.save()