Return is a protected python word so it can not be used as a parameter name in a method definition.

These fixes are there to allow for the usage or return_url or return as parameter name.

1 files changed, 5 insertions, 2 deletions

diff --git a/src/saml2/discovery.py b/src/saml2/discovery.py
index 26b9de90..160b3940 100644
--- a/src/saml2/discovery.py
+++ b/src/saml2/discovery.py
@@ -62,9 +62,12 @@ class DiscoveryServer(Entity):
 
     # -------------------------------------------------------------------------
 
-    def create_discovery_service_response(self, return_url,
+    def create_discovery_service_response(self, return_url=None,
                                           returnIDParam="entityID",
-                                          entity_id=None):
+                                          entity_id=None, **kwargs):
+        if return_url is None:
+            return_url = kwargs["return"]
+            
         if entity_id:
             qp = urlencode({returnIDParam: entity_id})