import sqlite3, os
from contextlib import contextmanager

DB_PATH = os.environ.get("DATABASE_PATH", "readitlater.db")

def init_db():
    conn = sqlite3.connect(DB_PATH)
    conn.executescript(open("schema.sql", "r", encoding="utf-8").read())
    conn.commit()
    conn.close()

@contextmanager
def get_db():
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    try:
        yield conn
    finally:
        conn.close()