常用小工具

常用的小工具，将持续更新

import time


def timecounter(func):
    def wrapper(*args,**kwargs):
        start_time = time.time()
        result = func(*args,**kwargs)
        end_time = time.time()
        print(f'----------->{func.__name__} cost {end_time-start_time}')
        return result
    return wrapper



def resault_cache(func):
    cache = {}
    def inner(*args,**kwargs):
        if args in cache:
            return cache[args]
        result = func(*args,**kwargs)
        cache[args] = result
        return result
    return inner


def validate_data(func):
    def inner(*args,**kwargs):
        valide = None
        # 验证过程
        if valide:
            return func(*args,**kwargs)
        else:
            raise ValueError("invalid data")

    return inner

def data_validate(func):
    def inner(*args,**kwargs):
        result = func(*args,**kwargs)
        valide_data = None
        # 验证数据是否符合要求的步骤
        if valide_data:
            return result
        else:
            raise ValueError("invalid data,check data")
    return inner

def log_record(func):
    def inner(*args,**kwargs):
        result = func(*args,**kwargs)
        with open('result.log','a') as log_file:
            log_file.write("----->需要记录的数据")
        return result
    return inner

def handle_error(func):

    def inner(*args,**kwargs):
        try:
            return func(*args,**kwargs)
        except Exception as e:
            with open('error.log', 'a') as log_file:
                log_file.write(f'Error in {func.__name__}----->{e}')

            return None

    return inner

# 调用的时候需要注意，当前装饰器，是装饰器工厂函数
def retry(max_count,delay):
    def wrapper(func):
        def inner(*args,**kwargs):
            count = 0
            while count< max_count:
                try:
                    return func(*args,**kwargs)
                except Exception as e:
                    print(f'--->{count+1} failed.retry in {delay} seconds')
                    count += 1
                    time.sleep(delay)
            raise Exception("Max Retry count")
        return inner
    return wrapper


def visualization(func):
    import matplotlib.pyplot as plt
    
    def inner(*args,**kwargs):
        result = func(*args,**kwargs)
        plt.figure()
        #可视化代码，优化
        plt.show()
        return result
    return inner

def debug(func):
    def inner(*args,**kwargs):
        print(f'{func.__name__},args is {args},kwargs{kwargs}')
        return func(*args,**kwargs)
    return inner


def warning_handler(func):
    
    def inner(*args,**kwargs):
        import warnings
        warnings.warn(f'{func.__name__} is not supported,will be removed')
        return func(*args,**kwargs) 
    return inner