Python中的装饰器介绍

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

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

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

一个简单的装饰器demo：

def my_decorator(func):
    def wrapper():
        print("函数调用前.")
        func()
        print("函数调用后.")
    return wrapper


@my_decorator
def say_hello():
    print("Hello，欢迎关注公众号#小博测试成长之路!")


say_hello()

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

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")

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

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

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()

Python中的装饰器就先简单的介绍到这了，如果还不是很理解装饰器的用法，自己写不出来的话，可以复制文章中的案例依葫芦画瓢改一下试试。至于装饰器的一些其他应用场景啥的，只要代码写的足够多的时候，就会想着去进行优化，这个时候获取就能派上一些用场。

End

如果觉得文章对你有帮助的话，欢迎点赞转发~
如果还没有进粉丝交流群的小伙伴，赶快添加好友(xiaobotester)邀请你们进群喔。

关注公众号，测试干货及时送达

为感谢所有粉丝的关注和长期以来的支持，特在公众号后台为大家准备了一些资料，可在后台回复【百宝箱/关键字】获取相关链接，资料会持续更新，有需要的可以先私聊作者补充文档。

往期精彩回顾

如何打造一篇优秀的简历

混迹职场多年的你，面试真的准备好了吗？

你还在抱怨职场的不公吗？

职场篇-初入职场如何选择行业和公司
面试逻辑题分享--字母数字映射关系推算题

非技术性的面试中的技巧性回答集锦（建议收藏）

你的求职简历中招了吗？

测试编程笔试题No.1-输入表格列的英文名称，返回其对应的数字序号

软件测试面试题分享-No.1

软件测试面试题分享-No.2

软件测试面试题分享-No.3

软件测试面试题分享-No.4

软件测试面试题分享-No.5

软件测试面试题分享-No.6（某某某公司面试题）

软件测试面试题分享-No.7（接口自动化面试题）

xxx公司测试岗位企业面试题