This commit is contained in:
3
storage_service/utils/__init__.py
Normal file
3
storage_service/utils/__init__.py
Normal file
@ -0,0 +1,3 @@
|
||||
#!/usr/bin/env python3
|
||||
from .require_decorators import json_required
|
||||
from .error_handlers import register_all_error_handlers
|
21
storage_service/utils/error_handlers.py
Normal file
21
storage_service/utils/error_handlers.py
Normal file
@ -0,0 +1,21 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
|
||||
def get_standard_error_handler(code: int):
|
||||
|
||||
def error_handler(err):
|
||||
return {"msg": str(err)}, code
|
||||
|
||||
return error_handler
|
||||
|
||||
|
||||
def register_all_error_handlers(app):
|
||||
"""
|
||||
function to register all handlers
|
||||
"""
|
||||
|
||||
error_codes_to_override = [404, 403, 401, 405, 400, 409, 422]
|
||||
|
||||
for code in error_codes_to_override:
|
||||
app.register_error_handler(code, get_standard_error_handler(code))
|
||||
|
19
storage_service/utils/require_decorators.py
Normal file
19
storage_service/utils/require_decorators.py
Normal file
@ -0,0 +1,19 @@
|
||||
#!/usr/bin/env python3
|
||||
from flask import request, current_app, abort
|
||||
|
||||
from functools import wraps
|
||||
|
||||
|
||||
|
||||
def json_required(f):
|
||||
|
||||
@wraps(f)
|
||||
def call(*args, **kwargs):
|
||||
|
||||
if request.is_json:
|
||||
return f(*args, **kwargs)
|
||||
else:
|
||||
abort(400, "JSON required")
|
||||
|
||||
return call
|
||||
|
Reference in New Issue
Block a user