201 lines
7.1 KiB
Python
201 lines
7.1 KiB
Python
"""
|
||
FastAPI应用主入口
|
||
"""
|
||
from fastapi import FastAPI
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from api.routes import config, trades, stats, dashboard, account, recommendations, system
|
||
import os
|
||
import logging
|
||
from pathlib import Path
|
||
from logging.handlers import RotatingFileHandler
|
||
|
||
# 加载.env文件
|
||
try:
|
||
from dotenv import load_dotenv
|
||
backend_dir = Path(__file__).parent.parent
|
||
project_root = backend_dir.parent
|
||
env_file = backend_dir / '.env'
|
||
if not env_file.exists():
|
||
env_file = project_root / '.env'
|
||
if env_file.exists():
|
||
load_dotenv(env_file)
|
||
else:
|
||
load_dotenv(project_root / '.env', override=False)
|
||
except ImportError:
|
||
pass
|
||
except Exception:
|
||
pass
|
||
|
||
# 配置日志
|
||
def setup_logging():
|
||
"""配置日志系统"""
|
||
# 获取日志级别
|
||
log_level = os.getenv('LOG_LEVEL', 'INFO').upper()
|
||
log_level_map = {
|
||
'DEBUG': logging.DEBUG,
|
||
'INFO': logging.INFO,
|
||
'WARNING': logging.WARNING,
|
||
'ERROR': logging.ERROR,
|
||
'CRITICAL': logging.CRITICAL
|
||
}
|
||
level = log_level_map.get(log_level, logging.INFO)
|
||
|
||
# 创建日志目录
|
||
backend_dir = Path(__file__).parent.parent
|
||
log_dir = backend_dir / 'logs'
|
||
log_dir.mkdir(exist_ok=True)
|
||
|
||
# 日志文件路径
|
||
log_file = log_dir / 'api.log'
|
||
|
||
# 配置根日志记录器
|
||
root_logger = logging.getLogger()
|
||
root_logger.setLevel(level)
|
||
|
||
# 清除现有的处理器
|
||
root_logger.handlers.clear()
|
||
|
||
# 设置日志时间格式为北京时间（UTC+8）
|
||
from datetime import datetime, timezone, timedelta
|
||
|
||
class BeijingTimeFormatter(logging.Formatter):
|
||
"""使用北京时间的日志格式化器"""
|
||
def formatTime(self, record, datefmt=None):
|
||
# 转换为北京时间（UTC+8）
|
||
beijing_tz = timezone(timedelta(hours=8))
|
||
dt = datetime.fromtimestamp(record.created, tz=beijing_tz)
|
||
if datefmt:
|
||
return dt.strftime(datefmt)
|
||
return dt.strftime('%Y-%m-%d %H:%M:%S')
|
||
|
||
# 创建格式器（使用北京时间）
|
||
formatter = BeijingTimeFormatter(
|
||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||
)
|
||
|
||
# 文件处理器（带轮转）
|
||
file_handler = RotatingFileHandler(
|
||
log_file,
|
||
maxBytes=10 * 1024 * 1024, # 10MB
|
||
backupCount=5,
|
||
encoding='utf-8'
|
||
)
|
||
file_handler.setLevel(level)
|
||
file_handler.setFormatter(formatter)
|
||
root_logger.addHandler(file_handler)
|
||
|
||
# 控制台处理器
|
||
console_handler = logging.StreamHandler()
|
||
console_handler.setLevel(level)
|
||
console_handler.setFormatter(formatter)
|
||
root_logger.addHandler(console_handler)
|
||
|
||
# 追加：将 ERROR 日志写入 Redis（不影响现有文件/控制台日志）
|
||
try:
|
||
from api.redis_log_handler import RedisErrorLogHandler, RedisLogConfig
|
||
|
||
redis_url = os.getenv("REDIS_URL", "redis://localhost:6379")
|
||
redis_use_tls = os.getenv("REDIS_USE_TLS", "False").lower() == "true"
|
||
redis_username = os.getenv("REDIS_USERNAME", None)
|
||
redis_password = os.getenv("REDIS_PASSWORD", None)
|
||
ssl_cert_reqs = os.getenv("REDIS_SSL_CERT_REQS", "required")
|
||
ssl_ca_certs = os.getenv("REDIS_SSL_CA_CERTS", None)
|
||
|
||
redis_cfg = RedisLogConfig(
|
||
redis_url=redis_url,
|
||
use_tls=redis_use_tls,
|
||
username=redis_username,
|
||
password=redis_password,
|
||
ssl_cert_reqs=ssl_cert_reqs,
|
||
ssl_ca_certs=ssl_ca_certs,
|
||
service="backend",
|
||
)
|
||
redis_handler = RedisErrorLogHandler(redis_cfg)
|
||
# 让 handler 自己按组筛选（error/warning/info），这里只需要放宽到 INFO
|
||
redis_handler.setLevel(logging.INFO)
|
||
root_logger.addHandler(redis_handler)
|
||
except Exception:
|
||
pass
|
||
|
||
# 设置第三方库的日志级别
|
||
logging.getLogger('uvicorn').setLevel(logging.WARNING)
|
||
logging.getLogger('uvicorn.access').setLevel(logging.WARNING)
|
||
|
||
return log_file
|
||
|
||
# 初始化日志
|
||
log_file = setup_logging()
|
||
logger = logging.getLogger(__name__)
|
||
logger.info(f"日志系统已初始化，日志文件: {log_file}")
|
||
logger.info(f"日志级别: {os.getenv('LOG_LEVEL', 'INFO')}")
|
||
|
||
# 检查 redis-py 是否可用（redis-py 4.2+ 同时支持同步和异步，可替代aioredis）
|
||
try:
|
||
import redis
|
||
# 检查是否是 redis-py 4.2+（支持异步）
|
||
if hasattr(redis, 'asyncio'):
|
||
logger.info(f"✓ redis-py 已安装 (版本: {redis.__version__ if hasattr(redis, '__version__') else '未知'})，支持同步和异步客户端")
|
||
logger.info(f" - redis.Redis: 同步客户端（用于config_manager）")
|
||
logger.info(f" - redis.asyncio.Redis: 异步客户端（用于trading_system，可替代aioredis）")
|
||
else:
|
||
logger.warning("⚠ redis-py 版本可能过低，建议升级到 4.2+ 以获得异步支持")
|
||
except ImportError as e:
|
||
import sys
|
||
logger.warning("⚠ redis-py 未安装，Redis/Valkey 缓存将不可用")
|
||
logger.warning(f" Python 路径: {sys.executable}")
|
||
logger.warning(f" 导入错误: {e}")
|
||
logger.warning(f" 提示: 请运行 'pip install redis>=4.2.0' 安装 redis-py")
|
||
logger.warning(f" 注意: redis-py 4.2+ 同时支持同步和异步，无需安装 aioredis")
|
||
logger.warning(f" 或者运行 'pip install -r backend/requirements.txt' 安装所有依赖")
|
||
|
||
app = FastAPI(
|
||
title="Auto Trade System API",
|
||
version="1.0.0",
|
||
description="币安自动交易系统API",
|
||
redirect_slashes=False # 禁用自动重定向，避免307重定向问题
|
||
)
|
||
|
||
# CORS配置（允许React前端访问）
|
||
# 默认包含：本地开发端口、主前端域名、推荐查看器域名
|
||
cors_origins_str = os.getenv('CORS_ORIGINS', 'http://localhost:3000,http://localhost:3001,http://localhost:5173,http://as.deepx1.com,http://asapi.deepx1.com,http://r.deepx1.com,https://r.deepx1.com')
|
||
cors_origins = [origin.strip() for origin in cors_origins_str.split(',') if origin.strip()]
|
||
|
||
logger.info(f"CORS允许的源: {cors_origins}")
|
||
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=cors_origins,
|
||
allow_credentials=True,
|
||
allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "PATCH"],
|
||
allow_headers=["*"],
|
||
expose_headers=["*"],
|
||
)
|
||
|
||
# 注册路由
|
||
app.include_router(config.router, prefix="/api/config", tags=["配置管理"])
|
||
app.include_router(trades.router, prefix="/api/trades", tags=["交易记录"])
|
||
app.include_router(stats.router, prefix="/api/stats", tags=["统计分析"])
|
||
app.include_router(dashboard.router, prefix="/api/dashboard", tags=["仪表板"])
|
||
app.include_router(account.router, prefix="/api/account", tags=["账户数据"])
|
||
app.include_router(recommendations.router, tags=["交易推荐"])
|
||
app.include_router(system.router, tags=["系统控制"])
|
||
|
||
|
||
@app.get("/")
|
||
async def root():
|
||
return {
|
||
"message": "Auto Trade System API",
|
||
"version": "1.0.0",
|
||
"docs": "/docs"
|
||
}
|
||
|
||
|
||
@app.get("/api/health")
|
||
async def health():
|
||
return {"status": "ok"}
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
uvicorn.run(app, host="0.0.0.0", port=8001)
|