python面试题，设计装饰器，接收一个int类型的参数，可以用来装饰任何的函数， 如果函数运行的时间大于number，则打印出函数名和函数的运行时间

Posted 2021-10-15 七月的小尾巴

题目：

(面试笔试题)请设计一个装饰器，接收一个int类型的参数number，可以用来装饰任何的函数，
如果函数运行的时间大于number，则打印出函数名和函数的运行时间

import time


def get_time(number):
    # 判断参数类型是否是 int 类型
    if isinstance(number, int):
        def decorator(func):
            def swapper(*args, **kwargs):
                # 定义开始时间
                start_time = time.time()
                res = func(*args, **kwargs)
                end_time = time.time()
                # 计算时间戳毫米级别，如果时间大于number，则打印 函数名称 和运行时间
                if int(round(end_time * 1000)) - int(round(start_time * 1000)) > number:
                    print("函数名称:{0}, 函数运行时间:{1}".format(func.__name__, end_time))
                return res

            return swapper

        return decorator
    else:
        raise TypeError("参数类型不正确")


@get_time("1111")
def test():
    print("开始执行函数")
    time.sleep(3)

拓展知识

在写完上面的装饰器的时候，下面我们来看一下，使用装饰器之后，本来想要直接打印一下test函数的名称和文档注释的，但是发现打印出来的是装饰器的函数内容。

之后去查了一些资料，可以使用 @wraps() 装饰器，用来消除使用装饰器之后的副作用，保留函数之前的属性。

import time
from functools import wraps


def get_time(number):
    # 判断参数类型是否是 int 类型
    if isinstance(number, int):
        def decorator(func):
            @wraps(func)
            def swapper(*args, **kwargs):
                # 定义开始时间
                start_time = time.time()
                res = func(*args, **kwargs)
                end_time = time.time()
                # 计算时间戳毫米级别，如果时间大于number，则打印 函数名称 和运行时间
                if int(round(end_time * 1000)) - int(round(start_time * 1000)) > number:
                    print("函数名称:{0}, 函数运行时间:{1}".format(func.__name__, end_time))
                return res

            return swapper

        return decorator
    else:
        raise TypeError("参数类型不正确")


@get_time(2222)
def test():
    """测试函数"""
    print("开始执行函数")
    time.sleep(3)

返回结果：
test
测试函数

Process finished with exit code 0