✨ adding user information

brazil-data-cube · Feb 24, 2023 · 680d387 · 680d387
1 parent dc42076
commit 680d387
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bdc_auth_client/decorators.py b/bdc_auth_client/decorators.py
@@ -67,6 +67,8 @@ def wrapped(*args, **kwargs):
                 kwargs.update(dict(roles=res['sub']['roles'] or []))
                 kwargs.update(dict(access_token=access_token))
                 kwargs.update(dict(user_id=res.get('user_id', None)))
+                kwargs.update(dict(user=res['sub']['user'] or []))
+
 
             else:
                 session = OAuth2Session(
@@ -104,9 +106,11 @@ def wrapped(*args, **kwargs):
                         abort(403, HTTP_403_MSG)
 
                     user_roles = res['sub'].get('roles', [])
+                    user = res['sub'].get('user', [])
                     kwargs.update(dict(roles=user_roles))
                     kwargs.update(dict(access_token=access_token))
                     kwargs.update(dict(user_id=res.get('user_id', None)))
+                    kwargs.update(dict(user=user))
                     for key, value in res['sub'].items():
                         if key != 'roles':
                             kwargs.update({key: value})