discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

error-handle-create-shout-2
All checks were successful
Deploy on push / deploy (push) Successful in 25s
Details

Browse Source
This commit is contained in:
Untone 2024-02-26 12:52:22 +03:00
parent 4a57866c3d
commit ced8c9f75c
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
services/auth.py
View File

@ -1,6 +1,6 @@
from functools import wraps from functools import wraps
import httpx import httpx
from starlette.exceptions import HTTPException
from settings import ADMIN_SECRET, AUTH_URL from settings import ADMIN_SECRET, AUTH_URL
from services.logger import root_logger as logger from services.logger import root_logger as logger
@ -45,6 +45,7 @@ async def check_auth(req):
} }
data = await request_data(gql) data = await request_data(gql)
if data: if data:
logger.debug(data)
user_data = data.get('data', {}).get(query_name, {}).get('claims', {}) user_data = data.get('data', {}).get(query_name, {}).get('claims', {})
user_id = user_data.get('sub') user_id = user_data.get('sub')
user_roles = user_data.get('allowed_roles') user_roles = user_data.get('allowed_roles')
@ -82,7 +83,9 @@ def login_required(f):
logger.info(f' got {user_id} roles: {user_roles}') logger.info(f' got {user_id} roles: {user_roles}')
context['user_id'] = user_id.strip() context['user_id'] = user_id.strip()
context['roles'] = user_roles context['roles'] = user_roles
return await f(*args, **kwargs) return await f(*args, **kwargs)
else:
raise HTTPException(status_code=401, detail='Unauthorized')
return decorated_function return decorated_function
@ -95,6 +98,8 @@ def auth_request(f):
if user_id: if user_id:
req['user_id'] = user_id.strip() req['user_id'] = user_id.strip()
req['roles'] = user_roles req['roles'] = user_roles
return await f(*args, **kwargs) return await f(*args, **kwargs)
else:
raise HTTPException(status_code=401, detail='Unauthorized')
return decorated_function return decorated_function