diff options
Diffstat (limited to 'bootstrap.py')
| -rw-r--r-- | bootstrap.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/bootstrap.py b/bootstrap.py index 1b4d484..e50cb38 100644 --- a/bootstrap.py +++ b/bootstrap.py @@ -57,6 +57,12 @@ args.add_argument( default=False, help='Should we run cProfile on this code?', ) +args.add_argument( + '--trace_memory', + action=ActionNoYes, + default=False, + help='Should we record/report on memory utilization?', +) original_hook = sys.excepthook @@ -221,11 +227,25 @@ def initialize(entry_point): else: config.parse(None) + if config.config['trace_memory']: + import tracemalloc + + tracemalloc.start() + # Initialize logging... and log some remembered messages from # config module. logging_utils.initialize_logging(logging.getLogger()) config.late_logging() + # Maybe log some info about the python interpreter itself. + logger.debug( + f'Platform: {sys.platform}, maxint=0x{sys.maxsize:x}, byteorder={sys.byteorder}' + ) + logger.debug(f'Python interpreter version: {sys.version}') + logger.debug(f'Python implementation: {sys.implementation}') + logger.debug(f'Python C API version: {sys.api_version}') + logger.debug(f'Python path: {sys.path}') + # Allow programs that don't bother to override the random seed # to be replayed via the commandline. import random @@ -265,6 +285,14 @@ def initialize(entry_point): logger.debug(f'{entry_point.__name__} (program entry point) returned {ret}.') + if config.config['trace_memory']: + snapshot = tracemalloc.take_snapshot() + top_stats = snapshot.statistics('lineno') + print() + print("--trace_memory's top 10 memory using files:") + for stat in top_stats[:10]: + print(stat) + if config.config['dump_all_objects']: dump_all_objects() |