ycg
569c1c8813
- 重构数据访问层：引入DAO模式，支持MySQL/SQLite双数据库 - 新增数据库架构：完整的股票数据、AI分析、自选股管理表结构 - 升级AI分析服务：集成豆包大模型，支持多维度分析 - 优化API路由：分离市场数据API，提供更清晰的接口设计 - 完善项目文档：添加数据库迁移指南、新功能指南等 - 清理冗余文件：删除旧的缓存文件和无用配置 - 新增调度器：支持定时任务和数据自动更新 - 改进前端模板：简化的股票展示页面 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
208 lines
8.5 KiB
Python
208 lines
8.5 KiB
Python
"""
|
|
股票数据访问对象
|
|
"""
|
|
from typing import Dict, List, Optional, Tuple
|
|
import json
|
|
from datetime import datetime, date
|
|
|
|
from .base_dao import BaseDAO
|
|
|
|
|
|
class StockDAO(BaseDAO):
|
|
"""股票数据访问对象"""
|
|
|
|
def get_stock_by_code(self, stock_code: str) -> Optional[Dict]:
|
|
"""根据股票代码获取股票信息"""
|
|
query = "SELECT * FROM stocks WHERE stock_code = %s"
|
|
return self._execute_single_query(query, (stock_code,))
|
|
|
|
def add_or_update_stock(self, stock_code: str, stock_name: str, market: str) -> int:
|
|
"""添加或更新股票信息"""
|
|
existing = self.get_stock_by_code(stock_code)
|
|
|
|
if existing:
|
|
# 更新现有股票
|
|
query = """
|
|
UPDATE stocks
|
|
SET stock_name = %s, market = %s, updated_at = CURRENT_TIMESTAMP
|
|
WHERE stock_code = %s
|
|
"""
|
|
self._execute_update(query, (stock_name, market, stock_code))
|
|
return existing['id']
|
|
else:
|
|
# 添加新股票
|
|
query = """
|
|
INSERT INTO stocks (stock_code, stock_name, market)
|
|
VALUES (%s, %s, %s)
|
|
"""
|
|
return self._execute_insert(query, (stock_code, stock_name, market))
|
|
|
|
def get_stock_data(self, stock_code: str, data_date: str = None) -> Optional[Dict]:
|
|
"""获取股票数据"""
|
|
if data_date is None:
|
|
data_date = self.get_today_date()
|
|
|
|
query = """
|
|
SELECT sd.*, s.stock_name
|
|
FROM stock_data sd
|
|
JOIN stocks s ON sd.stock_code = s.stock_code
|
|
WHERE sd.stock_code = %s AND sd.data_date = %s
|
|
"""
|
|
return self._execute_single_query(query, (stock_code, data_date))
|
|
|
|
def save_stock_data(self, stock_code: str, stock_info: Dict, data_date: str = None) -> bool:
|
|
"""保存股票数据"""
|
|
if data_date is None:
|
|
data_date = self.get_today_date()
|
|
|
|
try:
|
|
# 确保股票信息存在
|
|
self.add_or_update_stock(
|
|
stock_code,
|
|
stock_info.get('name', ''),
|
|
'SH' if stock_code.startswith('6') else 'SZ'
|
|
)
|
|
|
|
# 检查是否已存在当日数据
|
|
existing = self.get_stock_data(stock_code, data_date)
|
|
|
|
if existing:
|
|
# 更新现有数据
|
|
query = """
|
|
UPDATE stock_data SET
|
|
price = %s,
|
|
change_percent = %s,
|
|
market_value = %s,
|
|
pe_ratio = %s,
|
|
pb_ratio = %s,
|
|
ps_ratio = %s,
|
|
dividend_yield = %s,
|
|
roe = %s,
|
|
gross_profit_margin = %s,
|
|
net_profit_margin = %s,
|
|
debt_to_assets = %s,
|
|
revenue_yoy = %s,
|
|
net_profit_yoy = %s,
|
|
bps = %s,
|
|
ocfps = %s,
|
|
from_cache = %s,
|
|
updated_at = CURRENT_TIMESTAMP
|
|
WHERE stock_code = %s AND data_date = %s
|
|
"""
|
|
self._execute_update(query, (
|
|
self.parse_float(stock_info.get('price')),
|
|
self.parse_float(stock_info.get('change_percent')),
|
|
self.parse_float(stock_info.get('market_value')),
|
|
self.parse_float(stock_info.get('pe_ratio')),
|
|
self.parse_float(stock_info.get('pb_ratio')),
|
|
self.parse_float(stock_info.get('ps_ratio')),
|
|
self.parse_float(stock_info.get('dividend_yield')),
|
|
self.parse_float(stock_info.get('roe')),
|
|
self.parse_float(stock_info.get('gross_profit_margin')),
|
|
self.parse_float(stock_info.get('net_profit_margin')),
|
|
self.parse_float(stock_info.get('debt_to_assets')),
|
|
self.parse_float(stock_info.get('revenue_yoy')),
|
|
self.parse_float(stock_info.get('net_profit_yoy')),
|
|
self.parse_float(stock_info.get('bps')),
|
|
self.parse_float(stock_info.get('ocfps')),
|
|
bool(stock_info.get('from_cache', False)),
|
|
stock_code,
|
|
data_date
|
|
))
|
|
else:
|
|
# 插入新数据
|
|
query = """
|
|
INSERT INTO stock_data (
|
|
stock_code, data_date, price, change_percent, market_value,
|
|
pe_ratio, pb_ratio, ps_ratio, dividend_yield,
|
|
roe, gross_profit_margin, net_profit_margin, debt_to_assets,
|
|
revenue_yoy, net_profit_yoy, bps, ocfps, from_cache
|
|
) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
|
"""
|
|
self._execute_insert(query, (
|
|
stock_code, data_date,
|
|
self.parse_float(stock_info.get('price')),
|
|
self.parse_float(stock_info.get('change_percent')),
|
|
self.parse_float(stock_info.get('market_value')),
|
|
self.parse_float(stock_info.get('pe_ratio')),
|
|
self.parse_float(stock_info.get('pb_ratio')),
|
|
self.parse_float(stock_info.get('ps_ratio')),
|
|
self.parse_float(stock_info.get('dividend_yield')),
|
|
self.parse_float(stock_info.get('roe')),
|
|
self.parse_float(stock_info.get('gross_profit_margin')),
|
|
self.parse_float(stock_info.get('net_profit_margin')),
|
|
self.parse_float(stock_info.get('debt_to_assets')),
|
|
self.parse_float(stock_info.get('revenue_yoy')),
|
|
self.parse_float(stock_info.get('net_profit_yoy')),
|
|
self.parse_float(stock_info.get('bps')),
|
|
self.parse_float(stock_info.get('ocfps')),
|
|
bool(stock_info.get('from_cache', False))
|
|
))
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
self.logger.error(f"保存股票数据失败: {stock_code}, 错误: {e}")
|
|
self.log_data_update('stock_data', stock_code, 'failed', str(e))
|
|
return False
|
|
|
|
def get_latest_stock_data(self, stock_code: str) -> Optional[Dict]:
|
|
"""获取最新的股票数据"""
|
|
query = """
|
|
SELECT sd.*, s.stock_name
|
|
FROM stock_data sd
|
|
JOIN stocks s ON sd.stock_code = s.stock_code
|
|
WHERE sd.stock_code = %s
|
|
ORDER BY sd.data_date DESC
|
|
LIMIT 1
|
|
"""
|
|
return self._execute_single_query(query, (stock_code,))
|
|
|
|
def get_multiple_stocks_data(self, stock_codes: List[str], data_date: str = None) -> List[Dict]:
|
|
"""批量获取股票数据"""
|
|
if not stock_codes:
|
|
return []
|
|
|
|
if data_date is None:
|
|
data_date = self.get_today_date()
|
|
|
|
placeholders = ','.join(['%s'] * len(stock_codes))
|
|
query = f"""
|
|
SELECT sd.*, s.stock_name
|
|
FROM stock_data sd
|
|
JOIN stocks s ON sd.stock_code = s.stock_code
|
|
WHERE sd.stock_code IN ({placeholders}) AND sd.data_date = %s
|
|
"""
|
|
|
|
return self._execute_query(query, tuple(stock_codes + [data_date]))
|
|
|
|
def get_stock_data_history(self, stock_code: str, days: int = 30) -> List[Dict]:
|
|
"""获取股票历史数据"""
|
|
query = """
|
|
SELECT sd.*, s.stock_name
|
|
FROM stock_data sd
|
|
JOIN stocks s ON sd.stock_code = s.stock_code
|
|
WHERE sd.stock_code = %s AND sd.data_date >= DATE_SUB(CURDATE(), INTERVAL %s DAY)
|
|
ORDER BY sd.data_date DESC
|
|
"""
|
|
return self._execute_query(query, (stock_code, days))
|
|
|
|
def delete_stock_data(self, stock_code: str, before_date: str = None) -> int:
|
|
"""删除股票数据"""
|
|
if before_date:
|
|
query = "DELETE FROM stock_data WHERE stock_code = %s AND data_date < %s"
|
|
return self._execute_update(query, (stock_code, before_date))
|
|
else:
|
|
query = "DELETE FROM stock_data WHERE stock_code = %s"
|
|
return self._execute_update(query, (stock_code,))
|
|
|
|
def get_stock_count(self) -> int:
|
|
"""获取股票总数"""
|
|
query = "SELECT COUNT(*) as count FROM stocks"
|
|
result = self._execute_single_query(query)
|
|
return result['count'] if result else 0
|
|
|
|
def get_data_date_range(self) -> Optional[Dict]:
|
|
"""获取数据的日期范围"""
|
|
query = "SELECT MIN(data_date) as min_date, MAX(data_date) as max_date FROM stock_data"
|
|
return self._execute_single_query(query) |