Python 装饰器 运行时间

import time

def timer(function):
    """
    装饰器函数timer
    :param function:想要计时的函数
    :return:
    """
    def wrapper(*args, **kwargs):
        time_start = time.time()
        res = function(*args, **kwargs)
        cost_time = time.time() - time_start
        print(f"【{function.__name__}】运行时间：【{cost_time}】秒")
        return res
    return wrapper

@timer
def main():
    time.sleep(1)