src/database/queries.py

import sqlite3
import pandas as pd
from src.config.settings import DB_PATH, PROPERTY_CSV, LEASE_CSV


def init_db():
    DB_PATH.parent.mkdir(parents=True, exist_ok=True)
    conn = sqlite3.connect(DB_PATH)

    props_df = pd.read_csv(PROPERTY_CSV)
    props_df.to_sql("properties", conn, if_exists="replace", index=False)

    lease_df = pd.read_csv(LEASE_CSV)
    lease_df.to_sql("leases", conn, if_exists="replace", index=False)

    conn.close()
    print(f"DB initialised at {DB_PATH}")


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