Декоратор в Python-Flask

Question

#========================================== 
# Current API 
#========================================== 

@blueprint.route('/list/<int:type_id>/', methods=["GET"]) 
@blueprint.route('/list/<int:type_id>/<int:object_id>', methods=["GET"]) 
@blueprint.route('/list/<int:type_id>/<int:object_id>/<int:cost_id>', methods=["GET"]) 
@login_required 
def get_list(type_id, object_id=None, cost_id=None): 
    # Do something 
    pass 

Мы используем чертежи для группировки API в нашем проекте Flask.

Теперь требование здесь, чтобы написать декоратор, который проверяет параметр API, переданный в URL, как TYPE_ID, object_id, cost_id и т.д.

#========================================== 
# New Requested feature in API 
#========================================== 
from functools import wraps 

def request_validator(): 
""" 
This function will validates requests and it's parameters if necessary 
""" 

    def wrap(f): 
     @wraps(f) 
     def wrapped(self, *args, **kwargs): 

      # TODO -- Here I want to validate before actual handler 
      # 1) type_id, 
      # 2) object_id, 
      # 3) cost_id 
      # And allow handler to process only if validation passes Here 

      if type_id not in [ 1,2,3,4,5 ]: 
       return internal_server_error(errormsg="Invalid Type ID") 

     return f(self, *args, **kwargs) 
    return wrapped 
return wrap 


@blueprint.route('/list/<int:type_id>/', methods=["GET"]) 
@blueprint.route('/list/<int:type_id>/<int:object_id>', methods=["GET"]) 
@blueprint.route('/list/<int:type_id>/<int:object_id>/<int:cost_id>', methods=["GET"]) 
@login_required 
@request_validator 
def get_list(type_id, object_id=None, cost_id=None): 
    # Do something 
    pass 

Но я получаю beow ошибку и не смог запустить приложение, Am Я что-то пропустил?

TypeError: request_validator() takes 0 positional arguments but 1 was given 

Answer 1

Ваш декоратор request_validator должен принимать функцию как аргумент. Когда вы пишете:

@request_validator 
def get_list(): 
    pass 

это означает то же самое, как:

def get_list(): 
    pass 
get_list = request_validator(get_list) 

Так что ваш декоратор должен выглядеть следующим образом (немного проще, чем в вашем примере):

def request_validator(f): 
    @wraps(f) 
    def wrapped(*args, **kwargs): 
     if type_id not in [ 1,2,3,4,5 ]: 
      return internal_server_error(errormsg="Invalid Type ID") 
     return f(*args, **kwargs) 
    return wrapped