Python装饰器的几种运用场景

讲下装饰器的运用了。以下都是需求+例子，大家且看且珍惜~

计时器

当我们需要统计一些方法的执行时间，我们可以这么写。
def time_dec(func):
def wrapper(*arg):
t = time.perf_counter()
res = func(*arg)
print(func.name, time.perf_counter() - t)
return res
return wrapper
复制代码
但你会发现装饰器的名字打印出来的时候，func.name 变成了wrapper，很正常，因为我们确实定义了个wrapper。
只需要在装饰器def wrapper上面加上一个装饰器:
@functools.wraps(func)

异常捕获器（来自测试开发笋货的思路）

写try和except写厌烦了吧？你可以这么玩:
def exceptions(func):
def wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except Exception as e:
raise Exception(e)

return wrapper

复制代码

web开发中判断用户是否登录

def login_required(func):
@wraps(func)
def wrapper(*args, **kwargs):
try:
headers = request.headers
token = headers.get(‘token’)
if token is None:
return jsonify(dict(code=401, msg=“用户信息认证失败, 请检查”))
# 解析用户信息
user_info = UserToken.parse_token(token)
# 这里把user信息写入kwargs
kwargs[“user_info”] = user_info
except Exception as e:
return jsonify(dict(code=401, msg=str(e)))
return func(*args, **kwargs)

    return wrapper

复制代码

数据驱动

@parameters(
(2, 4, 6),
(5, 6, 11),
)
def test_add(a, b, expected):
assert a + b == expected
复制代码
类似于ddt库提供的数据驱动方法，用于测试用例。

日志记录

用例重跑

UI用例执行异常自动截图

这个与笋货类似，捕获异常以后进行额外的处理。

给变量加锁

import functools

def synchronized(lock):
“”" Synchronization decorator “”"
def wrap(f):
@functools.wraps(f)
def newFunction(*args, **kw):
with lock:
return f(*args, **kw)
return newFunction
return wrap
复制代码

今天的内容就简单介绍到这里了，收尾明显有点潦草，因为有个同事催我下班了，哈哈~
下次再会···
目前python前景最好，学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。下面有一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

一、Python所有方向的学习路线

Python所有方向路线就是把Python常用的技术点做整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

二、学习软件

工欲善其事必先利其器。学习Python常用的开发软件都在这里了，给大家节省了很多时间。

三、全套PDF电子书

书籍的好处就在于权威和体系健全，刚开始学习的时候你可以只看视频或者听某个人讲课，但等你学完之后，你觉得你掌握了，这时候建议还是得去看一下书籍，看权威技术书籍也是每个程序员必经之路。

四、入门学习视频

我们在看视频学习的时候，不能光动眼动脑不动手，比较科学的学习方法是在理解之后运用它们，这时候练手项目就很适合了。

五、实战案例

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

这份完整版的Python全套学习资料已经上传CSDN，朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费】。