Skip to content

Commit

Permalink
feat: add auth methods
Browse files Browse the repository at this point in the history
  • Loading branch information
Valbou committed May 25, 2024
1 parent 768f3a4 commit 1832db4
Showing 1 changed file with 16 additions and 6 deletions.
22 changes: 16 additions & 6 deletions src/libs/flask/utils/request_helpers.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,9 @@
import logging

from flask import Request

logger = logging.getLogger(__name__)

BEARER = "Bearer "


Expand All @@ -11,18 +15,24 @@ def get_bearer_token(request: Request) -> str | None:


def get_payload_token(request: Request) -> str | None:
token = request.get_json().get("token")
return token
return request.get_json().get("token")


def get_querystring_token(request: Request) -> str | None:
return request.args.to_dict().get("token")


def get_auth_token(request: Request) -> str | None:
token = None
methods = [get_bearer_token, get_payload_token]
methods = [get_bearer_token, get_payload_token, get_querystring_token]

for method in methods:
token = method(request=request)
if token:
break
try:
token = method(request=request)
if token:
break
except Exception as e:
logger.debug(f"Error in method {method.__name__}: {e}")

return token

Expand Down

0 comments on commit 1832db4

Please sign in to comment.