sporada/nopaque
1
0
Fork 0
mirror of https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git synced 2024-12-25 02:44:18 +00:00
Code Issues Projects Releases Wiki Activity

combine content_negotiation related decorators

Browse Source
This commit is contained in:
Patrick Jentsch 2023-03-07 16:32:15 +01:00
parent cb830a6f9b
commit 9272150212
1 changed files with 17 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
app/decorators.py
View File

@ -65,52 +65,34 @@ def background(f):
return wrapped return wrapped
def consumes(mime_type: str, *_mime_types: str):
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
provided = request.mimetype
consumeables = {mime_type, *_mime_types}
if provided not in consumeables:
raise NotAcceptable()
return f(*args, **kwargs)
return decorated_function
return decorator
def produces(mime_type: str, *_mime_types: str):
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
accepted = {*request.accept_mimetypes.values()}
produceables = {mime_type, *_mime_types}
if len(produceables & accepted) == 0:
raise NotAcceptable()
return f(*args, **kwargs)
return decorated_function
return decorator
def content_negotiation( def content_negotiation(
produces: Union[str, List[str]], produces: Union[str, List[str], None] = None,
consumes: Union[str, List[str]] consumes: Union[str, List[str], None] = None
): ):
def decorator(f): def decorator(f):
@wraps(f) @wraps(f)
def decorated_function(*args, **kwargs): def decorated_function(*args, **kwargs):
provided = request.mimetype provided = request.mimetype
if isinstance(consumes, str): if consumes is None:
consumeables = None
elif isinstance(consumes, str):
consumeables = {consumes} consumeables = {consumes}
else: elif isinstance(consumes, list) and all(isinstance(x, str) for x in consumes):
consumeables = {*consumes} consumeables = {*consumes}
accepted = {*request.accept_mimetypes.values()}
if isinstance(produces, str):
produceables = {produces}
else: else:
raise TypeError()
accepted = {*request.accept_mimetypes.values()}
if produces is None:
produceables = None
elif isinstance(produces, str):
produceables = {produces}
elif isinstance(produces, list) and all(isinstance(x, str) for x in produces):
produceables = {*produces} produceables = {*produces}
if len(produceables & accepted) == 0: else:
raise TypeError()
if produceables is not None and len(produceables & accepted) == 0:
raise NotAcceptable() raise NotAcceptable()
if provided not in consumeables: if consumeables is not None and provided not in consumeables:
raise NotAcceptable() raise NotAcceptable()
return f(*args, **kwargs) return f(*args, **kwargs)
return decorated_function return decorated_function