p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed log paths for profiling decorator

This commit is contained in:
VakarisZ 2020-04-30 16:28:46 +03:00
parent 4dcae80a64
commit 9be8d4af1b
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/monkey_island/cc/testing/profiler_decorator.py
View File

@ -1,6 +1,9 @@
from cProfile import Profile from cProfile import Profile
import os
import pstats import pstats
PROFILER_LOG_DIR = "./profiler_logs/"
def profile(sort_args=['cumulative'], print_args=[100]): def profile(sort_args=['cumulative'], print_args=[100]):
profiler = Profile() profiler = Profile()
@ -11,7 +14,11 @@ def profile(sort_args=['cumulative'], print_args=[100]):
try: try:
result = profiler.runcall(fn, *args, **kwargs) result = profiler.runcall(fn, *args, **kwargs)
finally: finally:
filename = _get_filename_for_function(fn) try:
os.mkdir(PROFILER_LOG_DIR)
except os.error:
pass
filename = PROFILER_LOG_DIR + _get_filename_for_function(fn)
with open(filename, 'w') as stream: with open(filename, 'w') as stream:
stats = pstats.Stats(profiler, stream=stream) stats = pstats.Stats(profiler, stream=stream)
stats.strip_dirs().sort_stats(*sort_args).print_stats(*print_args) stats.strip_dirs().sort_stats(*sort_args).print_stats(*print_args)