Function import_cprofile_data

scripts_python/profiling_statistics.py:185–260

(cursor: sqlite3.Cursor, run_id: int, prof_path: Path)

def import_cprofile_data(cursor: sqlite3.Cursor, run_id: int, prof_path: Path):
    if not prof_path.exists():
        return

    stats = pstats.Stats(str(prof_path))

    cursor.execute(
        """
        UPDATE profiling_runs
        SET total_function_calls = ?,
            primitive_calls = ?,
            total_time_seconds = ?
        WHERE id = ?
    """,
        (stats.total_calls, stats.prim_calls, stats.total_tt, run_id),
    )

    for func_tuple, (cc, nc, tt, ct, callers) in stats.stats.items():
        function_id = get_or_create_function(cursor, func_tuple)

        time_per_call = tt / nc if nc > 0 else 0
        cumulative_per_call = ct / cc if cc > 0 else 0
        time_percentage = (tt / stats.total_tt * 100) if stats.total_tt > 0 else 0

        cursor.execute(
            """
            INSERT INTO function_stats 
            (run_id, function_id, call_count, primitive_call_count, 
             total_time, cumulative_time, time_per_call, cumulative_per_call, time_percentage)
            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
        """,
            (
                run_id,
                function_id,
                nc,
                cc,
                tt,
                ct,
                time_per_call,
                cumulative_per_call,
                time_percentage,
            ),
        )

        for caller_tuple, caller_stats in callers.items():
            caller_function_id = get_or_create_function(cursor, caller_tuple)

            if isinstance(caller_stats, tuple):
                caller_nc, caller_cc, caller_tt, caller_ct = caller_stats
                cursor.execute(
                    """
                    INSERT INTO call_relationships 
                    (run_id, caller_function_id, callee_function_id, call_count, total_time, cumulative_time)
                    VALUES (?, ?, ?, ?, ?, ?)
                """,
                    (run_id, caller_function_id, function_id, caller_nc, caller_tt, caller_ct),
                )
            else: