2022.04.14-python学习之装饰器

# File : learning_012_decorators.py.py
# Datetime : 2022/4/14 0014 3:27

import time


# 装饰器
# 本质上,装饰器是一个函数,用来处理其他函数,
# 可以让其他函数在不需要修改代码的前提下增加额外的功能,装饰器返回的对象也是一个函数.

# 常用的装饰器需求场景有:插入日志、性能检测、事务处理、缓存、权限校验等

# 代码演练装饰器场景示例： 代码性能检测,可以选择性的插入日志
def insert_log(flag=False):
    def show_time(function_name):
        def wrapper():
            start_date = time.time()
            function_name()
            time.sleep(1)  # 设置程序停留时长
            end_date = time.time()
            if flag:
                print("日志插入成功!")
            print("程序执行时间为", end_date - start_date)

        return wrapper

    return show_time


# @show_time  # 相当于my_function=show_time(my_function)
@insert_log(True)  # 相当于insert_log=show_time(my_function)
def my_function():
    print("我的测试代码")


my_function()