model-service/model_service/utils/require_decorators.py
marcsello 38509c5a39
All checks were successful
continuous-integration/drone/push Build is passing
Revised API endpoints
2020-10-02 03:28:40 +02:00

29 lines
529 B
Python

#!/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
def multipart_required(f):
@wraps(f)
def call(*args, **kwargs):
if request.form:
return f(*args, **kwargs)
else:
abort(400, "multipart/form-data required")
return call