优化图片映射

data-pipeline/migrate_image_urls.py

@@ -17,6 +17,23 @@ from pathlib import Path
 import asyncpg
 
 
+# ======================================================================
+# 加载 .env（与环境变量）
+# ======================================================================
+def _load_env():
+    env_file = Path(__file__).resolve().parent.parent / ".env"
+    if env_file.exists():
+        with open(env_file, encoding="utf-8") as fh:
+            for line in fh:
+                line = line.strip()
+                if not line or line.startswith("#") or "=" not in line:
+                    continue
+                key, _, val = line.partition("=")
+                os.environ.setdefault(key.strip(), val.strip())
+
+
+_load_env()
+
 # ======================================================================
 # 配置
 # ======================================================================
@@ -24,7 +41,7 @@ DB_HOST = os.environ.get("POSTGRES_HOST", "localhost")
 DB_PORT = os.environ.get("POSTGRES_PORT", "5432")
 DB_NAME = os.environ.get("POSTGRES_DB", "pharmacopoeia")
 DB_USER = os.environ.get("POSTGRES_USER", "postgres")
-DB_PASSWORD = os.environ.get("POSTGRES_PASSWORD", "postgres")
+DB_PASSWORD = os.environ.get("POSTGRES_PASSWORD", "pharma2025")
 
 # 新旧 URL 前缀
 OLD_PREFIX = "/images/drugs/"