cProfile, как записать данные в файл

Question

Я новичок в использовании Python, и меня просили регистрировать профиль каждый раз, когда я запускаю симуляцию. Моя симуляция имеет 4 класса и 10 методов, которые работают в цикле, пока не будет достигнуто определенное условие.

Я написал следующий файл, который создает файл:

LOG_FILE = open(os.path.join(os.getcwd(), "logs", "Log_Log_{}.txt".format(
   str(datetime.datetime.now()).replace(":", "_"))), mode="w") 

В самом низу скрипта я добавил:

if __name__ == "__main__":
    cProfile.run("main()", LOG_FILE, sort="tottime")

Почему мой файл Log_Log пуст и cProfile ничего не возвращает?

Answer 1

используйте pstats для вывода вывода cProfile в удобочитаемом формате

def sample():
   # initialize cProfile
   profiler_object = cProfile.Profile()
   profiler_object.enable()

   # execute something here
   a = [i for i in range(0, 100)]

   profiler_object.disable()

   # dump the profiler stats 
   s = io.StringIO()
   ps = pstats.Stats(profiler_object, stream=s).sort_stats('cumulative')
   ps.dump_stats('enter your dump file path here')

   # convert to human readable format
   out_stream = open('enter your log file path here', 'w')
   ps = pstats.Stats('enter your dump file path here', stream=out_stream)
   ps.strip_dirs().sort_stats('cumulative').print_stats()
   return True

sample()

Пример вывода:

Sat Sep  8 07:07:34 2018    your_dump_file_path

     2 function calls in 0.000 seconds

     Ordered by: cumulative time

     ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    0.000    0.000    0.000    0.000    <ipython-input-31-4e18ee895e20>:4(<listcomp>)
        1    0.000    0.000    0.000    0.000    {method 'disable' of '_lsprof.Profiler' objects}

Вы можете импровизировать, используя декораторов

def profiler(func):
     def wrapper(*args, **kwargs):
         # profiler initialization 
         res = func(*args, **kwargs)
         # further processing
     return wrapper


@profiler
def a():
    print('hello world')