fix: server.py 支持环境变量配置DB连接 + deploy_remote.sh 适配远程PG15端口5432 + deploy.sh 设置DB环境变量
This commit is contained in:
+9
-1
@@ -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")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user