This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/saml_idp/models.py

55 lines
1.8 KiB
Python
Raw Normal View History

2018-11-16 08:10:35 +00:00
"""passbook saml_idp Models"""
from django.db import models
from django.shortcuts import reverse
2018-11-26 16:18:56 +00:00
from django.utils.translation import gettext as _
2018-11-16 08:10:35 +00:00
2018-11-26 16:17:32 +00:00
from passbook.core.models import Provider
from passbook.lib.utils.reflection import class_to_path, path_to_class
2018-11-16 08:10:35 +00:00
from passbook.saml_idp.base import Processor
2018-11-26 16:17:32 +00:00
class SAMLProvider(Provider):
2018-11-16 08:10:35 +00:00
"""Model to save information about a Remote SAML Endpoint"""
2018-11-26 16:17:32 +00:00
name = models.TextField()
2018-11-16 08:10:35 +00:00
acs_url = models.URLField()
processor_path = models.CharField(max_length=255, choices=[])
issuer = models.TextField()
assertion_valid_for = models.IntegerField(default=86400)
signing = models.BooleanField(default=True)
signing_cert = models.TextField()
signing_key = models.TextField()
2018-11-16 08:10:35 +00:00
2018-11-26 21:40:10 +00:00
form = 'passbook.saml_idp.forms.SAMLProviderForm'
_processor = None
2018-11-26 21:40:10 +00:00
2018-11-16 08:10:35 +00:00
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._meta.get_field('processor_path').choices = get_provider_choices()
2018-11-16 08:10:35 +00:00
@property
def processor(self):
2018-12-26 16:26:17 +00:00
"""Return selected processor as instance"""
if not self._processor:
self._processor = path_to_class(self.processor_path)(self)
return self._processor
2018-11-16 08:10:35 +00:00
def __str__(self):
2018-11-26 16:17:32 +00:00
return "SAMLProvider %s (processor=%s)" % (self.name, self.processor_path)
def link_download_metadata(self):
"""Get link to download XML metadata for admin interface"""
return reverse('passbook_saml_idp:metadata_xml',
kwargs={'provider_id': self.pk})
2018-11-26 16:17:32 +00:00
class Meta:
verbose_name = _('SAML Provider')
verbose_name_plural = _('SAML Providers')
def get_provider_choices():
"""Return tuple of class_path, class name of all providers."""
return [(class_to_path(x), x.__name__) for x in Processor.__subclasses__()]