【Python基础】装饰器之日志

背景：目前只是简单的写了一个日志，后面继续优化~
希望各位大佬指点，勿喷 ^-^

# -*- coding: utf-8 -*-
# @Time    : 2021/3/31 11:10
# @Author  : Du
# @File    : test.py
# @Projebct:
import datetime
import time
from colorama import Fore, Style
from icecream import ic


def get_run_time(func):
    def wrapper(*args, **kwargs):
        # 定义开始时间
        begin_time = datetime.datetime.now()
        # ret是装饰函数的运行
        ret = func(*args, **kwargs)
        # 定义结束时间
        end_time = datetime.datetime.now()
        # 计算运行时间
        run_time = (end_time - begin_time)
        ic(f'{str(func.__name__)} "函数运行时间为": {str(run_time)}')
        return ret

    return wrapper


def print_logger(func1):
    def out_log(*args, **kwargs):
        now_time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
        res = func1(*args, **kwargs)
        lines = func1.__code__
        _ = (str(lines).split(",")[-1])
        print(
            f'{Fore.BLUE} {now_time} {Fore.MAGENTA} | INFO | {Fore.CYAN}'
            f'{Fore.RED} {func1.__module__}: {func1.__name__} {Style.BRIGHT + Fore.GREEN}{_}')
        return res

    return out_log


@get_run_time
@print_logger
def test01():
    time.sleep(0.05)


if __name__ == '__main__':
    test01()