Объединить функцию декоратора как класс - PullRequest
Новая
       32

Объединить функцию декоратора как класс

1 голос
/ 02 мая 2010

Необходимо создать класс, который будет выполнять все функции в качестве функции "слияния". В классе я буду изменять, обрабатывать и добавлять новые аргументы. 

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

1 Ответ

2 голосов
/ 03 мая 2010

Я не уверен, что понял, о чем вы спрашиваете. Ваша реализация работает нормально, и вы не сможете обойтись без двух уровней косвенности, если захотите создать параметризованный декоратор любого типа.

Чтобы объединить класс, вы можете сделать это 

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')
...