Allow to set additional custom keys on services

[cascardo/ipsilon.git] / ipsilon / tools / saml2metadata.py

diff --git a/ipsilon/tools/saml2metadata.py b/ipsilon/tools/saml2metadata.py
index b86e727..27eddb9 100755 (executable)
--- a/ipsilon/tools/saml2metadata.py
+++ b/ipsilon/tools/saml2metadata.py
@@ -107,10 +107,12 @@ class Metadata(object):
         if enccert:
             self.add_cert(enccert.get_cert(), 'encryption')
 
-    def add_service(self, service, location):
+    def add_service(self, service, location, **kwargs):
         svc = mdElement(self.role, service[0])
         svc.set('Binding', service[1])
         svc.set('Location', location)
+        for key, value in kwargs.iteritems():
+            svc.set(key, value)
 
     def add_allowed_name_format(self, name_format):
         nameidfmt = mdElement(self.role, 'NameIDFormat')