You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
466 B
Python
23 lines
466 B
Python
2 years ago
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import cProfile
|
||
|
|
from functools import wraps
|
||
|
|
import pstats
|
||
|
|
|
||
|
|
|
||
|
|
LIST_MAX_FUNCTION_COUNT = 40
|
||
|
|
|
||
|
|
|
||
|
|
def profile(fun):
|
||
|
|
@wraps(fun)
|
||
|
|
def _fun(*args, **kwargs):
|
||
|
|
with cProfile.Profile() as pf:
|
||
|
|
ret = fun(*args, **kwargs)
|
||
|
|
pstats.Stats(pf).sort_stats(pstats.SortKey.CUMULATIVE).print_stats(
|
||
|
|
"entertainment_decider",
|
||
|
|
LIST_MAX_FUNCTION_COUNT,
|
||
|
|
)
|
||
|
|
return ret
|
||
|
|
|
||
|
|
return _fun
|