web/flask-home-vod
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update Media Access Request to accept Token in Query

Browse source
This commit is contained in:
Gergely Hegedus 2023-03-30 13:38:46 +03:00
parent 03728447ef
commit 2258ffce66
2 changed files with 30 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

21
server/flask/application/backend/media_access_requests.py
View file

@ -2,10 +2,10 @@ from flask import request, jsonify
from .require_decorators import get_cropped_token
from .data.data_models import ResponseCode
from .data import dao_session
from .data import dao_users
from urllib.parse import parse_qs
def handle_has_media_access():
media_token = get_cropped_token(request.headers.get('Media-Authorization'))
media_token = get_cropped_token(_get_token_from_request('Media-Authorization'))
if (media_token is None):
errorResponse = jsonify({'message':'Missing Authorization!','code':ResponseCode.MISSING_MEDIA_AUTHORIZATION})
return errorResponse, 401
@ -13,4 +13,19 @@ def handle_has_media_access():
if (user_id is None):
errorResponse = jsonify({'message':'Invalid Authorization!','code':ResponseCode.INVALID_MEDIA_AUTHORIZATION})
return errorResponse, 401
return jsonify({'message':'Access Granted','code': ResponseCode.SUCCESS_MEDIA_ACCESS}), 200
return jsonify({'message':'Access Granted','code': ResponseCode.SUCCESS_MEDIA_ACCESS}), 200
def _get_token_from_request(key: str):
token = request.headers.get(key)
if (token is not None):
return token
original_uri = request.headers.get('X-Original-URI')
if not isinstance(original_uri,str):
return None
query_string = original_uri[original_uri.find('?')+1:]
return _get_first_token_from_query_string(query_string = query_string, key = key)
def _get_first_token_from_query_string(query_string: str, key: str):
query_dict = parse_qs(query_string)
tokens = query_dict.get(key, [None])
return tokens[0]