diff --git a/api/saml/metadata/federations/loader.py b/api/saml/metadata/federations/loader.py index 3e4e21a359..c30c96a3ab 100644 --- a/api/saml/metadata/federations/loader.py +++ b/api/saml/metadata/federations/loader.py @@ -3,13 +3,10 @@ from onelogin.saml2.idp_metadata_parser import OneLogin_Saml2_IdPMetadataParser from onelogin.saml2.xmlparser import tostring -from api.saml.metadata.federations.model import ( - SAMLFederatedIdentityProvider, - SAMLFederation, -) from api.saml.metadata.federations.validator import SAMLFederatedMetadataValidator from api.saml.metadata.parser import SAMLMetadataParser from core.exceptions import BasePalaceException +from core.model import SAMLFederatedIdentityProvider, SAMLFederation from core.util import first_or_default diff --git a/api/saml/metadata/federations/validator.py b/api/saml/metadata/federations/validator.py index 1679f21ca2..4de11976ed 100644 --- a/api/saml/metadata/federations/validator.py +++ b/api/saml/metadata/federations/validator.py @@ -5,8 +5,8 @@ from onelogin.saml2.utils import OneLogin_Saml2_Utils from onelogin.saml2.xmlparser import fromstring -from api.saml.metadata.federations.model import SAMLFederation from core.exceptions import BasePalaceException +from core.model import SAMLFederation from core.util.datetime_helpers import from_timestamp, utc_now diff --git a/bin/configuration/add_saml_federations.py b/bin/configuration/add_saml_federations.py index 71ba92f1f5..d7dd10d439 100755 --- a/bin/configuration/add_saml_federations.py +++ b/bin/configuration/add_saml_federations.py @@ -10,8 +10,7 @@ sys.path.append(os.path.abspath(package_dir)) from api.saml.metadata.federations import incommon -from api.saml.metadata.federations.model import SAMLFederation -from core.model import production_session +from core.model import SAMLFederation, production_session with closing(production_session()) as db: incommon_federation = (