将装饰器函数合并为类

Question

需要创建一个将完成“合并”功能的类。在课堂上我会改变，处理和添加新的论点。

def merge(*arg, **kwarg): # get decorator args & kwargs 
    def func(f): 
     def tmp(*args, **kwargs): # get function args & kwargs  
      kwargs.update(kwarg) # merge two dictionaries 
      return f(*args, **kwargs) # return merged data 
     return tmp 
    return func 

用法：

@other_decorator # return *args and **kwarg 
@merge(list=['one','two','three']) # need to merge with @other_decorator 
def test(*a, **k): # get merged args and kwargs 
    print 'args:', a 
    print 'kwargs:', k 

Answer 1

我不知道我很明白你问什么。你的实现工作正常，如果你想创建任何类型的参数化装饰器，你就不会有两层间接寻址。

要合并的一类，你可以做到这一点

class Merge(object): 
    def __init__(self, **extra_kws): 
     self.extra_kws = extra_kws 
    def __call__(self, function): 
     def _wrapper(*args, **kws): 
      kws.update(self.extra_kws) 
      return function(*args, **kws) 
     return _wrapper 

然后，你可以这样做：

@Merge(foo='bar') 
def test(*args, **kws): 
    print *args 
    print **kws 

但是你说你要添加的变化和处理新的论据。所以大概你想装饰者本身是活的，所以你可以这样做：

test.extra_kws['sun'] = 'dock' 

装饰器已经应用之后。在这种情况下，你可能不希望合并为一类，但你希望它产生的一类，因此test被修改的情况下更换：

def merge(**extra_kws): 
    class _Merge(object): 
     def __init__(self, function): 
      self.extra_kws = extra_kws 
      self.function = function 
     def __call__(self, *args, **kws): 
      kws.update(self.extra_kws) 
      return self.function(*args, **kws) 
    return _Merge 

@merge(foo='bar') 
def test(*args, **kws): 
    print 'args:', args 
    print 'kws:', kws 

test(sun='dock') 
test.extra_kws['trog'] = 'cube' 
test(sun='dock') 

这就允许您在更改关键字稍后一个特定的装饰功能。

你也可以做函数参数相同的事情，而不类：

def merge(**extra_kws): 
    def _decorator(function): 
     def _wrapper(*args, **kws): 
      kws.update(_wrapper.extra_kws) 
      return function(*args, **kws) 
     _wrapper.extra_kws = extra_kws 
     return _wrapper 
    return _decorator 

@merge(foo='bar') 
def test(*args, **kws): 
    print 'kws:', kws 

test(sun='dock') 
test.extra_kws['trog'] = 'cube' 
test(sun='dock')