Python装饰器可以将多个函数用作参数？

Question

我了解了Python装饰器的基本原理，并且喜欢语法。 但是，他们似乎对您的工作有所限制。

是否有一种优雅的方式可以将多个函数作为如下参数处理？

def parent_fn(child_fn1, child_fn2):
    def wrapper():
        print('stuff is happening here')
        child_fn1()
        print('other stuff is happening here')
        child_fn2()
    return wrapper

@parent_fn
def decorated():
    print('child_fn1 stuff here')

decorated()

我可以在哪里放置child_fn2代码？ 我尝试过的一些想法似乎摆脱了装饰器的简单和优雅。

Answer 1

您可以这样做：

import functools

def sequence_with(f):
    def wrap(g):
        @functools.wraps(g)
        def sequenced_func():
            f()
            g()
        return sequenced_func
    return wrap

def func1():
    print('func1')

@sequence_with(func1)
def func2():
    print('func2')

但是对于装饰师来说，这可能不是一个很好的概念选择。 从概念上讲，装饰器应该更改其装饰的单个功能的功能。 一个在相等位置上对其他两个函数进行排序的函数作为装饰器可能没有多大意义。 例如，装饰器创建的新功能现在以名称func2代替了原来的func2 。 这仅对某些特定用例有意义，而避免使用装饰器语法将为您提供更大的灵活性。

不使用装饰器语法更有意义：

def sequence(f, g):
   def sequenced_func():
       f()
       g()
   return sequenced_func

def func1():
    print('func1')

def func2():
    print('func2')

sequenced_func = sequence(func1, func2)