python通用运行时间装饰器

@wrapt.decorator
def universal(wrapped, instance, args, kwargs):
    if instance is None:
        if inspect.isclass(wrapped):
            # 装饰一个类
            # Decorator was applied to a class.
            return wrapped(*args, **kwargs)
        else:
           # 装饰一个普通函数或者静态方法
            # Decorator was applied to a function or staticmethod.
            ret = consume_time(wrapped=wrapped, args=args, kwargs=kwargs)
            return ret
    else:
        if inspect.isclass(instance):
            # 装饰类方法
            # Decorator was applied to a classmethod.
            ret = consume_time(wrapped=wrapped, args=args, kwargs=kwargs)
            return ret
        else:
           # 装饰实例方法
            ret = consume_time(wrapped=wrapped, args=args, kwargs=kwargs)
            return ret
           
def consume_time(wrapped, args, kwargs):
    local_time = datetime.datetime.now()
    ret = wrapped(*args, **kwargs)
    times = (datetime.datetime.now() - local_time).seconds
    funcname = wrapped.name
    print(‘运行 {} 函数消耗的时间为 {} 分 {} 秒’.format(
        funcname, times // 60, times % 60))
    # print(‘Run {} time is {} minutes {} seconds’.format(
    #     funcname, times // 60, times % 60))
    return ret

既然说是通用装饰器，怎么个通用法呢？它可以作用于以下内容


* 类
* 普通函数
* 类静态方法
* 类方法
* 实例方法


接下来我们测试一下


【未加装饰器的程序】

import time

class People:
    # 定义基本属性
    name = “”
    age = 0

# 初始化函数
    def init(self, name, age):
        self.name = name
        self.age = age 
    
    # 静态方法
    @staticmethod
    def work():
        time.sleep(3)
        print(“I could work”)

# 类方法
    @classmethod
    def speak(cls, words):
        time.sleep(1)
        print(words)

# 实例方法
    def eat(self, food):
        time.sleep(5)
        print(f"I am eating {food}“)
    
# 普通函数
def read(bookname):
    time.sleep(1)
    print(f"I am reading {bookname}”)

# 调用普通函数
read(“《Linux 二进制分析》”)

# 调用静态方法
People.work()

# 调用类方法
People.speak(“hello world”)

## 调用实例方法
# 实例化一个对象
xiaoming = People(name=“xiaoming”, age=18)
xiaoming.eat(“a sandwich”)

运行后结果



![图片](https://img-blog.csdnimg.cn/img_convert/94d1c3855dfa02a91a99138d7baa185b.png)


现在我们加上时间装饰器，来看看这些函数都消耗了多少时间


【已经加上装饰器的程序】


为了演示装饰不同内容，我们给装饰器每个分支都加上 `print`

import datetime
import time
import wrapt
import inspect

@wrapt.decorator
def universal(wrapped, instance, args, kwargs):
    funcname = wrapped.name
    if instance is None:
        if inspect.isclass(wrapped):
            print(“-” * 50)
            print(“正在装饰一个类”)
            return wrapped(*args, **kwargs)
        else:
            print(“-” * 50)
            print(f"正在装饰一个普通函数或者静态方法: {funcname}“)
            ret = consume_time(wrapped=wrapped, args=args, kwargs=kwargs)
            return ret
    else:
        if inspect.isclass(instance):
            print(”-" * 50)
            print(f"正在装饰类方法: {funcname}“)
            ret = consume_time(wrapped=wrapped, args=args, kwargs=kwargs)
            return ret
        else:
            print(”-" * 50)
            print(f"正在装饰实例方法: {funcname}“)
            ret = consume_time(wrapped=wrapped, args=args, kwargs=kwargs)
            print(”-" * 50)
            return ret

def consume_time(wrapped, args, kwargs):
    local_time = datetime.datetime.now()
    ret = wrapped(*args, **kwargs)
    times = (datetime.datetime.now() - local_time).seconds
    funcname = wrapped.name
    print(‘运行 {} 函数消耗的时间为 {} 分 {} 秒’.format(
        funcname, times // 60, times % 60))
    # print(‘Run {} time is {} minutes {} seconds’.format(
    #     funcname, times // 60, times % 60))
    return ret

@universal
class People:
    # 定义基本属性
    name = “”
    age = 0

# 初始化函数
    def init(self, name, age):
        self.name = name
        self.age = age 
    
    # 静态方法
    @universal
    @staticmethod
    def work():
        time.sleep(3)
        print(“I could work”)

# 类方法
    @universal
    @classmethod
    def speak(cls, words):
        time.sleep(1)
        print(words)

# 实例方法
    @universal
    def eat(self, food):
        time.sleep(5)
        print(f"I am eating {food}“)
    
# 普通函数
@universal
def read(bookname):
    time.sleep(1)
    print(f"I am reading {bookname}”)

# 调用普通函数
read(“《Linux 二进制分析》”)

# 调用静态方法
People.work()

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数网络安全工程师，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年网络安全全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上网络安全知识点！真正的体系化！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上网络安全知识点！真正的体系化！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！