Python中的装饰器介绍 - 文章 - 开发者社区

装饰器是Python编程语言中一种强大的特性，用于修改或增强函数或类的行为，而无需对它们本身进行修改。装饰器通常被用于在不改变原始代码的情况下，向函数或方法添加额外的功能，如日志记录、权限检查、数据格式转换等。装饰器本质上是一个函数，它接受一个函数作为参数，并返回一个经过修改的函数或可调用对象。

在装饰器的内部，您通常会定义一个新的函数，这个函数执行了装饰器所要添加的操作，然后调用原始函数，并返回其结果。这就是为什么装饰器的返回值通常是一个函数。

然后，通过使用 @装饰器函数名语法将装饰器应用于其他函数或方法，使其成为经过装饰的版本。这样，当您调用经过装饰的函数时，实际上是调用了装饰器内部的新函数，从而实现了对原始函数的修改或增强。

一个简单的装饰器demo：

  
def my_decorator(func):  
 def wrapper():  
 print("函数调用前.")  
 func()  
 print("函数调用后.")  
 return wrapper  
  
@my_decorator  
def say_hello():  
 print("Hello，欢迎关注公众号#小博测试成长之路!")  
  
say_hello()

picture.image

再看一下最近工作中写脚本，之前经常涉及到对字段进行一些额外的处理操作，比如两个字段值相加，但是源数据给的是字符串，这个时候就可以利用装饰器，对数据进行格式转换之后再操作：

  
def add_decorator_convert_int(func):  
 def wrapper(a, b):  
 result = func(int(a),int(b))  
 print(f"{a} + {b} = {result}")  
 return result  
 return wrapper  
  
@add_decorator_convert_int  
def add(a, b):  
 return a + b  
  
result = add(5, 3)  
result = add("5","3")

picture.image

上面演示的是不带参数的装饰器，下面再来看看带参数的装饰器：

带参数的装饰器可以接收额外的参数，并在内部使用这些参数来定义装饰器的行为。这些参数可以在装饰器函数内部使用，以在调用原始函数之前或之后执行某些操作。

  
def validate_and_transform_data_format(data_type):  
 def decorator(func):  
 def wrapper(*args, **kwargs):  
 validated_args = []  
 for arg in args:  
 if isinstance(arg, data_type):  
 validated_args.append(arg)  
 else:  
 try:  
 validated_arg = data_type(arg)  
 validated_args.append(validated_arg)  
 except ValueError:  
 print(f"Invalid argument: {arg}. Skipping...")  
 return func(*validated_args, **kwargs)  
  
 return wrapper  
  
 return decorator  
  
  
@validate_and_transform_data_format(int)  
def calculate_sum(a, b, c):  
 return a + b + c  
  
  
result = calculate_sum("10", 20, "30")  
print("Result:", result)  
输出结果：Result: 60

上面的代码，当调用 calculate_sum("10", 20, "30") 时，装饰器将会将字符串 "10" 和 "30" 转换为整数类型，然后执行函数并返回结果。

  
def with_parameters_decorator(arg1, arg2):  
 def inner_decorator(func):  
 def wrapper(*args, **kwargs):  
 # 在调用原始函数之前执行操作  
 print(f"函数{func.__name__}被调用前")  
 result = func(*args, **kwargs)  
 # 在调用原始函数之后执行操作  
 print(f"函数{func.__name__}被调用后")  
 return result  
 return wrapper  
 return inner_decorator  
  
@with_parameters_decorator("Hello", "World")  
def my_func():  
 print("被装饰器装饰的函数被调用啦")  
  
my_func()