32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
|
"""SAML base views"""
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from django.http import HttpRequest, HttpResponse
|
||
|
|
from django.shortcuts import get_object_or_404
|
||
|
|
from django.views import View
|
||
|
|
|
||
|
|
from passbook.core.models import Application
|
||
|
|
from passbook.core.views.access import AccessMixin
|
||
|
|
from passbook.providers.samlv2.saml.constants import SESSION_KEY
|
||
|
|
from passbook.providers.samlv2.saml.parser import SAMLRequest
|
||
|
|
|
||
|
|
|
||
|
|
class BaseSAMLView(AccessMixin, View):
|
||
|
|
"""Base SAML View to resolve app_slug"""
|
||
|
|
|
||
|
|
application: Application
|
||
|
|
|
||
|
|
def setup(self, request: HttpRequest, *args, **kwargs):
|
||
|
|
View.setup(self, request, *args, **kwargs)
|
||
|
|
self.application = self.get_application(self.kwargs.get("app_slug"))
|
||
|
|
|
||
|
|
def get_application(self, app_slug: str) -> Optional[Application]:
|
||
|
|
"""Return application or raise 404"""
|
||
|
|
return get_object_or_404(Application, slug=app_slug)
|
||
|
|
|
||
|
|
def handle_saml_request(self, request: SAMLRequest) -> HttpResponse:
|
||
|
|
"""Handle SAML Request"""
|
||
|
|
self.request.SESSION[SESSION_KEY] = request
|
||
|
|
if self.application.skip_authorization:
|
||
|
|
pass
|