Function run_profiling_and_import

scripts_python/profiling_statistics.py:619–660

(
    n_runs: int = 200,
    max_retries: int = 10,
    profiling_script: str = "./scripts_bash/start_profiling_session.sh",
    extra_args: list = None,
    summaries_dir: str = "profiling/summaries",
    base_dir: str = "profiling",
)

def run_profiling_and_import(
    n_runs: int = 200,
    max_retries: int = 10,
    profiling_script: str = "./scripts_bash/start_profiling_session.sh",
    extra_args: list = None,
    summaries_dir: str = "profiling/summaries",
    base_dir: str = "profiling",
) -> str:
    extra_args = extra_args or []
    summaries_path = Path(summaries_dir)
    summaries_path.mkdir(parents=True, exist_ok=True)

    db_path = generate_db_path(base_dir)
    conn = create_database(db_path)
    conn.close()

    successful_runs = 0
    total_attempts = 0

    with tqdm(total=n_runs, desc="Profiling runs") as pbar:
        while successful_runs < n_runs and total_attempts < n_runs + max_retries:
            pbar.set_postfix_str(f"Attempt {total_attempts + 1}")
            total_attempts += 1

            if run_profiling_session(profiling_script, extra_args):
                latest_folder = get_latest_summary_folder(summaries_path)
                if latest_folder:
                    conn = sqlite3.connect(db_path)
                    if import_single_folder(conn, latest_folder):
                        successful_runs += 1
                        pbar.update(1)
                        pbar.set_postfix_str(f"Success {successful_runs}/{n_runs}")
                    conn.close()

    tqdm.write(
        f"\nCompleted {successful_runs}/{n_runs} successful runs in {total_attempts} attempts"
    )
    tqdm.write(f"Database: {db_path}")

    generate_all_charts(db_path)

    return db_path


if __name__ == "__main__":