import sys
from functools import wraps
from termcolor import cprint
from line_profiler import LineProfiler
def line_used_time_parse(func):
@wraps(func)
def wrap(*args, **kwargs):
profile = LineProfiler(func)
profile.enable()
cprint(f"start run:{func.__name__}".center(30, "="), "green")
res = func(*args, **kwargs)
profile.disable()
profile.print_stats(sys.stdout, output_unit=1e-3)
return res
return wrap