Function benchmark_reset

scripts/clear_database.py:258–284

Clear benchmark state while preserving the fixed paper collection.

(
    db_path: Path = DB_PATH,
    roles_path: Path = ROLES_FILE,
    backup_dir: Path = BACKUP_DIR,
    roles: Optional[Iterable[str]] = None,
)

def benchmark_reset(
    db_path: Path = DB_PATH,
    roles_path: Path = ROLES_FILE,
    backup_dir: Path = BACKUP_DIR,
    roles: Optional[Iterable[str]] = None,
) -> Dict[str, Any]:
    """Clear benchmark state while preserving the fixed paper collection."""
    backup_path = backup_database(db_path=db_path, backup_dir=backup_dir)
    roles_meta = load_roles_meta(roles_path=roles_path)

    conn = get_connection(db_path=db_path)
    try:
        before_counts = get_table_counts(conn)
        clear_tables(conn, ("behavior_logs", "task_status"))
        seeded_profiles = reset_user_profiles(conn, roles_meta, roles=roles)
        reset_sqlite_sequences(conn, ("profiles", "behavior_logs", "task_status"))
        conn.commit()
        after_counts = get_table_counts(conn)
    finally:
        conn.close()

    return {
        "backup_path": str(backup_path),
        "before_counts": before_counts,
        "after_counts": after_counts,
        "seeded_profiles": seeded_profiles,
    }


def parse_args() -> argparse.Namespace: