背景#

装饰器：Python开发者的效率利器！ 🛠️

在Python的世界里，装饰器绝对是一把强大的“瑞士军刀”。它能帮我们优雅地封装通用逻辑，大幅减少重复代码，真正实现事半功倍的开发效率。

如果你是第一次接触装饰器这个概念，强烈建议先找些基础资料了解一下它的核心思想和工作原理（别担心，小编当初也是一头雾水，看别人的代码完全摸不着头脑）。打好基础再往下看，理解起来会顺畅得多，相信小编！

小编环境#

1
import sys
2

3
print('python 版本：',sys.version.split('|')[0])
4
#python 版本： 3.11.11

计算耗时#

想知道函数执行耗时？一个装饰器轻松搞定！⏱️

还在手动写 time.time() 包裹你的函数来计算耗时吗？太麻烦啦！只需几行代码定义一个计时装饰器，轻轻松松给任何函数“戴上”，就能自动获取精准的执行时间

1
def time_it(func):
2
    @wraps(func)  # 保留原始函数的元数据
3
    def wrapper(*args, **kwargs):
4
        print("开始执行---->")
5
        start_time=datetime.datetime.now()
6
        result = func(*args, **kwargs)
7
        end_time=datetime.datetime.now()
8
        print(f"结束执行，消耗时长为：{end_time - start_time}")
9
        return result
10
    return wrapper

重试机制#

在程序中调用外部API、访问数据库或进行网络请求时，网络环境的不稳定性往往是导致程序“意外扑街”的头号元凶！面对这种不可控因素，与其祈祷网络永远畅通，不如主动出击——引入重试机制，来增加程序的健壮性

1
def retry(func=None,*,times=3):
2
    if func is None:
3
        return partial(retry,times=times)
4

5
    @wraps(func)
6
    def wrapper(*args,**kwargs):
7
        for attempt in range(1,times+1):
8
            try:
9
                return func(*args,**kwargs)
10
            except Exception as exc:
11
                print(f"函数 {func.__name__} 进行第 {attempt} 次尝试，遇到错误：{exc}")
12
                sleep(SLEEP_TIME * attempt)
13
        print(f"所有尝试均失败！！！")
14
        return None
15

16
    return wrapper

完整代码#

1
from time import sleep
2
import datetime
3
from functools import wraps,partial  # 导入 wraps 装饰器
4

5
SLEEP_TIME=1
6

7
def time_it(func):
8
    @wraps(func)  # 保留原始函数的元数据
9
    def wrapper(*args, **kwargs):
10
        print("开始执行---->")
11
        start_time=datetime.datetime.now()
12
        result = func(*args, **kwargs)
13
        end_time=datetime.datetime.now()
14
        print(f"结束执行，消耗时长为：{end_time - start_time}")
15
        return result
16
    return wrapper
17

18

19
def retry(func=None,*,times=3):
20
    if func is None:
21
        return partial(retry,times=times)
22

23
    @wraps(func)
24
    def wrapper(*args,**kwargs):
25
        for attempt in range(1,times+1):
26
            try:
27
                return func(*args,**kwargs)
28
            except Exception as exc:
29
                print(f"函数 {func.__name__} 进行第 {attempt} 次尝试，遇到错误：{exc}")
30
                sleep(SLEEP_TIME * attempt)
31
        print(f"所有尝试均失败！！！")
32
        return None
33

34
    return wrapper
35

36

37
if __name__=='__main__':
38

39
    @retry
40
    @time_it
41
    def cal(x):
42
        total=0
43
        for i in range(x):
44
            total += i
45

46
        return total
47

48
    print(cal(100000000))
49
    print(cal('100000'))

运行测试结果

历史相关文章#

以上是自己实践中遇到的一些问题，分享出来供大家参考学习，欢迎关注微信公众号：DataShare ，不定期分享干货