24.Python进阶_通用装饰器写法06

通用装饰器要求 ：

1.被装饰函数 支持 不管有没有返回值都需要支持，
2. 被装饰函数支持 ，不管有多少参数 ，装饰器都要支持

结合之前的学习的带参数的和带返回值的可以写出以下形式的装饰器：

def test(func):
    def test_in(*args, **kwargs):
        print("log 日志为------------  ")
        a = func(*args, **kwargs)
        return a

    return test_in


@test
def f1():
    return "装饰有返回值的函数"

@test
def f2():
    print("装饰没有返回值的函数")

@test
def f3(a):
    return a

@test
def f4(a, b, c):
    return a + b + c



print(f1())  # 有返回值
print(f2())  # 没有返回值
print(f3(3))  # 有返回值 ，并且传入一个参数
print(f4(1, 2, c=3))  # 有返回值 并且传入多个参数

执行结果：

log 日志为------------  
装饰有返回值的函数
log 日志为------------  
装饰没有返回值的函数
None
log 日志为------------  
3
log 日志为------------  
6