Python-带参数的装饰器？

一尘不染

Python-带参数的装饰器？

python

我在装饰器传递变量'insurance_mode'时遇到问题。我可以通过以下装饰器语句来做到这一点：

 @execute_complete_reservation(True)
 def test_booking_gta_object(self):
     self.test_select_gta_object()

但不幸的是，该声明不起作用。也许也许有更好的方法来解决此问题。

def execute_complete_reservation(test_case,insurance_mode):
    def inner_function(self,*args,**kwargs):
        self.test_create_qsf_query()
        test_case(self,*args,**kwargs)
        self.test_select_room_option()
        if insurance_mode:
            self.test_accept_insurance_crosseling()
        else:
            self.test_decline_insurance_crosseling()
        self.test_configure_pax_details()
        self.test_configure_payer_details

    return inner_function

阅读 313

2020-02-12

共1个答案

一尘不染

带参数的装饰器的语法有些不同-带参数的装饰器应返回一个函数，该函数将接受一个函数并返回另一个函数。因此，它实际上应该返回一个普通的装饰器。有点混乱吧？我的意思是：

def decorator_factory(argument):
    def decorator(function):
        def wrapper(*args, **kwargs):
            funny_stuff()
            something_with_argument(argument)
            result = function(*args, **kwargs)
            more_funny_stuff()
            return result
        return wrapper
    return decorator

在这里，你可以阅读有关该主题的更多信息-也可以使用可调用对象来实现此目的，这也在那里进行了说明。

2020-02-12