openstack/deb-python-pysaml2
Code Issues Proposed changes

Handle possible lack of extensions

Browse Source
This commit is contained in:
Roland Hedberg
2014-04-08 17:37:56 +02:00
parent 3740d80810
commit 69c5e078e0
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/saml2/mdstore.py
View File

@@ -815,14 +815,18 @@ class MetadataStore(object):
:type entity_id: string :type entity_id: string
:rtype: dict :rtype: dict
""" """
ext = self.__getitem__(entity_id)["extensions"]
res = {} res = {}
try:
ext = self.__getitem__(entity_id)["extensions"]
except KeyError:
return res
for elem in ext["extension_elements"]: for elem in ext["extension_elements"]:
if elem["__class__"] == ENTITYATTRIBUTES: if elem["__class__"] == ENTITYATTRIBUTES:
for attr in elem["attribute"]: for attr in elem["attribute"]:
if attr["name"] not in res: if attr["name"] not in res:
res[attr["name"]] = [] res[attr["name"]] = []
res[attr["name"]] += [v["text"] for v in attr["attribute_value"]] res[attr["name"]] += [v["text"] for v in attr[
"attribute_value"]]
return res return res
def bindings(self, entity_id, typ, service): def bindings(self, entity_id, typ, service):