mirror of
https://github.com/ZhuLinsen/daily_stock_analysis
synced 2026-09-20 10:53:33 +08:00
144 lines
4.5 KiB
Python
144 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
===================================
|
|
WebUI 入口文件 (向后兼容)
|
|
===================================
|
|
|
|
本文件保持向后兼容，实际实现已迁移到 web/ 包
|
|
|
|
结构说明:
|
|
web/
|
|
├── __init__.py - 包初始化
|
|
├── server.py - HTTP 服务器
|
|
├── router.py - 路由分发
|
|
├── handlers.py - 请求处理器
|
|
├── services.py - 业务服务层
|
|
└── templates.py - HTML 模板
|
|
|
|
API Endpoints:
|
|
GET / - 配置页面
|
|
GET /health - 健康检查
|
|
GET /analysis?code=xxx - 触发单只股票异步分析
|
|
GET /tasks - 查询任务列表
|
|
GET /task?id=xxx - 查询任务状态
|
|
POST /update - 更新配置
|
|
|
|
Usage:
|
|
python webui.py
|
|
WEBUI_HOST=0.0.0.0 WEBUI_PORT=8000 python webui.py
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import logging
|
|
|
|
# 从 web 包导入（新架构）
|
|
from web.server import WebServer, run_server_in_thread, run_server
|
|
from web.router import Router, get_router
|
|
from web.services import ConfigService, AnalysisService, get_config_service, get_analysis_service
|
|
from web.handlers import PageHandler, ApiHandler
|
|
from web.templates import render_config_page, render_error_page
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# 导出所有公共接口（保持向后兼容）
|
|
__all__ = [
|
|
# 服务器
|
|
'WebServer',
|
|
'run_server_in_thread',
|
|
'run_server',
|
|
# 路由
|
|
'Router',
|
|
'get_router',
|
|
# 服务
|
|
'ConfigService',
|
|
'AnalysisService',
|
|
'get_config_service',
|
|
'get_analysis_service',
|
|
# 处理器
|
|
'PageHandler',
|
|
'ApiHandler',
|
|
# 模板
|
|
'render_config_page',
|
|
'render_error_page',
|
|
]
|
|
|
|
|
|
def _start_bot_stream_clients() -> None:
|
|
"""启动 Bot Stream 模式客户端（如果已配置）"""
|
|
from config import get_config
|
|
config = get_config()
|
|
|
|
# 钉钉 Stream 模式
|
|
if config.dingtalk_stream_enabled:
|
|
try:
|
|
from bot.platforms import start_dingtalk_stream_background, DINGTALK_STREAM_AVAILABLE
|
|
if DINGTALK_STREAM_AVAILABLE:
|
|
if start_dingtalk_stream_background():
|
|
logger.info("[WebUI] 钉钉 Stream 客户端已在后台启动")
|
|
else:
|
|
logger.warning("[WebUI] 钉钉 Stream 客户端启动失败")
|
|
else:
|
|
logger.warning("[WebUI] 钉钉 Stream 模式已启用但 SDK 未安装")
|
|
logger.warning("[WebUI] 请运行: pip install dingtalk-stream")
|
|
except Exception as e:
|
|
logger.error(f"[WebUI] 启动钉钉 Stream 客户端失败: {e}")
|
|
|
|
# 飞书 Stream 模式
|
|
if getattr(config, 'feishu_stream_enabled', False):
|
|
try:
|
|
from bot.platforms import start_feishu_stream_background, FEISHU_SDK_AVAILABLE
|
|
if FEISHU_SDK_AVAILABLE:
|
|
if start_feishu_stream_background():
|
|
logger.info("[WebUI] 飞书 Stream 客户端已在后台启动")
|
|
else:
|
|
logger.warning("[WebUI] 飞书 Stream 客户端启动失败")
|
|
else:
|
|
logger.warning("[WebUI] 飞书 Stream 模式已启用但 SDK 未安装")
|
|
logger.warning("[WebUI] 请运行: pip install lark-oapi")
|
|
except Exception as e:
|
|
logger.error(f"[WebUI] 启动飞书 Stream 客户端失败: {e}")
|
|
|
|
|
|
def main() -> int:
|
|
"""
|
|
主入口函数
|
|
|
|
支持环境变量配置:
|
|
WEBUI_HOST: 监听地址 (默认 127.0.0.1)
|
|
WEBUI_PORT: 监听端口 (默认 8000)
|
|
"""
|
|
host = os.getenv("WEBUI_HOST", "127.0.0.1")
|
|
port = int(os.getenv("WEBUI_PORT", "8000"))
|
|
|
|
print(f"WebUI running: http://{host}:{port}")
|
|
print("API Endpoints:")
|
|
print(" GET / - 配置页面")
|
|
print(" GET /health - 健康检查")
|
|
print(" GET /analysis?code=xxx - 触发分析")
|
|
print(" GET /tasks - 任务列表")
|
|
print(" GET /task?id=xxx - 任务状态")
|
|
print(" POST /update - 更新配置")
|
|
print()
|
|
print("Bot Webhooks:")
|
|
print(" POST /bot/feishu - 飞书机器人")
|
|
print(" POST /bot/dingtalk - 钉钉机器人")
|
|
print(" POST /bot/wecom - 企业微信机器人")
|
|
print(" POST /bot/telegram - Telegram 机器人")
|
|
print()
|
|
|
|
# 启动 Bot Stream 客户端（如果配置了）
|
|
_start_bot_stream_clients()
|
|
|
|
try:
|
|
run_server(host=host, port=port)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|