Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-05-11 16:53:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added accept_request method to PageViewRestriction

Browse source 
This allows custom code to check whether a PageViewRestriction will pass
a request or not without needing to depend on any implementation
details.
This commit is contained in:
Karl Hobley 2016-11-28 11:17:19 +00:00 committed by Matt Westcott
parent 7994d19f65
commit ecd585abd1
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
wagtail/wagtailcore/models.py
View file

@ -31,6 +31,7 @@ from django.utils.translation import ugettext_lazy as _
from modelcluster.models import ClusterableModel, get_all_child_relations
from treebeard.mp_tree import MP_Node
from wagtail.utils.compat import user_is_authenticated
from wagtail.wagtailcore.query import PageQuerySet, TreeQuerySet
from wagtail.wagtailcore.signals import page_published, page_unpublished
from wagtail.wagtailcore.url_routing import RouteResult
@ -1833,6 +1834,25 @@ class PageViewRestriction(models.Model):
password = models.CharField(verbose_name=_('password'), max_length=255, blank=True)
groups = models.ManyToManyField(Group, blank=True)
def accept_request(self, request):
if self.restriction_type == PageViewRestriction.PASSWORD:
passed_restrictions = request.session.get('passed_page_view_restrictions', [])
if self.id not in passed_restrictions:
return False
elif self.restriction_type == PageViewRestriction.LOGIN:
if not user_is_authenticated(request.user):
return False
elif self.restriction_type == PageViewRestriction.GROUPS:
if not request.user.is_superuser:
current_user_groups = request.user.groups.all()
if not any(group in current_user_groups for group in self.groups.all()):
return False
return True
class Meta:
verbose_name = _('page view restriction')
verbose_name_plural = _('page view restrictions')