"""sso models""" from django.utils.translation import gettext as _ from passbook.core.models import Policy from passbook.policies.struct import PolicyRequest, PolicyResult class SSOLoginPolicy(Policy): """Policy that applies to users that have authenticated themselves through SSO""" form = 'passbook.policies.sso.forms.SSOLoginPolicyForm' def passes(self, request: PolicyRequest) -> PolicyResult: """Check if user instance passes this policy""" from passbook.factors.view import AuthenticationView is_sso_login = request.user.session.get(AuthenticationView.SESSION_IS_SSO_LOGIN, False) return PolicyResult(is_sso_login) class Meta: verbose_name = _('SSO Login Policy') verbose_name_plural = _('SSO Login Policies')