記帳網站（三）：換上雲端資料庫，正式上線

"""資料庫連線：有環境變數 DATABASE_URL 時連到 Postgres（Neon），沒有時使用本機 SQLite。

兩種資料庫的差異都集中在這個檔案：
- 參數符號：SQLite 用 ?，Postgres 用 %s。程式其他地方一律寫 ?，由 run() 轉換。
- 建表語法：分別放在 schema_sqlite.sql 與 schema_postgres.sql。
"""
import os
import pathlib
import sqlite3

from flask import current_app, g

HERE = pathlib.Path(__file__).resolve().parent


def database_url():
    return current_app.config.get("DATABASE_URL") or os.environ.get("DATABASE_URL")


def is_postgres():
    return bool(database_url())


def get_db():
    """每個請求共用一個連線，查詢結果一律是 dict。"""
    if "db" not in g:
        if is_postgres():
            import psycopg
            from psycopg.rows import dict_row
            g.db = psycopg.connect(database_url(), row_factory=dict_row)
        else:
            path = pathlib.Path(current_app.config["SQLITE_PATH"])
            path.parent.mkdir(parents=True, exist_ok=True)
            g.db = sqlite3.connect(path)
            g.db.row_factory = lambda cur, row: {d[0]: v for d, v in zip(cur.description, row)}
    return g.db


def close_db(_exc=None):
    db = g.pop("db", None)
    if db is not None:
        db.close()


def run(sql, params=()):
    """執行一段 SQL 並回傳所有結果。SQL 中的參數一律寫 ?。"""
    if is_postgres():
        sql = sql.replace("?", "%s")
    cur = get_db().execute(sql, params)
    return cur.fetchall() if cur.description else []


def commit():
    get_db().commit()


def init_db(seed=True):
    """建立資料表，並放入示範資料（會清除原有資料）。"""
    schema = "schema_postgres.sql" if is_postgres() else "schema_sqlite.sql"
    db = get_db()
    scripts = [(HERE / schema).read_text(encoding="utf-8")]
    if seed:
        scripts.append((HERE / "seed.sql").read_text(encoding="utf-8"))
    for script in scripts:
        if is_postgres():
            db.execute(script)
        else:
            db.executescript(script)
    db.commit()