You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

135 lines
3.4 KiB

# -*- coding: utf-8 -*-
"""
===================================
A股自选股智能分析系统 - 分析服务层
===================================
职责
1. 封装核心分析逻辑支持多调用方CLIWebUIBot
2. 提供清晰的API接口不依赖于命令行参数
3. 支持依赖注入便于测试和扩展
4. 统一管理分析流程和配置
"""
import uuid
from typing import List, Optional
from src.analyzer import AnalysisResult
from src.config import get_config, Config
from src.notification import NotificationService
from src.enums import ReportType
from src.core.pipeline import StockAnalysisPipeline
from src.core.market_review import run_market_review
def analyze_stock(
stock_code: str,
config: Config = None,
full_report: bool = False,
notifier: Optional[NotificationService] = None
) -> Optional[AnalysisResult]:
"""
分析单只股票
Args:
stock_code: 股票代码
config: 配置对象可选默认使用单例
full_report: 是否生成完整报告
notifier: 通知服务可选
Returns:
分析结果对象
"""
if config is None:
config = get_config()
# 创建分析流水线
pipeline = StockAnalysisPipeline(
config=config,
query_id=uuid.uuid4().hex,
query_source="cli"
)
# 使用通知服务(如果提供)
if notifier:
pipeline.notifier = notifier
# 根据full_report参数设置报告类型
report_type = ReportType.FULL if full_report else ReportType.SIMPLE
# 运行单只股票分析
result = pipeline.process_single_stock(
code=stock_code,
skip_analysis=False,
single_stock_notify=notifier is not None,
report_type=report_type
)
return result
def analyze_stocks(
stock_codes: List[str],
config: Config = None,
full_report: bool = False,
notifier: Optional[NotificationService] = None
) -> List[AnalysisResult]:
"""
分析多只股票
Args:
stock_codes: 股票代码列表
config: 配置对象可选默认使用单例
full_report: 是否生成完整报告
notifier: 通知服务可选
Returns:
分析结果列表
"""
if config is None:
config = get_config()
results = []
for stock_code in stock_codes:
result = analyze_stock(stock_code, config, full_report, notifier)
if result:
results.append(result)
return results
def perform_market_review(
config: Config = None,
notifier: Optional[NotificationService] = None
) -> Optional[str]:
"""
执行大盘复盘
Args:
config: 配置对象可选默认使用单例
notifier: 通知服务可选
Returns:
复盘报告内容
"""
if config is None:
config = get_config()
# 创建分析流水线以获取analyzer和search_service
pipeline = StockAnalysisPipeline(
config=config,
query_id=uuid.uuid4().hex,
query_source="cli"
)
# 使用提供的通知服务或创建新的
review_notifier = notifier or pipeline.notifier
# 调用大盘复盘函数
return run_market_review(
notifier=review_notifier,
analyzer=pipeline.analyzer,
search_service=pipeline.search_service
)