pyt/profiling/profiler.py at 8githubScanRefactor · python-security/pyt · GitHub

Repository navigation

import argparse

from . import fine_timer

parser = argparse.ArgumentParser()

parser.add_argument('project_file', help='Path to the file where PyT should start to analyse.', type=str)

parser.add_argument('-pr', '--project', help='Path to where the project is located which PyT should analyse.', type=str)

parser.add_argument('-n', '--number-of-results', help='Number of results to be shown. Default: 10.', type=int)

parser.add_argument('-v', '--visualise', help='Visualise the results in an interactive way.', action='store_true')

parser.add_argument('-fp', '--fixed-point', help='Run line profiling on the fixed point algorithm.', action='store_true')

parser.add_argument('-k', '--keep-intermediate-files', help='Keep files from cProfile and the line profiler.', action='store_true')

args = parser.parse_args()

number_of_results = 10

if args.number_of_results:

number_of_results = args.number_of_results

fine_timer.run(args.project, args.project_file, number_of_results)

if args.fixed_point:

fine_timer.fixed_point_timer(args.project, args.project_file)

if args.visualise:

try:

print()

print('#############################')

print('C-c to to clean up and exit.')

print('#############################')

print()

fine_timer.visualise()

except KeyboardInterrupt:

pass

if not args.keep_intermediate_files:

fine_timer.clean_up()