Upload 4 files

main.py

@@ -35,26 +35,46 @@ async def require_api_key(x_api_key: str | None = Header(default=None)):
 # DB
 # --------------------------
 import pathlib
-
-# 1) DBファイルの場所を /data に（環境変数で上書き可）
-DB_URL = os.getenv("DATABASE_URL")  # 例: sqlite:////data/app.db
-if not DB_URL:
-    # デフォルトは /data/app.db にしておく
-    DB_PATH = "/data/app.db"
-    # パスのディレクトリを作成（存在しなくてもOK）
-    pathlib.Path(os.path.dirname(DB_PATH)).mkdir(parents=True, exist_ok=True)
-    DB_URL = f"sqlite:///{DB_PATH}"
+import os
+from sqlmodel import create_engine
+
+def _ensure_dir(path: str) -> bool:
+    try:
+        pathlib.Path(path).mkdir(parents=True, exist_ok=True)
+        return True
+    except Exception:
+        return False
+
+def _make_sqlite_url(db_path: str) -> str:
+    # 引数は絶対パスを想定
+    return f"sqlite:///{db_path}"
+
+# 1) 環境変数があれば最優先
+DB_URL = os.getenv("DATABASE_URL")
+
+if DB_URL and DB_URL.startswith("sqlite"):
+    # sqlite:///... または sqlite:////...
+    raw = DB_URL.replace("sqlite:///", "", 1)
+    # absolute/relative を正規化
+    raw = raw if raw.startswith("/") else os.path.abspath(raw)
+    dir_ = os.path.dirname(raw)
+    if not _ensure_dir(dir_):
+        # 失敗時は /tmp にフォールバック
+        fallback = "/tmp/app.db"
+        _ensure_dir(os.path.dirname(fallback))
+        DB_URL = _make_sqlite_url(fallback)
 else:
-    # DATABASE_URL が sqlite のときは、必要ならディレクトリを作成
-    if DB_URL.startswith("sqlite"):
-        # sqlite:///relative/path または sqlite:////absolute/path の両対応
-        raw = DB_URL.replace("sqlite:///", "", 1)
-        # 先頭2スラッシュの正規化（'//data/app.db' -> '/data/app.db'）
-        raw = raw[1:] if raw.startswith("/") else raw
-        dir_ = os.path.dirname("/" + raw)  # 先頭に1つ'/'を戻して絶対パス化
-        pathlib.Path(dir_).mkdir(parents=True, exist_ok=True)
-
-# 2) SQLite のスレッド設定（FastAPIと相性良く）
+    # 2) /data → 3) /tmp → 4) ./ の順でフォールバック
+    candidates = ["/data/app.db", "/tmp/app.db", os.path.abspath("./app.db")]
+    chosen = None
+    for p in candidates:
+        if _ensure_dir(os.path.dirname(p)):
+            chosen = p
+            break
+    assert chosen is not None, "No writable location found for SQLite DB"
+    DB_URL = _make_sqlite_url(chosen)
+
+# 3) SQLite のスレッド設定
 engine = create_engine(
     DB_URL,
     echo=False,