一个有参装饰器，它可作用于任何函数上

最近Python网络班学员去一个公司面试，遇到了一个很有意思的问题：

请设计一个有参装饰器decorator，它可作用于任何函数上，要求可以接受一个int作为参数，该参数为要求的执行秒数，如果该函数的执行时间大于规定的执行秒数，请打印改函数名字和执行时间。

马哥教育Python网络班企业教练思路：

#我们先来分析下：1、有参数的装饰器 --> 可以使用高阶函数实现2、执行时间 --> 可以使用python的time 库

实现代码：

from functools import wrapsimport time

class TakeTime:    def __init__(self, para: int):        self.para = para
    def __call__(self, func):        @wraps(func)        def wrapped(*args, **kwargs):            start_time = time.time()            name = func.__name__            func(*args, **kwargs)            count_time = time.time() - start_time            if self.para >= count_time:                print("the {name} take {time}, less time".format(name=name, time=count_time))            else:                print("the {name} take {time}, more time".format(name=name, time=count_time))            return func        return wrapped

def func_time(para: int):def decorator(func):        @wraps(func)        def wrapper(*args, **kwargs):            start_time = time.time()            name = func.__name__            func(*args, **kwargs)            count_time = time.time() - start_time            if para >= count_time:                print("the {name} take {time}, less time".format(name=name, time=count_time))            else:                print("the {name} take {time}, more time".format(name=name, time=count_time))            return func        return wrapper    return decorator
@TakeTime(10)def log():    for x in range(10):        time.sleep(0.5)

@func_time(10)def log1():    for x in range(10):        time.sleep(0.5)

代码执行结果：

the log take 5.030340194702148, less time<function log at 0x10216d950>