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")