fmt

services/auth.py

@@ -9,67 +9,67 @@ from settings import ADMIN_SECRET, AUTH_URL
 
 async def request_data(gql, headers=None):
     if headers is None:
-        headers = {'Content-Type': 'application/json'}
+        headers = {"Content-Type": "application/json"}
     try:
         async with httpx.AsyncClient() as client:
             response = await client.post(AUTH_URL, json=gql, headers=headers)
             if response.status_code == 200:
                 data = response.json()
-                errors = data.get('errors')
+                errors = data.get("errors")
                 if errors:
-                    logger.error(f'HTTP Errors: {errors}')
+                    logger.error(f"HTTP Errors: {errors}")
                 else:
                     return data
     except Exception as e:
         # Handling and logging exceptions during authentication check
-        logger.error(f'request_data error: {e}')
+        logger.error(f"request_data error: {e}")
     return None
 
 
 async def check_auth(req):
-    token = req.headers.get('Authorization')
-    user_id = ''
+    token = req.headers.get("Authorization")
+    user_id = ""
     user_roles = []
     if token:
         # Logging the authentication token
-        logger.debug(f'{token}')
-        query_name = 'validate_jwt_token'
-        operation = 'ValidateToken'
-        variables = {'params': {'token_type': 'access_token', 'token': token}}
+        logger.debug(f"{token}")
+        query_name = "validate_jwt_token"
+        operation = "ValidateToken"
+        variables = {"params": {"token_type": "access_token", "token": token}}
 
         gql = {
-            'query': f'query {operation}($params: ValidateJWTTokenInput!)  {{'
-            + f'{query_name}(params: $params) {{ is_valid claims }} '
-            + '}',
-            'variables': variables,
-            'operationName': operation,
+            "query": f"query {operation}($params: ValidateJWTTokenInput!)  {{"
+            + f"{query_name}(params: $params) {{ is_valid claims }} "
+            + "}",
+            "variables": variables,
+            "operationName": operation,
         }
         data = await request_data(gql)
         if data:
             logger.debug(data)
-            user_data = data.get('data', {}).get(query_name, {}).get('claims', {})
-            user_id = user_data.get('sub', '')
-            user_roles = user_data.get('allowed_roles', [])
+            user_data = data.get("data", {}).get(query_name, {}).get("claims", {})
+            user_id = user_data.get("sub", "")
+            user_roles = user_data.get("allowed_roles", [])
     return user_id, user_roles
 
 
 async def add_user_role(user_id):
-    logger.info(f'add author role for user_id: {user_id}')
-    query_name = '_update_user'
-    operation = 'UpdateUserRoles'
+    logger.info(f"add author role for user_id: {user_id}")
+    query_name = "_update_user"
+    operation = "UpdateUserRoles"
     headers = {
-        'Content-Type': 'application/json',
-        'x-authorizer-admin-secret': ADMIN_SECRET,
+        "Content-Type": "application/json",
+        "x-authorizer-admin-secret": ADMIN_SECRET,
     }
-    variables = {'params': {'roles': 'author, reader', 'id': user_id}}
+    variables = {"params": {"roles": "author, reader", "id": user_id}}
     gql = {
-        'query': f'mutation {operation}($params: UpdateUserInput!) {{ {query_name}(params: $params) {{ id roles }} }}',
-        'variables': variables,
-        'operationName': operation,
+        "query": f"mutation {operation}($params: UpdateUserInput!) {{ {query_name}(params: $params) {{ id roles }} }}",
+        "variables": variables,
+        "operationName": operation,
     }
     data = await request_data(gql, headers)
     if data:
-        user_id = data.get('data', {}).get(query_name, {}).get('id')
+        user_id = data.get("data", {}).get(query_name, {}).get("id")
         return user_id
 
 
@@ -77,15 +77,15 @@ def login_required(f):
     @wraps(f)
     async def decorated_function(*args, **kwargs):
         info = args[1]
-        req = info.context.get('request')
+        req = info.context.get("request")
         authorized = await check_auth(req)
         if authorized:
             logger.info(authorized)
             user_id, user_roles = authorized
             if user_id and user_roles:
-                logger.info(f' got {user_id} roles: {user_roles}')
-                info.context['user_id'] = user_id.strip()
-                info.context['roles'] = user_roles
+                logger.info(f" got {user_id} roles: {user_roles}")
+                info.context["user_id"] = user_id.strip()
+                info.context["roles"] = user_roles
         return await f(*args, **kwargs)
 
     return decorated_function
@@ -99,11 +99,11 @@ def auth_request(f):
         if authorized:
             user_id, user_roles = authorized
             if user_id and user_roles:
-                logger.info(f' got {user_id} roles: {user_roles}')
-                req['user_id'] = user_id.strip()
-                req['roles'] = user_roles
+                logger.info(f" got {user_id} roles: {user_roles}")
+                req["user_id"] = user_id.strip()
+                req["roles"] = user_roles
             return await f(*args, **kwargs)
         else:
-            raise HTTPException(status_code=401, detail='Unauthorized')
+            raise HTTPException(status_code=401, detail="Unauthorized")
 
     return decorated_function