add get_attr support

2 files changed, 33 insertions, 1 deletions

diff --git a/python/environs/py_logout.c b/python/environs/py_logout.c
index 60560477..5a28217f 100644
--- a/python/environs/py_logout.c
+++ b/python/environs/py_logout.c
@@ -41,6 +41,36 @@ PyObject *LassoLogout_wrap(LassoLogout *logout) {
 
 /******************************************************************************/
 
+PyObject *logout_getattr(PyObject *self, PyObject *args) {
+  PyObject *logout_obj;
+  LassoLogout *logout;
+  const char *attr;
+
+  if (CheckArgs(args, "OS:logout_get_attr")) {
+    if (!PyArg_ParseTuple(args, "Os:logout_get_attr", &logout_obj, &attr))
+      return NULL;
+  }
+  else return NULL;
+
+  logout = LassoLogout_get(logout_obj);
+
+  if (!strcmp(attr, "__members__"))
+    return Py_BuildValue("[ssss]", "user", "msg_url", "msg_body",
+			 "msg_relayState");
+  if (!strcmp(attr, "user"))
+    return (LassoUser_wrap(LASSO_PROFILE_CONTEXT(logout)->user));
+  if (!strcmp(attr, "msg_url"))
+    return (charPtrConst_wrap(LASSO_PROFILE_CONTEXT(logout)->msg_url));
+  if (!strcmp(attr, "msg_body"))
+    return (charPtrConst_wrap(LASSO_PROFILE_CONTEXT(logout)->msg_body));
+  if (!strcmp(attr, "msg_relayState"))
+    return (charPtrConst_wrap(LASSO_PROFILE_CONTEXT(logout)->msg_relayState));
+
+  Py_INCREF(Py_None);
+  return (Py_None);
+}
+
+
 PyObject *logout_new(PyObject *self, PyObject *args) {
   PyObject    *server_obj, *user_obj;
   LassoLogout *logout;
diff --git a/python/environs/py_logout.h b/python/environs/py_logout.h
index 8b86e92c..91880e52 100644
--- a/python/environs/py_logout.h
+++ b/python/environs/py_logout.h
@@ -39,11 +39,13 @@ typedef struct {
 #define LassoLogout_get(v) (((v) == Py_None) ? NULL : (((LassoLogout_object *)(PyObject_GetAttr(v, PyString_FromString("_o"))))->obj))
 PyObject *LassoLogout_wrap(LassoLogout *logout);
 
-PyObject *logout_new(PyObject *self, PyObject *args);
+PyObject *logout_getattr(PyObject *self, PyObject *args);
+
 PyObject *logout_build_request_msg(PyObject *self, PyObject *args);
 PyObject *logout_build_response_msg(PyObject *self, PyObject *args);
 PyObject *logout_destroy(PyObject *self, PyObject *args);
 PyObject *logout_init_request(PyObject *self, PyObject *args);
+PyObject *logout_new(PyObject *self, PyObject *args);
 PyObject *logout_process_request_msg(PyObject *self, PyObject *args);
 PyObject *logout_process_response_msg(PyObject *self, PyObject *args);