Python装饰器使用

装饰器（decorators）是 Python 中的一种高级功能，它允许你动态地修改函数或类的行为。

装饰器是一种函数，它接受一个函数作为参数，并返回一个新的函数或修改原来的函数。

装饰器的语法使用 @xxx 来应用在函数或方法上。

基本语法

函数实现返回字符串的长度，装饰器在函数调用前后打印一句话

# 自定义函数名，声明装饰器decorator_new，参数是一个函数名，这里任意写一个参数名
def decorator_new(fun_get_len):
    # wrapper 是内部函数，它是实际会被调用的新函数，它包裹了原始函数的调用，并在其前后增加了额外的行为。
    def wrapper(str):
        print('开始前')
        # 形参调用，最后执行时调用的是目标函数，这里没有实现过程
        len = fun_get_len(str)
        print('结束后')
        return len
    return wrapper

# 注解使用装饰器，标在要被改变的函数上
@decorator_new
def get_len_new(str):
    # 这里是原函数的实现过程，装饰器里面调用的就是这个
    print('在装饰器中执行吗?')
    len = 0
    for i in str:
        len = len + 1
    return len    

print('调用原方法')
print(get_len_new('123'))
print('装饰器结束')

输出

调用原方法
开始前
在装饰器中执行吗?
结束后
3
装饰器结束

带参数的装饰器

带参数的装饰器，这里的参数是给装饰器用的，在修改原函数实现过程中使用，而不是指原函数参数

函数实现打印参数，装饰器传入参数表示打印n次

# 自定义装饰器名，带参数的装饰器，参数n是给装饰器用，用在改造原函数上
def fun_arg(n):
    # 自定义decorator_arg，自定义原函数名func
    def decorator_arg(func):
        # str是原函数的参数，不是装饰器的参数
        def wrapper(str):
            # n是装饰器的参数
            for i in range(n):
                func(str)
        return wrapper
    return decorator_arg

# 传入装饰器的参数
@fun_arg(3)
def print_str(str):
    print(str)

# 传入原函数的参数，并调用
print_str('hello')

输出

hello
hello
hello