91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
技术分析结果存储模块
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from collections import defaultdict
|
|||
|
|
from typing import Dict, List, Optional
|
|||
|
|
import threading
|
|||
|
|
|
|||
|
|
from ..models import TechTrendState, TechTrendChange
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TechStore:
|
|||
|
|
"""技术分析结果存储（只负责数据CRUD）"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
# 趋势状态: {SYMBOL: {PERIOD: TechTrendState}}
|
|||
|
|
self._trend_states: Dict[str, Dict[str, TechTrendState]] = defaultdict(lambda: defaultdict(dict))
|
|||
|
|
self._lock = threading.RLock()
|
|||
|
|
|
|||
|
|
# 趋势转换历史: {SYMBOL: [TechTrendChange, ...]}
|
|||
|
|
self._trend_changes: Dict[str, List[TechTrendChange]] = defaultdict(list)
|
|||
|
|
|
|||
|
|
# 最大历史记录数
|
|||
|
|
self.MAX_CHANGES = 20
|
|||
|
|
|
|||
|
|
print("[TechStore] 技术分析存储已初始化")
|
|||
|
|
|
|||
|
|
# ==================== 趋势状态 ====================
|
|||
|
|
|
|||
|
|
def save_trend_state(self, state: TechTrendState):
|
|||
|
|
"""保存趋势状态"""
|
|||
|
|
with self._lock:
|
|||
|
|
self._trend_states[state.symbol][state.period] = state
|
|||
|
|
|
|||
|
|
def get_trend_state(self, symbol: str, period: str = None) -> Dict:
|
|||
|
|
"""获取趋势状态"""
|
|||
|
|
with self._lock:
|
|||
|
|
if period:
|
|||
|
|
state = self._trend_states[symbol].get(period)
|
|||
|
|
return state.to_dict() if state else {}
|
|||
|
|
return {p: s.to_dict() for p, s in self._trend_states[symbol].items()}
|
|||
|
|
|
|||
|
|
def get_trend_state_object(self, symbol: str, period: str) -> Optional[TechTrendState]:
|
|||
|
|
"""获取趋势状态对象"""
|
|||
|
|
with self._lock:
|
|||
|
|
return self._trend_states[symbol].get(period)
|
|||
|
|
|
|||
|
|
def get_all_trend_states(self, symbol: str) -> Dict[str, TechTrendState]:
|
|||
|
|
"""获取某品种所有周期的趋势状态"""
|
|||
|
|
with self._lock:
|
|||
|
|
return dict(self._trend_states[symbol])
|
|||
|
|
|
|||
|
|
# ==================== 趋势转换历史 ====================
|
|||
|
|
|
|||
|
|
def add_trend_change(self, symbol: str, change: TechTrendChange):
|
|||
|
|
"""添加趋势转换记录"""
|
|||
|
|
with self._lock:
|
|||
|
|
self._trend_changes[symbol].append(change)
|
|||
|
|
# 限制数量
|
|||
|
|
if len(self._trend_changes[symbol]) > self.MAX_CHANGES:
|
|||
|
|
self._trend_changes[symbol] = self._trend_changes[symbol][-self.MAX_CHANGES:]
|
|||
|
|
|
|||
|
|
def get_trend_changes(self, symbol: str, count: int = 10) -> List[Dict]:
|
|||
|
|
"""获取趋势转换历史"""
|
|||
|
|
with self._lock:
|
|||
|
|
changes = self._trend_changes[symbol][-count:]
|
|||
|
|
return [c.to_dict() for c in changes]
|
|||
|
|
|
|||
|
|
# ==================== 清理 ====================
|
|||
|
|
|
|||
|
|
def clear_symbol(self, symbol: str):
|
|||
|
|
"""清除某品种的数据"""
|
|||
|
|
with self._lock:
|
|||
|
|
if symbol in self._trend_states:
|
|||
|
|
del self._trend_states[symbol]
|
|||
|
|
if symbol in self._trend_changes:
|
|||
|
|
del self._trend_changes[symbol]
|
|||
|
|
|
|||
|
|
# ==================== 状态 ====================
|
|||
|
|
|
|||
|
|
def get_status(self) -> Dict:
|
|||
|
|
"""获取状态"""
|
|||
|
|
with self._lock:
|
|||
|
|
symbols = list(self._trend_states.keys())
|
|||
|
|
total_states = sum(len(periods) for periods in self._trend_states.values())
|
|||
|
|
return {
|
|||
|
|
"symbols": symbols,
|
|||
|
|
"total_states": total_states
|
|||
|
|
}
|