Puis-je combiner deux décorateurs en un seul en Python?

Question

Existe-t-il un moyen de combiner deux décorateurs en un nouveau décorateur en python?

Je me rends compte que je peux juste appliquer plusieurs décorateurs à une fonction, mais j'étais curieux de savoir s'il y avait un moyen simple de combiner deux dans un nouveau.

Answer 1

Un peu plus général:

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

Puis

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

est équivalent à

@dec1 
@dec2 
def some(f): 
    pass 

Answer 2

Si les décorateurs ne prennent pas d'arguments supplémentaires, vous pouvez utiliser

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

combined_decorator = compose(decorator1, decorator2) 

maintenant

@combined_decorator 
def f(): 
    pass 

sera équivalent à

@decorator1 
@decorator2 
def f(): 
    pass 

Answer 3

Oui. Voir la définition d'un décorateur, here.

Quelque chose comme cela devrait fonctionner:

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

@multiple_decorators 
def foo(): pass 

Answer 4

Si vous ne t voulez-vous répéter trop dans une suite de tests, vous pouvez faire comme ceci ::

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 _ 

vous pouvez l'utiliser dans votre suite de test au lieu d'une quantité folle de décorateurs au-dessus de chaque function ::

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