create a saml contract in python

main.py
from onelogin.saml2.authn_context import AuthnContextClassRef
from onelogin.saml2.constants import OneLogin_Saml2_Constants
from onelogin.saml2.idp_metadata_parser import OneLogin_Saml2_IdPMetadataParser
from onelogin.saml2.settings import OneLogin_Saml2_Settings
from onelogin.saml2.utils import OneLogin_Saml2_Utils

# Load the IdP metadata from a file or URL
idp_metadata = OneLogin_Saml2_IdPMetadataParser.parse_remote_metadata("https://idp.example.com/metadata")

# Create the SAML settings object
saml_settings = OneLogin_Saml2_Settings(idp_metadata=idp_metadata)

# Set up the SAML assertion parameters
name_id = "user@example.com"
session_index = "123456"
authn_context = AuthnContextClassRef.PASSWORD_PROTECTED_TRANSPORT
not_on_or_after = OneLogin_Saml2_Utils.unix_time_to_datetime(OneLogin_Saml2_Utils.now() + 3600)

# Create the SAML assertion
saml_assertion = OneLogin_Saml2_Utils.generate_saml_assertion(
    saml_settings, name_id, session_index, authn_context=authn_context, not_on_or_after=not_on_or_after
)

# Sign the SAML assertion with the IdP private key
saml_assertion_signed = OneLogin_Saml2_Utils.add_sign(
    saml_assertion, saml_settings.get_sp_key_file(), saml_settings.get_sp_cert_file()
)

# Encode the SAML assertion as XML
saml_assertion_xml = OneLogin_Saml2_Utils.compress_node(saml_assertion_signed)

# Print the SAML assertion XML
print(saml_assertion_xml)
1392 chars
34 lines