Python-自定义修饰器

最新推荐文章于 2025-05-25 21:16:38 发布

原创最新推荐文章于 2025-05-25 21:16:38 发布 · 903 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

46 篇文章

订阅专栏

本文介绍了一种通过构建修饰器函数来提升代码整洁性的方法，以代码耗时检测为例，展示了如何使用修饰器函数对目标函数进行修饰，并提供了一个判断质数的函数实例，以及如何利用修饰器对其进行性能监控。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言：为了代码的整洁性，一般可以通过构建一个修饰器函数，来对目标函数做修饰。例如：需要对一个代码进行耗时检测，此时就可以将代码计时函数作为一个修饰器函数，然后其作用对象就是目标函数。

代码实现：

# -*- coding:utf-8 -*-
import time

#函数：判断输入数据是否为质数
def is_prime(num):
    if num<2:
        return False
    elif num==2:
        return True
    else:
        for i in range(2,num):
            if num % i ==0:
                return False
        return True

#修饰函数
def code_using_time(func):
    def operation_code(*args):   # *args表示有多个参数可以带入
        t1 = time.time()
        result = func(*args)    #被修饰目标函数
        t2 = time.time()
        print("Total time: {:.4} s".format(t2-t1))
        return result
    return operation_code


#调用修饰函数
@code_using_time
def prime_nums(maxnum):
    count = 0
    result_list = []
    for i in range(0,maxnum):
        if is_prime(i):
            count = count +1
            result_list.append(i)
    return count,result_list

count = prime_nums(10)
print(count)

注：添加修饰器的代码较不加修饰器的代码，运行速度并没有提升。

参考资料：https://www.bilibili.com/video/av25698102?from=search&seid=17702678529678288129