mirror of
https://gitlab.crans.org/bde/nk20
synced 2024-11-26 10:27:07 +00:00
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from django.conf import settings
|
|
from django.contrib.admin import AdminSite
|
|
from django.contrib.sites.admin import Site, SiteAdmin
|
|
|
|
from member.views import CustomLoginView
|
|
from .middlewares import get_current_session
|
|
|
|
|
|
class StrongAdminSite(AdminSite):
|
|
def has_permission(self, request):
|
|
"""
|
|
Authorize only staff that have the correct permission mask
|
|
"""
|
|
session = get_current_session()
|
|
return request.user.is_active and request.user.is_staff and session.get("permission_mask", -1) >= 42
|
|
|
|
def login(self, request, extra_context=None):
|
|
return CustomLoginView.as_view()(request)
|
|
|
|
|
|
# Instantiate admin site and register some defaults
|
|
admin_site = StrongAdminSite()
|
|
admin_site.register(Site, SiteAdmin)
|
|
|
|
# Add external apps model
|
|
if "oauth2_provider" in settings.INSTALLED_APPS:
|
|
from oauth2_provider.admin import Application, ApplicationAdmin, Grant, \
|
|
GrantAdmin, AccessToken, AccessTokenAdmin, RefreshToken, RefreshTokenAdmin
|
|
admin_site.register(Application, ApplicationAdmin)
|
|
admin_site.register(Grant, GrantAdmin)
|
|
admin_site.register(AccessToken, AccessTokenAdmin)
|
|
admin_site.register(RefreshToken, RefreshTokenAdmin)
|
|
|
|
if "django_htcpcp_tea" in settings.INSTALLED_APPS:
|
|
from django_htcpcp_tea.admin import *
|
|
from django_htcpcp_tea.models import *
|
|
admin_site.register(Pot, PotAdmin)
|
|
admin_site.register(TeaType, TeaTypeAdmin)
|
|
admin_site.register(Addition, AdditionAdmin)
|
|
|
|
if "mailer" in settings.INSTALLED_APPS:
|
|
from mailer.admin import *
|
|
from mailer.models import *
|
|
admin_site.register(Message, MessageAdmin)
|
|
admin_site.register(DontSendEntry, DontSendEntryAdmin)
|
|
admin_site.register(MessageLog, MessageLogAdmin)
|
|
|
|
if "rest_framework" in settings.INSTALLED_APPS:
|
|
from rest_framework.authtoken.admin import *
|
|
from rest_framework.authtoken.models import *
|
|
admin_site.register(Token, TokenAdmin)
|