【面试出题】基础-装饰器

实现一个装饰器，统计函数执行的时间和被调用的次数

# !/usr/bin/env python
# coding:utf-8
import time
from functools import wraps

# 第一种 函数装饰器
def timeCounter(func):
    num = 0
    @wraps(func)
    def wrapper(*args, **kwargs):
        nonlocal num
        startTime = time.time()
        func(*args, **kwargs)
        num += 1
        endTime = time.time()
        return (num,round(endTime-startTime, 2))
    return wrapper

@timeCounter
def add(a, b):
    time.sleep(2.115)
    return a+b

print(add(1,2))
print(add(1,3))

# 第二种 类装饰器
class TimeCounter:
    def __init__(self, func):
        self.func = func
        self.count = 0

    def __call__(self, *args, **kwargs):
        start_time = time.time()
        self.func(*args, **kwargs)
        end_time = time.time()
        self.count += 1
        return (self.count, round(end_time - start_time, 2))
@TimeCounter
def add(a, b):
    time.sleep(2.115)
    return a+b
print(add(1, 3))