Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-05-12 17:23:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Wrap all admin urls to add access_admin permission

Browse source
This commit is contained in:
Karl Hobley 2015-01-19 12:46:30 +00:00
parent 58ec974653
commit 27da6b2ddd
2 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
wagtail/utils/urlpatterns.py Normal file
View file

@ -0,0 +1,9 @@
def decorate_urlpatterns(urlpatterns, decorator):
for pattern in urlpatterns:
if hasattr(pattern, 'url_patterns'):
decorate_urlpatterns(pattern.url_patterns, decorator)
if hasattr(pattern, '_callback'):
pattern._callback = decorator(pattern.callback)
return urlpatterns

10
wagtail/wagtailadmin/urls.py
View file

@ -1,8 +1,10 @@
from django.conf.urls import url
from django.contrib.auth.decorators import permission_required
from wagtail.wagtailadmin.forms import PasswordResetForm
from wagtail.wagtailadmin.views import account, chooser, home, pages, tags, userbar, page_privacy
from wagtail.wagtailcore import hooks
from wagtail.utils.urlpatterns import decorate_urlpatterns
urlpatterns = [
@ -106,3 +108,11 @@ for fn in hooks.get_hooks('register_admin_urls'):
urls = fn()
if urls:
urlpatterns += urls
urlpatterns = decorate_urlpatterns(urlpatterns,
permission_required(
'wagtailadmin.access_admin',
login_url='wagtailadmin_login'
)
)