profiling_2.py

import json
from typing import Dict
from typing import List
from typing import Union


def export_profiling_events(output_path: str):
    """
    Dump profiling events into a JSON file that can be provided 
    to the Chrome Tracing Viewer
    """
    if not PROFILE:
        return

    events: List[Dict[str, Union[str, int, float]]] = []
    while not PROFILING_EVENTS_QUEUE.empty():
        name, tid, t_begin, t_end = PROFILING_EVENTS_QUEUE.get()
        events.append({"name": name, "ph": "B",
                      "ts": t_begin*1e6, "tid": tid, "pid": 0})
        events.append({"name": name, "ph": "E",
                      "ts": t_end*1e6, "tid": tid, "pid": 0})

    with open(output_path, "w", encoding="utf-8") as _f:
        json.dump({"traceEvents": events}, _f)

    print("Open Chrome, type chrome://tracing/ and "
          f"load the file located at {os.path.abspath(output_path)}")