109 lines
3.2 KiB
Python
109 lines
3.2 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
便捷启动脚本 - 支持命令行参数指定环境
|
|||
|
|
用法：
|
|||
|
|
python start.py # 使用默认配置
|
|||
|
|
python start.py --env dev # 开发环境
|
|||
|
|
python start.py --env test # 测试环境
|
|||
|
|
python start.py --env prod # 生产环境
|
|||
|
|
python start.py --port 9000 # 临时指定端口
|
|||
|
|
"""
|
|||
|
|
import argparse
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(description="启动 FastAPI 应用")
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--env",
|
|||
|
|
choices=["dev", "test", "prod"],
|
|||
|
|
help="指定运行环境 (dev/test/prod)"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--host",
|
|||
|
|
help="指定主机地址"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--port",
|
|||
|
|
type=int,
|
|||
|
|
help="指定端口号"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--workers",
|
|||
|
|
type=int,
|
|||
|
|
help="指定 worker 数量（生产环境）"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--reload",
|
|||
|
|
action="store_true",
|
|||
|
|
help="启用热重载"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--log-level",
|
|||
|
|
choices=["critical", "error", "warning", "info", "debug"],
|
|||
|
|
help="日志级别"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
# 设置环境变量
|
|||
|
|
if args.env:
|
|||
|
|
env_map = {"dev": "development", "test": "testing", "prod": "production"}
|
|||
|
|
os.environ["ENVIRONMENT"] = env_map[args.env]
|
|||
|
|
print(f"[INFO] 启动环境: {env_map[args.env]} (配置文件: .env.{args.env})")
|
|||
|
|
|
|||
|
|
# 临时覆盖配置
|
|||
|
|
if args.host:
|
|||
|
|
os.environ["HOST"] = args.host
|
|||
|
|
if args.port:
|
|||
|
|
os.environ["PORT"] = str(args.port)
|
|||
|
|
if args.workers:
|
|||
|
|
os.environ["WORKERS"] = str(args.workers)
|
|||
|
|
if args.reload:
|
|||
|
|
os.environ["RELOAD"] = "True"
|
|||
|
|
if args.log_level:
|
|||
|
|
os.environ["LOG_LEVEL"] = args.log_level
|
|||
|
|
|
|||
|
|
# 导入并启动应用
|
|||
|
|
try:
|
|||
|
|
from main import app
|
|||
|
|
from app.core.config import settings
|
|||
|
|
import uvicorn
|
|||
|
|
|
|||
|
|
print(f"[CONFIG] 当前配置:")
|
|||
|
|
print(f" - 环境: {settings.ENVIRONMENT}")
|
|||
|
|
print(f" - 主机: {settings.HOST}")
|
|||
|
|
print(f" - 端口: {settings.PORT}")
|
|||
|
|
print(f" - 调试: {settings.DEBUG}")
|
|||
|
|
print(f" - 热重载: {settings.RELOAD}")
|
|||
|
|
print(f" - Workers: {settings.WORKERS}")
|
|||
|
|
print(f" - 日志级别: {settings.LOG_LEVEL}")
|
|||
|
|
|
|||
|
|
if settings.DOCS_URL:
|
|||
|
|
print(f"[DOCS] API 文档: http://{settings.HOST}:{settings.PORT}{settings.DOCS_URL}")
|
|||
|
|
|
|||
|
|
print("-" * 50)
|
|||
|
|
|
|||
|
|
# 根据环境决定是否使用字符串形式的 app（支持热重载）
|
|||
|
|
app_str = "main:app" if settings.RELOAD else app
|
|||
|
|
|
|||
|
|
uvicorn.run(
|
|||
|
|
app_str,
|
|||
|
|
host=settings.HOST,
|
|||
|
|
port=settings.PORT,
|
|||
|
|
reload=settings.RELOAD,
|
|||
|
|
log_level=settings.LOG_LEVEL.lower(),
|
|||
|
|
workers=1 if settings.RELOAD else settings.WORKERS,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except ImportError as e:
|
|||
|
|
print(f"[ERROR] 导入错误: {e}")
|
|||
|
|
print("请确保在正确的目录中运行此脚本")
|
|||
|
|
sys.exit(1)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 启动失败: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|