Extract element with namescape from xml by Python xml module

xml_extract.py

    from xml.etree import ElementTree as ET
    for prefix, uri in NAMESPACES.items():
        ET.register_namespace(prefix, uri)
    root_xml_element = ET.fromstring(xml)
    token_xml_element = root_xml_element.find(
        'soap:Body', namespaces=NAMESPACES
    ).find(
        'wst:RequestSecurityTokenResponseCollection', namespaces=NAMESPACES
    ).find(
        'wst:RequestSecurityTokenResponse', namespaces=NAMESPACES
    ).find(
        'wst:RequestedSecurityToken', namespaces=NAMESPACES
    ).find(
        'saml2:Assertion', namespaces=NAMESPACES
    )
    token_xml = ET.tostring(token_xml_element)