"""saml sp models""" from django.db import models from django.urls import reverse_lazy from django.utils.translation import gettext_lazy as _ from passbook.core.models import Source from passbook.core.types import UILoginButton from passbook.crypto.models import CertificateKeyPair class SAMLSource(Source): """SAML Source""" issuer = models.TextField( blank=True, default=None, verbose_name=_("Issuer"), help_text=_("Also known as Entity ID. Defaults the Metadata URL."), ) idp_url = models.URLField(verbose_name=_("IDP URL")) idp_logout_url = models.URLField( default=None, blank=True, null=True, verbose_name=_("IDP Logout URL") ) auto_logout = models.BooleanField(default=False) signing_kp = models.ForeignKey( CertificateKeyPair, default=None, null=True, help_text=_( "Certificate Key Pair of the IdP which Assertions are validated against." ), on_delete=models.SET_NULL, ) form = "passbook.sources.saml.forms.SAMLSourceForm" @property def ui_login_button(self) -> UILoginButton: return UILoginButton( name=self.name, url=reverse_lazy( "passbook_sources_saml:login", kwargs={"source_slug": self.slug} ), icon_path="", ) @property def ui_additional_info(self) -> str: metadata_url = reverse_lazy( "passbook_sources_saml:metadata", kwargs={"source_slug": self.slug} ) return f'Metadata Download' def __str__(self): return f"SAML Source {self.name}" class Meta: verbose_name = _("SAML Source") verbose_name_plural = _("SAML Sources")