python 装饰器在Flask中验证发布请求

Posted 2021-05-09

from functools import wraps
from flask import request, abort

def validate_post(validator, status_code):
    """Validate the POST request with the given validator function.
    Abort with the given status_code upon validation failure.
    """
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if not validator(request):
                abort(status_code)
            return func(*args, **kwargs)
        return wrapper
    return decorator