import sqlite3
import pandas as pd
from config import DB_PATH, COLUMNS_TO_KEEP, TABLE_NAME

def init_db_from_csv(csv_path: str = "weather_data.csv"):
    conn = sqlite3.connect(DB_PATH)

    df = pd.read_csv(csv_path, usecols=COLUMNS_TO_KEEP)
    df.to_sql(TABLE_NAME, conn, if_exists="replace", index=False)

    conn.close()

# init_db_from_csv()

def get_connection():
    return sqlite3.connect(DB_PATH)