Update src/db.py

src/db.py

@@ -1,25 +1,30 @@
-# src/db.py
 from sqlmodel import SQLModel, create_engine, Session
 import os
 from pathlib import Path
 from dotenv import load_dotenv
 
 load_dotenv()
+
+# 既定は /data（Hugging Face Spaces 等の書き込み可能領域）
 DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:////data/app.db")
 
 def _ensure_sqlite_dir(url: str):
-    # sqlite:////absolute/path.db も sqlite:///relative/path.db も対応
-    if url.startswith("sqlite"):
-        # "sqlite:///" を基準にパス部分を取り出す
-        path_part = url.split("sqlite:///")[-1]
-        p = Path(path_part)
-        # 相対なら CWD 基準。HF Spaces は相対パスは書けないので /data を推奨
-        dirpath = p.parent
-        try:
-            dirpath.mkdir(parents=True, exist_ok=True)
-        except PermissionError:
-            # 読み取り専用領域だった場合は無視（/data を使うのが前提）
-            pass
+    """sqlite URL のパス部分のディレクトリを作成（/data 対応）"""
+    if not url.startswith("sqlite"):
+        return
+    # sqlite:/// 相対 or sqlite://// 絶対 の両対応でパス抽出
+    # 例: sqlite:////data/app.db -> /data/app.db
+    #     sqlite:///./data/app.db -> ./data/app.db
+    parts = url.split("sqlite:///")
+    path_part = parts[-1] if len(parts) > 1 else url.replace("sqlite://", "")
+    # 万一 '////' を含むケースも Path が吸収
+    p = Path(path_part)
+    dirpath = p.parent
+    try:
+        dirpath.mkdir(parents=True, exist_ok=True)
+    except PermissionError:
+        # 読み取り専用領域のときは無視（/data を使う前提）
+        pass
 
 _ensure_sqlite_dir(DATABASE_URL)