¿Puedo combinar dos decoradores en uno solo en Python?

Question

¿Hay alguna manera de combinar dos decoradores en un nuevo decorador en python?

Me doy cuenta de que puedo aplicar múltiples decoradores a una función, pero tenía curiosidad sobre si hay alguna manera simple de combinar dos en una nueva.

Answer 1

Un poco más general:

def composed(*decs): 
    def deco(f): 
     for dec in reversed(decs): 
      f = dec(f) 
     return f 
    return deco 

Entonces

@composed(dec1, dec2) 
def some(f): 
    pass 

es equivalente a

@dec1 
@dec2 
def some(f): 
    pass 

Answer 2

Si los decoradores no toman argumentos adicionales, usted podría utilizar

def compose(f, g): 
    return lambda x: f(g(x)) 

combined_decorator = compose(decorator1, decorator2) 

Ahora

@combined_decorator 
def f(): 
    pass 

será equivalente a

@decorator1 
@decorator2 
def f(): 
    pass 

Answer 3

Sí. Consulte la definición de decorador, here.

Algo como esto debería funcionar:

def multiple_decorators(func): 
    return decorator1(decorator2(func)) 

@multiple_decorators 
def foo(): pass 

Answer 4

Si no, quiero que repetir demasiado en un banco de pruebas, se podría hacer como esto ::

def apply_patches(func): 
    @functools.wraps(func) 
    @mock.patch('foo.settings.USE_FAKE_CONNECTION', False) 
    @mock.patch('foo.settings.DATABASE_URI', 'li://foo') 
    @mock.patch('foo.connection.api.Session.post', autospec=True) 
    def _(*args, **kwargs): 
     return func(*args, **kwargs) 

    return _ 

ahora puede usar eso en su conjunto de pruebas en lugar de una cantidad loca de decoradores por encima de cada función ::

def ChuckNorrisCase(unittest.TestCase): 
    @apply_patches 
    def test_chuck_pwns_none(self): 
     self.assertTrue(None)