【python】装饰器听了N次，实际工作中到底能干嘛？

时间：2020-09-04 23:45:43 阅读：86 评论：0 收藏：0 [点我收藏+]

终于知道python的装饰器是怎么回事，那在工作中，到底能干吗用呢？
尤其对我这个只会写写脚本又不做python开发的小测试/手动无辜脸。
技术分享图片

先说结论，肯定是有用处滴。

一、自动化测试中使用

就拿写的自动化测试来说吧，如果我想统一的输出点东西，比如：case的运行时长，case名称等等，那就可以用起来。

首先，看下最简单的case，没有装饰器：

import pytest

def test_01():
    a = 1
    b = 2
    assert a < b

def test_02():
    a = 1
    b = 1
    assert a-b == 0

if __name__ == ‘__main__‘:
    pytest.main([‘demo_test.py‘])

这个用例文件里有2个case，运行一下：

============================= test session starts =============================
platform win32 -- Python 3.8.5, pytest-6.0.1, py-1.9.0, pluggy-0.13.1
rootdir: D:\练习
collected 2 items

demo_test.py ..                                                          [100%]

============================== 2 passed in 0.01s ==============================
[Finished in 0.4s]

运行通过就是.，所以2个case都通过了，有2个.。

如果我想在运行结果里看到每个case的执行时长，那么就可以写一个装饰器去处理：

import pytest
import functools
import time

def log_execute_time(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        start = time.perf_counter()
        res = func(*args, **kwargs)
        end = time.perf_counter()
        print("{} 用时 {} ms".format(func.__name__, (end - start) * 1000))
        return res
    return wrapper

@log_execute_time
def test_01():
    a = 1
    b = 2
    assert a < b

@log_execute_time
def test_02():
    a = 1
    b = 1
    assert a-b == 0

if __name__ == ‘__main__‘:
    pytest.main([‘-s‘,‘demo_test.py‘])

运行一下：

============================= test session starts =============================
platform win32 -- Python 3.8.5, pytest-6.0.1, py-1.9.0, pluggy-0.13.1
rootdir: D:\练习
collected 2 items

demo_test.py 
test_01 用时 0.0007999999999674934 ms .
test_02 用时 0.0012999999999818712 ms .

============================== 2 passed in 0.03s ==============================
[Finished in 0.4s]

可以看到，每个case执行会打印出执行耗时。

上述的这个例子是在实现自动化测试时的场景，具体还是要看你怎么去挖掘你的需求。

二、其他python开发向的用法

我也去查了下其他的用法，其实在python开发中，用途还是很多的，其中就还有大家熟悉的"身份认证"。

比如大家爱上博客园，你不登录账号也可以浏览博客。当你看着兴起，自己也想评论或者发文的时候，发现
提示你需要登录，这种场景就可以用上装饰器。代码举例：

import functools

def authenticate(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        request = args[0]
        if check_user_logged_in(request): # 检查用户是否登录
            return func(*args, **kwargs) # 如果登录了，就可以执行函数post_comment() 发送评论
        else:
            raise Exception(‘Authentication failed‘) # 否则，身份验证失败
    return wrapper
    
@authenticate
def post_comment(request)
    pass

上述代码只是说了这个意思，辅助描述下场景。