add rols to session

This commit is contained in:
Cayo Puigdefabregas 2023-06-02 09:54:14 +02:00
parent 4a6c82ef55
commit 4db71b2af4
2 changed files with 29 additions and 2 deletions

View File

@ -74,6 +74,7 @@ class LoginForm(FlaskForm):
user.get_dlt_keys(self.password.data).get('data', {}).get('api_token') user.get_dlt_keys(self.password.data).get('data', {}).get('api_token')
) )
session['token_dlt'] = token_dlt session['token_dlt'] = token_dlt
session['rols'] = user.get_rols()
return user.is_active return user.is_active

View File

@ -2,6 +2,7 @@ import json
from uuid import uuid4 from uuid import uuid4
from citext import CIText from citext import CIText
from ereuseapi.methods import API
from flask import current_app as app from flask import current_app as app
from flask import g, session from flask import g, session
from flask_login import UserMixin from flask_login import UserMixin
@ -140,8 +141,6 @@ class User(UserMixin, Thing):
if 'trublo' not in app.blueprints.keys(): if 'trublo' not in app.blueprints.keys():
return return
from ereuseapi.methods import API
if not api_token: if not api_token:
api_token = session.get('token_dlt', '.') api_token = session.get('token_dlt', '.')
target_user = api_token.split(".")[0] target_user = api_token.split(".")[0]
@ -155,6 +154,33 @@ class User(UserMixin, Thing):
result = apiUser1.issue_credential("Operator", target_user) result = apiUser1.issue_credential("Operator", target_user)
return result return result
def get_rols(self):
if session.get('rols'):
return session.get('rols')
if 'trublo' not in app.blueprints.keys():
return []
if not session.get('token_dlt'):
return []
token_dlt = session.get('token_dlt')
api_dlt = app.config.get('API_DLT')
if not token_dlt or not api_dlt:
return []
api = API(api_dlt, token_dlt, "ethereum")
result = api.check_user_roles()
if result.get('Status') != 200:
return []
if 'Success' not in result.get('Data', {}).get('status'):
return []
rols = result.get('Data', {}).get('data', {})
return [(k, k) for k, v in rols.items() if v]
class UserInventory(db.Model): class UserInventory(db.Model):
"""Relationship between users and their inventories.""" """Relationship between users and their inventories."""