【问题标题】:Merge decorator function as class将装饰器功能合并为类
【发布时间】:2011-02-14 19:27:39
【问题描述】:

需要创建一个类,它将作为“合并”功能执行所有操作。在课堂上我会改变、处理和添加新的论点。

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

【问题讨论】:

  • 你想让类装饰器做什么?该示例仅显示了一个函数装饰器。
  • 需要创建一个类来作为“合并”功能来做所有事情。在课堂上,我将改变、处理和添加新的论点。功能“合并”有效,但我有太多的流程子功能。

标签: python class arguments decorator keyword-argument


【解决方案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'

在应用装饰器之后。在这种情况下,您可能不希望 merge 成为一个类,但您希望它生成一个类,以便将 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')

【讨论】:

    猜你喜欢
    • 2011-01-30
    • 2017-05-13
    • 2010-11-22
    • 1970-01-01
    • 2022-01-09
    • 2021-12-05
    • 2023-03-28
    • 2021-12-15
    • 2020-04-25
    相关资源
    最近更新 更多