178 lines
5.5 KiB
Python
178 lines
5.5 KiB
Python
# filename: main.py
|
|
|
|
import subprocess
|
|
import os
|
|
import logging
|
|
from collections import deque
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI, HTTPException, Query
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.responses import FileResponse
|
|
|
|
# 引入调度器
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
|
|
# 复用现有manager
|
|
from core.manager import StrategyManager
|
|
|
|
# ================== 初始化 ==================
|
|
app = FastAPI(title="策略控制台")
|
|
manager = StrategyManager()
|
|
|
|
# 初始化调度器
|
|
scheduler = AsyncIOScheduler()
|
|
|
|
# 配置日志
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# ================== 新增：获取 Git 版本信息 ==================
|
|
def get_git_commit_info():
|
|
"""
|
|
[智能版] 获取 Git 仓库的最新提交信息。
|
|
此函数会自动从当前文件位置向上查找项目根目录（.git 文件夹所在位置）。
|
|
"""
|
|
try:
|
|
# 1. 获取 main.py 所在的目录
|
|
script_dir = Path(__file__).resolve().parent
|
|
|
|
# 2. 推断出项目根目录 (即 main.py 所在目录的上一级)
|
|
project_root = script_dir.parent
|
|
|
|
# 3. 检查项目根目录下是否存在 .git 文件夹
|
|
git_dir = project_root / ".git"
|
|
if not git_dir.is_dir():
|
|
return "Git repo not found in parent directory"
|
|
|
|
# 4. 在推断出的项目根目录下执行 git 命令
|
|
# 使用 cwd (current working directory) 参数是关键
|
|
result = subprocess.run(
|
|
['git', 'log', '-1', '--pretty=format:%h - %s (%cr)'],
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
check=True,
|
|
encoding='utf-8',
|
|
cwd=project_root # <--- 在这里指定命令的执行目录
|
|
)
|
|
return result.stdout.strip()
|
|
except (FileNotFoundError, subprocess.CalledProcessError) as e:
|
|
logger.warning(f"无法获取 Git 提交信息: {e}")
|
|
return "获取 Git 信息失败"
|
|
|
|
# ================== 定时任务逻辑 (保持不变) ==================
|
|
|
|
def scheduled_restart_task():
|
|
"""
|
|
定时任务：重启所有正在运行的策略
|
|
"""
|
|
logger.info("⏰ [定时任务] 触发自动重启流程...")
|
|
|
|
# 获取当前所有策略状态
|
|
status = manager.get_status()
|
|
running_strategies = [
|
|
name for name, info in status['strategies'].items()
|
|
if info['status'] == 'running'
|
|
]
|
|
|
|
if not running_strategies:
|
|
logger.info("⏰ [定时任务] 当前无运行中的策略，无需重启")
|
|
return
|
|
|
|
logger.info(f"⏰ [定时任务] 即将重启以下策略: {running_strategies}")
|
|
|
|
for name in running_strategies:
|
|
try:
|
|
manager.restart_strategy(name)
|
|
logger.info(f"✅ [定时任务] {name} 重启成功")
|
|
except Exception as e:
|
|
logger.error(f"❌ [定时任务] {name} 重启失败: {e}")
|
|
|
|
logger.info("⏰ [定时任务] 自动重启流程结束")
|
|
|
|
|
|
# ================== FastAPI 事件钩子 (保持不变) ==================
|
|
|
|
@app.on_event("startup")
|
|
async def start_scheduler():
|
|
scheduler.add_job(
|
|
scheduled_restart_task,
|
|
CronTrigger(hour=8, minute=58),
|
|
id="restart_morning",
|
|
replace_existing=True
|
|
)
|
|
scheduler.add_job(
|
|
scheduled_restart_task,
|
|
CronTrigger(hour=20, minute=58),
|
|
id="restart_evening",
|
|
replace_existing=True
|
|
)
|
|
scheduler.start()
|
|
logger.info("📅 定时任务调度器已启动 (计划时间: 08:58, 20:58)")
|
|
|
|
|
|
@app.on_event("shutdown")
|
|
async def stop_scheduler():
|
|
scheduler.shutdown()
|
|
|
|
|
|
# ================== API 路由 ==================
|
|
|
|
@app.get("/api/status")
|
|
def get_status():
|
|
# [核心修改] 在返回状态的同时，注入 Git 版本信息
|
|
status_data = manager.get_status()
|
|
status_data['git_info'] = get_git_commit_info()
|
|
return status_data
|
|
|
|
|
|
@app.post("/api/strategy/{name}/start")
|
|
def start_strategy(name: str):
|
|
if manager.start_strategy(name):
|
|
return {"success": True}
|
|
raise HTTPException(400, "启动失败")
|
|
|
|
|
|
@app.post("/api/strategy/{name}/stop")
|
|
def stop_strategy(name: str):
|
|
if manager.stop_strategy(name):
|
|
return {"success": True}
|
|
raise HTTPException(400, "停止失败")
|
|
|
|
|
|
@app.post("/api/strategy/{name}/restart")
|
|
def restart_strategy(name: str):
|
|
if manager.restart_strategy(name):
|
|
return {"success": True}
|
|
raise HTTPException(400, "重启失败，请检查日志")
|
|
|
|
|
|
@app.get("/api/logs/{name}")
|
|
def get_logs(name: str, lines: int = Query(50, le=500)):
|
|
try:
|
|
if '_' not in name:
|
|
return {"lines": []}
|
|
strategy_name, symbol = name.split('_', 1)
|
|
log_file = Path(f"logs/{strategy_name}/{symbol}.log")
|
|
|
|
if not log_file.exists():
|
|
return {"lines": ["日志文件尚未生成"]}
|
|
|
|
with open(log_file, 'r', encoding='utf-8', errors='replace') as f:
|
|
last_lines = deque(f, maxlen=lines)
|
|
return {"lines": [line.rstrip() for line in last_lines]}
|
|
except Exception as e:
|
|
raise HTTPException(500, f"读取日志失败: {e}")
|
|
|
|
|
|
# ================== 静态文件挂载 (保持不变) ==================
|
|
# 注意: 这里的路径是示例，请确保它与你的项目结构匹配
|
|
# 假设你的HTML文件在 "frontend/" 目录下
|
|
app.mount("/static", StaticFiles(directory="frontend/dist"), name="static")
|
|
|
|
@app.get("/")
|
|
def serve_frontend():
|
|
return FileResponse("frontend/dist/index.html") |