diff options
| author | Ivan Kanakarakis <ivan.kanak@gmail.com> | 2020-11-04 19:42:47 +0200 |
|---|---|---|
| committer | Ivan Kanakarakis <ivan.kanak@gmail.com> | 2020-11-04 21:45:14 +0200 |
| commit | 17ba9b1894824451957305d6348e3dc6bcb0402f (patch) | |
| tree | fd08be9eccd430761229125c26984e7054e3fabd | |
| parent | dde54a0508ef434a643212f43869a996d9809599 (diff) | |
| download | pysaml2-17ba9b1894824451957305d6348e3dc6bcb0402f.tar.gz | |
Restructure the registration_info method
Signed-off-by: Ivan Kanakarakis <ivan.kanak@gmail.com>
| -rw-r--r-- | src/saml2/mdstore.py | 35 |
1 files changed, 21 insertions, 14 deletions
diff --git a/src/saml2/mdstore.py b/src/saml2/mdstore.py index 5c393bd9..96acfa1e 100644 --- a/src/saml2/mdstore.py +++ b/src/saml2/mdstore.py @@ -1432,22 +1432,29 @@ class MetadataStore(MetaData): :type entity_id: string :rtype: dict """ - res = { - 'registration_authority': None, - 'registration_instant': None, - 'registration_policy': {} - } try: - ext = self.__getitem__(entity_id)["extensions"] + ext = self.__getitem__(entity_id) except KeyError: - return res - for elem in ext["extension_elements"]: - if elem["__class__"] == classnames["mdrpi_registration_info"]: - res["registration_authority"] = elem["registration_authority"] - res["registration_instant"] = elem.get("registration_instant") - for policy in elem.get('registration_policy', []): - if policy["__class__"] == classnames["mdrpi_registration_policy"]: - res['registration_policy'][policy["lang"]] = policy["text"] + ext = {} + + ext_elems = ext.get("extensions", {}).get("extension_elements", []) + reg_info = next( + ( + elem + for elem in ext_elems + if elem["__class__"] == classnames["mdrpi_registration_info"] + ), + {}, + ) + res = { + "registration_authority": reg_info.get("registration_authority"), + "registration_instant": reg_info.get("registration_instant"), + "registration_policy": { + policy["lang"]: policy["text"] + for policy in reg_info.get("registration_policy", []) + if policy["__class__"] == classnames["mdrpi_registration_policy"] + }, + } return res def _lookup_elements_by_cls(self, root, cls): |