Merge pull request #573 from skanct/ck_add_entity_category_support_attribute

Add metadata attribute for entity_category_support

Author: c00kiemon5ter

src/saml2/config.py

@@ -54,6 +54,7 @@
     "session_storage",
     "assurance_certification",
     "entity_category",
+    "entity_category_support",
     "xmlsec_path",
     "extension_schemas",
     "cert_handler_extra_class",
@@ -224,6 +225,7 @@ def __init__(self, homedir="."):
         self.name_qualifier = ""
         self.assurance_certification = []
         self.entity_category = []
+        self.entity_category_support = []
         self.crypto_backend = 'xmlsec1'
         self.id_attr_name = None
         self.scope = ""

src/saml2/metadata.py

@@ -724,6 +724,15 @@ def entity_descriptor(confd):
         )
         _add_attr_to_entity_attributes(entd.extensions, attr)
 
+    if confd.entity_category_support:
+        if not entd.extensions:
+            entd.extensions = md.Extensions()
+        ava = [AttributeValue(text=c) for c in confd.entity_category_support]
+        attr = Attribute(
+            attribute_value=ava, name="http://macedir.org/entity-category-support"
+        )
+        _add_attr_to_entity_attributes(entd.extensions, attr)
+
     for item in algorithm_support_in_metadata(confd.xmlsec_binary):
         if not entd.extensions:
             entd.extensions = md.Extensions()