openstack/deb-python-pysaml2
Code Issues Proposed changes

Handled possible exception in the entity_categories method and add a supported_entity_categories method.

Browse Source
This commit is contained in:
Roland Hedberg
2014-03-18 09:02:49 +01:00
parent 4f767b9ff6
commit bbe62602e8
1 changed files with 34 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/saml2/mdstore.py
View File

@@ -56,6 +56,8 @@ REQ2SRV = {
ENTITYATTRIBUTES = "urn:oasis:names:tc:SAML:metadata:attribute&EntityAttributes"
ENTITY_CATEGORY = "http://macedir.org/entity-category"
ENTITY_CATEGORY_SUPPORT = "http://macedir.org/entity-category-support"
# ---------------------------------------------------
@@ -774,13 +776,35 @@ class MetadataStore(object):
def entity_categories(self, entity_id):
ent = self.__getitem__(entity_id)
ext = ent["extensions"]
res = []
try:
ext = ent["extensions"]
except KeyError:
pass
else:
for elem in ext["extension_elements"]:
if elem["__class__"] == ENTITYATTRIBUTES:
for attr in elem["attribute"]:
if attr["name"] == "http://macedir.org/entity-category":
res.extend([v["text"] for v in attr["attribute_value"]])
if attr["name"] == ENTITY_CATEGORY:
res.extend([v["text"] for v in
attr["attribute_value"]])
return res
def supported_entity_categories(self, entity_id):
ent = self.__getitem__(entity_id)
res = []
try:
ext = ent["extensions"]
except KeyError:
pass
else:
for elem in ext["extension_elements"]:
if elem["__class__"] == ENTITYATTRIBUTES:
for attr in elem["attribute"]:
if attr["name"] == ENTITY_CATEGORY_SUPPORT:
res.extend([v["text"] for v in
attr["attribute_value"]])
return res