From ac150d1af398d609201ca6a9c43c3e1a93544256 Mon Sep 17 00:00:00 2001 From: dandds Date: Wed, 18 Jul 2018 09:51:20 -0400 Subject: [PATCH] add atat_role to user session data --- atst/handler.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/atst/handler.py b/atst/handler.py index 5543a43a..485873c0 100644 --- a/atst/handler.py +++ b/atst/handler.py @@ -15,7 +15,9 @@ class BaseHandler(tornado.web.RequestHandler): @tornado.gen.coroutine def login(self, user): - user["atat_permissions"] = yield self._get_user_permissions(user["id"]) + user_permissions = yield self._get_user_permissions(user["id"]) + user["atat_permissions"] = user_permissions["atat_permissions"] + user["atat_role"] = user_permissions["atat_role"] session_id = self.sessions.start_session(user) self.set_secure_cookie("atat", session_id) return self.redirect("/home") @@ -25,7 +27,7 @@ class BaseHandler(tornado.web.RequestHandler): response = yield self.authz_client.post( "/users", json={"id": user_id, "atat_role": "ccpo"} ) - return response.json["atat_permissions"] + return response.json def get_current_user(self): cookie = self.get_secure_cookie("atat")