Function create_database

scripts_python/profiling_statistics.py:26–140

(db_path: str)

def create_database(db_path: str) -> sqlite3.Connection:
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()

    cursor.execute("""
        CREATE TABLE IF NOT EXISTS profiling_runs (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            folder_name TEXT UNIQUE NOT NULL,
            timestamp DATETIME,
            execution_time_seconds REAL,
            n_concurrent_trials INTEGER,
            disable_diffs BOOLEAN,
            agent_name TEXT,
            cprofile_exists BOOLEAN,
            flamegraph_exists BOOLEAN,
            args_exists BOOLEAN,
            total_function_calls INTEGER,
            primitive_calls INTEGER,
            total_time_seconds REAL,
            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    """)

    cursor.execute("CREATE INDEX IF NOT EXISTS idx_runs_timestamp ON profiling_runs(timestamp)")
    cursor.execute("CREATE INDEX IF NOT EXISTS idx_runs_agent_name ON profiling_runs(agent_name)")

    cursor.execute("""
        CREATE TABLE IF NOT EXISTS functions (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            filename TEXT NOT NULL,
            line_number INTEGER NOT NULL,
            function_name TEXT NOT NULL,
            module_name TEXT,
            is_builtin BOOLEAN,
            UNIQUE(filename, line_number, function_name)
        )
    """)

    cursor.execute("CREATE INDEX IF NOT EXISTS idx_functions_name ON functions(function_name)")
    cursor.execute("CREATE INDEX IF NOT EXISTS idx_functions_filename ON functions(filename)")
    cursor.execute("CREATE INDEX IF NOT EXISTS idx_functions_module ON functions(module_name)")

    cursor.execute("""
        CREATE TABLE IF NOT EXISTS function_stats (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            run_id INTEGER NOT NULL,
            function_id INTEGER NOT NULL,
            call_count INTEGER NOT NULL,
            primitive_call_count INTEGER NOT NULL,
            total_time REAL NOT NULL,
            cumulative_time REAL NOT NULL,
            time_per_call REAL,
            cumulative_per_call REAL,
            time_percentage REAL,
            FOREIGN KEY (run_id) REFERENCES profiling_runs(id) ON DELETE CASCADE,
            FOREIGN KEY (function_id) REFERENCES functions(id) ON DELETE CASCADE,
            UNIQUE(run_id, function_id)
        )