Python3 украшает условно?

Можно ли украсить функцию на основе условия?

a'la:

if she.weight() == duck.weight(): 
    @burn
def witch():
    pass

Мне просто интересно, можно ли использовать логику (когда вызывается witch?), чтобы выяснить, украшать ли witch @burn?

Если нет, то можно ли создать такое же условие в декораторе? (witch называется неукрашенным.)


python conditional-statements decorator
person MadSc13ntist    schedule 22.09.2010    source источник

Ответы (3)


arrow_upward
11
arrow_downward

Вы можете создать «условно» декоратор:

>>> def conditionally(dec, cond):
    def resdec(f):
        if not cond:
            return f
        return dec(f)
    return resdec

Пример использования следующий:

>>> def burn(f):
    def blah(*args, **kwargs):
        print 'hah'
        return f(*args, **kwargs)
    return blah

>>> @conditionally(burn, True)
def witch(): pass
>>> witch()
hah

>>> @conditionally(burn, False)
def witch(): pass
>>> witch()
person Claudiu    schedule 05.10.2010
comment
+1 Мне это нравится больше, чем мое решение, так как оно соответствует теме декоратора и выглядит довольно чисто, имхо. - person nilamo; 25.10.2010
comment
Это старый ответ, но кто-нибудь знает, как передать аргументы декоратору с условной оболочкой, используя этот шаблон? - person Nicholas Tulach; 01.11.2017

arrow_upward
5
arrow_downward

Декораторы — это просто синтаксический сахар для переопределения функции, например:

def wrapper(f):
    def inner(f, *args):
        return f(*args)
    return lambda *args: inner(f, *args)

def foo():
    return 4
foo = wrapper(foo)

Это означает, что вы могли бы сделать это по-старому, до того, как появился синтаксический сахар:

def foo():
    return 4
if [some_condition]:
    foo = wrapper(foo)
person nilamo    schedule 05.10.2010

arrow_upward
5
arrow_downward

Можно включить/отключить декораторы путем переназначения.

def unchanged(func):
    "This decorator doesn't add any behavior"
    return func

def disabled(func):
    "This decorator disables the provided function, and does nothing"
    def empty_func(*args,**kargs):
        pass
    return empty_func

# define this as equivalent to unchanged, for nice symmetry with disabled
enabled = unchanged

#
# Sample use
#

GLOBAL_ENABLE_FLAG = True

state = enabled if GLOBAL_ENABLE_FLAG else disabled
@state
def special_function_foo():
    print "function was enabled"
person MadSc13ntist    schedule 30.09.2010