How to Create a Database Helper Class for Beginners in Python

import sqlite3


class DatabaseHelper:
    def __init__(self, db_path):
        self.connection = sqlite3.connect(db_path)
        self.cursor = self.connection.cursor()

    def create_table_with_index(self, table_name, columns, indexed_column):
        columns_sql = ", ".join(f"{name} {dtype}" for name, dtype in columns)
        self.cursor.execute(f"CREATE TABLE IF NOT EXISTS {table_name} ({columns_sql})")
        self.cursor.execute(f"CREATE INDEX IF NOT EXISTS idx_{table_name}_{indexed_column} ON {table_name} ({indexed_column})")
        self.connection.commit()

    def insert_many_optimized(self, table_name, data):
        placeholders = ", ".join("?" * len(data[0]))
        self.cursor.executemany(f"INSERT INTO {table_name} VALUES ({placeholders})", data)
        self.connection.commit()

    def query_with_index(self, table_name, indexed_column, value):
        self.cursor.execute(f"SELECT * FROM {table_name} WHERE {indexed_column} = ?", (value,))
        return self.cursor.fetchall()

    def close(self):
        self.connection.close()


if __name__ == "__main__":
    db = DatabaseHelper("scaling_example.db")

    db.create_table_with_index(
        "users",
        [("id", "INTEGER PRIMARY KEY"), ("name", "TEXT"), ("age", "INTEGER")],
        indexed_column="age"
    )

    sample_data = [
        (1, "Alice", 30),
        (2, "Bob", 25),
        (3, "Charlie", 35)
    ]
    db.insert_many_optimized("users", sample_data)

    results = db.query_with_index("users", "age", 25)
    print("Optimized query result:", results)

    db.close()