fix: server.py 支持环境变量配置DB连接 + deploy_remote.sh 适配远程PG15端口5432 + deploy.sh 设置DB环境变量

This commit is contained in:
selfrelease
2026-07-15 11:41:07 +08:00
parent 2b0c35220a
commit 5aad0eb8f1
3 changed files with 40 additions and 18 deletions
+9 -1
View File
@@ -8,6 +8,7 @@ from __future__ import annotations
import datetime as dt
import json
import os
import pathlib
import re
import subprocess
@@ -27,7 +28,14 @@ from pydantic import BaseModel
ROOT = pathlib.Path(__file__).resolve().parent
STATIC_DIR = ROOT / "static"
SOCKET_DIR = ROOT.parent / "db" / "socket"
DSN = f"host={SOCKET_DIR} port=5434 dbname=wxchat_sales"
# DSN:优先环境变量 DATABASE_URL,其次 DB_HOST/DB_PORT,最后默认本地 socket
if os.environ.get("DATABASE_URL"):
DSN = os.environ["DATABASE_URL"]
elif os.environ.get("DB_HOST"):
DSN = f"host={os.environ['DB_HOST']} port={os.environ.get('DB_PORT', '5432')} dbname={os.environ.get('DB_NAME', 'wxchat_sales')} user={os.environ.get('DB_USER', 'postgres')} password={os.environ.get('DB_PASSWORD', '')}"
else:
DSN = f"host={SOCKET_DIR} port=5434 dbname=wxchat_sales"
app = FastAPI(title="微信营销管理系统 MVP")