acces to RelayState in logout object

1 files changed, 16 insertions, 12 deletions

diff --git a/python/lasso.py b/python/lasso.py
index 1190445a..2f77c19a 100644
--- a/python/lasso.py
+++ b/python/lasso.py
@@ -990,26 +990,30 @@ class Logout:
     def __isprivate(self, name):
         return name == '_o'
 
-    def __init__(self, _obj):
-        """
-        The constructor
-        """
-        self._o = _obj
-
     def __getattr__(self, name):
         if self.__isprivate(name):
             return self.__dict__[name]
         if name[:2] == "__" and name[-2:] == "__" and name != "__members__":
             raise AttributeError, name
-
         ret = lassomod.logout_getattr(self, name)
-        if ret:
-            if name=="user":
-                ret = User(_obj=ret)
+        if ret is None:
+            return None
+        if name == "user":
+            ret = User(_obj=ret)
+        if name == "request":
+            ret = LogoutRequest(_obj=ret)
+        if name == "response":
+            ret = LogoutResponse(_obj=ret)
         return ret
 
-    def new(cls, provider_type, server, user = None):
-        obj = lassomod.logout_new(provider_type, server, user)
+    def __init__(self, _obj):
+        """
+        The constructor
+        """
+        self._o = _obj
+
+    def new(cls, server, provider_type):
+        obj = lassomod.logout_new(server, provider_type)
         return Logout(obj)
     new = classmethod(new)