1
This commit is contained in:
parent
2601a4db17
commit
9d8e223df8
Binary file not shown.
@ -53,9 +53,16 @@ class Settings(BaseSettings):
|
||||
|
||||
# 新闻/政策催化采集
|
||||
news_collection_enabled: bool = True
|
||||
news_tushare_sources: str = "sina,eastmoney,10jqka,wallstreetcn"
|
||||
news_tushare_enabled: bool = False
|
||||
news_tushare_sources: str = ""
|
||||
news_tushare_sources_per_run: int = 1
|
||||
news_tushare_daily_quota: int = 0
|
||||
news_tushare_alert_errors: bool = False
|
||||
news_rss_sources: str = "" # name|url,name|url
|
||||
news_web_sources: str = "eastmoney_roll|https://roll.eastmoney.com/"
|
||||
news_akshare_enabled: bool = True
|
||||
news_akshare_stock_limit: int = 20
|
||||
news_akshare_news_per_stock: int = 3
|
||||
news_fetch_lookback_hours: int = 24
|
||||
news_fetch_limit_per_source: int = 30
|
||||
news_analyze_limit_per_run: int = 50
|
||||
|
||||
Binary file not shown.
@ -14,6 +14,7 @@ from app.data.cache import cache
|
||||
from app.db.error_logger import log_error_background
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
_NEWS_QUOTA_ERROR_KEYWORDS = ("频率超限", "每分钟", "每小时", "每天", "权限")
|
||||
|
||||
|
||||
class TushareClient:
|
||||
@ -300,8 +301,14 @@ class TushareClient:
|
||||
return result
|
||||
except Exception as e:
|
||||
logger.warning("Tushare 新闻请求失败 source=%s: %s", source, e)
|
||||
log_error_background("tushare_news", f"Tushare 新闻请求失败 source={source}: {e}")
|
||||
if settings.news_tushare_alert_errors and not _is_news_quota_error(e):
|
||||
log_error_background("tushare_news", f"Tushare 新闻请求失败 source={source}: {e}")
|
||||
return pd.DataFrame()
|
||||
|
||||
|
||||
tushare_client = TushareClient()
|
||||
|
||||
|
||||
def _is_news_quota_error(exc: Exception) -> bool:
|
||||
message = str(exc)
|
||||
return any(keyword in message for keyword in _NEWS_QUOTA_ERROR_KEYWORDS)
|
||||
|
||||
Binary file not shown.
Binary file not shown.
@ -7,20 +7,29 @@ import html
|
||||
import logging
|
||||
import re
|
||||
import xml.etree.ElementTree as ET
|
||||
from datetime import datetime, timedelta
|
||||
from datetime import date, datetime, timedelta
|
||||
from importlib import import_module
|
||||
|
||||
import httpx
|
||||
import pandas as pd
|
||||
from sqlalchemy import text
|
||||
|
||||
from app.config import settings
|
||||
from app.data.tushare_client import tushare_client
|
||||
from app.db.database import get_db
|
||||
from app.news.models import NewsItem
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
_tushare_source_cursor = 0
|
||||
_tushare_calls_by_day: dict[str, int] = {}
|
||||
|
||||
RSS_HEADERS = {
|
||||
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36",
|
||||
}
|
||||
WEB_HEADERS = {
|
||||
"Referer": "https://www.eastmoney.com/",
|
||||
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36",
|
||||
}
|
||||
|
||||
|
||||
async def collect_news_sources(
|
||||
@ -38,6 +47,15 @@ async def collect_news_sources(
|
||||
except Exception as e:
|
||||
logger.warning("Tushare 新闻源采集失败 source=%s error=%s", source, e)
|
||||
|
||||
web_sources = _parse_named_url_sources(settings.news_web_sources)
|
||||
if web_sources:
|
||||
async with httpx.AsyncClient(headers=WEB_HEADERS, timeout=10.0, follow_redirects=True) as client:
|
||||
for name, url in web_sources:
|
||||
try:
|
||||
items.extend(await _collect_web_page(client, name, url, lookback_hours, limit_per_source))
|
||||
except Exception as e:
|
||||
logger.warning("网页新闻源采集失败 source=%s url=%s error=%s", name, url, e)
|
||||
|
||||
rss_sources = _parse_rss_sources(settings.news_rss_sources)
|
||||
if rss_sources:
|
||||
async with httpx.AsyncClient(headers=RSS_HEADERS, timeout=10.0, follow_redirects=True) as client:
|
||||
@ -47,6 +65,12 @@ async def collect_news_sources(
|
||||
except Exception as e:
|
||||
logger.warning("RSS 新闻源采集失败 source=%s url=%s error=%s", name, url, e)
|
||||
|
||||
if settings.news_akshare_enabled:
|
||||
try:
|
||||
items.extend(await _collect_akshare_stock_news(lookback_hours, limit_per_source))
|
||||
except Exception as e:
|
||||
logger.warning("AKShare 个股新闻采集失败: %s", e)
|
||||
|
||||
return _dedup_in_memory(items)
|
||||
|
||||
|
||||
@ -77,6 +101,23 @@ async def _collect_tushare_news(source: str, lookback_hours: int, limit: int) ->
|
||||
return items[:limit]
|
||||
|
||||
|
||||
async def _collect_web_page(
|
||||
client: httpx.AsyncClient,
|
||||
source: str,
|
||||
url: str,
|
||||
lookback_hours: int,
|
||||
limit: int,
|
||||
) -> list[NewsItem]:
|
||||
resp = await client.get(url)
|
||||
resp.raise_for_status()
|
||||
text = resp.text
|
||||
|
||||
if source == "eastmoney_roll":
|
||||
return _parse_eastmoney_roll(text, limit)
|
||||
|
||||
return _parse_generic_links(text, source=source, limit=limit)
|
||||
|
||||
|
||||
async def _collect_rss(
|
||||
client: httpx.AsyncClient,
|
||||
source: str,
|
||||
@ -112,11 +153,124 @@ async def _collect_rss(
|
||||
return items
|
||||
|
||||
|
||||
async def _collect_akshare_stock_news(lookback_hours: int, limit_per_source: int) -> list[NewsItem]:
|
||||
"""采集推荐池/关注池中标的的东方财富个股新闻。
|
||||
|
||||
AKShare 是可选增强依赖:安装时启用,未安装时安静跳过,避免影响主服务。
|
||||
"""
|
||||
ak = _load_akshare()
|
||||
if ak is None:
|
||||
logger.info("未安装 AKShare,跳过个股新闻补充源")
|
||||
return []
|
||||
|
||||
symbols = await _load_focus_stock_symbols(limit=settings.news_akshare_stock_limit)
|
||||
if not symbols:
|
||||
return []
|
||||
|
||||
items: list[NewsItem] = []
|
||||
per_stock = max(1, min(settings.news_akshare_news_per_stock, limit_per_source))
|
||||
cutoff = datetime.now() - timedelta(hours=lookback_hours)
|
||||
for ts_code, name in symbols:
|
||||
try:
|
||||
df = await _call_akshare_stock_news(ak, _strip_market(ts_code))
|
||||
except Exception as e:
|
||||
logger.debug("AKShare 个股新闻失败 ts_code=%s error=%s", ts_code, e)
|
||||
continue
|
||||
items.extend(_parse_akshare_stock_news(df, ts_code=ts_code, name=name, cutoff=cutoff, limit=per_stock))
|
||||
return items
|
||||
|
||||
|
||||
def _load_akshare():
|
||||
try:
|
||||
return import_module("akshare")
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
|
||||
async def _call_akshare_stock_news(ak, symbol: str) -> pd.DataFrame:
|
||||
import asyncio
|
||||
|
||||
return await asyncio.to_thread(ak.stock_news_em, symbol=symbol)
|
||||
|
||||
|
||||
async def _load_focus_stock_symbols(limit: int) -> list[tuple[str, str]]:
|
||||
async with get_db() as db:
|
||||
result = await db.execute(
|
||||
text(
|
||||
"SELECT ts_code, name FROM recommendations "
|
||||
"WHERE action_plan IN ('可操作', '重点关注', '观察') "
|
||||
"ORDER BY created_at DESC, score DESC LIMIT :limit"
|
||||
),
|
||||
{"limit": limit},
|
||||
)
|
||||
rows = result.fetchall()
|
||||
|
||||
seen: set[str] = set()
|
||||
symbols: list[tuple[str, str]] = []
|
||||
for row in rows:
|
||||
ts_code = str(row._mapping["ts_code"] or "")
|
||||
if not ts_code or ts_code in seen:
|
||||
continue
|
||||
seen.add(ts_code)
|
||||
symbols.append((ts_code, str(row._mapping["name"] or ts_code)))
|
||||
return symbols
|
||||
|
||||
|
||||
def _parse_akshare_stock_news(
|
||||
df: pd.DataFrame,
|
||||
ts_code: str,
|
||||
name: str,
|
||||
cutoff: datetime,
|
||||
limit: int,
|
||||
) -> list[NewsItem]:
|
||||
if df is None or df.empty:
|
||||
return []
|
||||
|
||||
items: list[NewsItem] = []
|
||||
for _, row in df.head(max(limit * 3, limit)).iterrows():
|
||||
title = _clean_text(_pick_row_value(row, "新闻标题", "title", "标题"))
|
||||
if not _is_useful_title(title):
|
||||
continue
|
||||
published_at = _parse_datetime(_pick_row_value(row, "发布时间", "datetime", "time", "日期"))
|
||||
if published_at and published_at < cutoff:
|
||||
continue
|
||||
content = _clean_text(_pick_row_value(row, "新闻内容", "content", "内容", "摘要"))
|
||||
source = _clean_text(_pick_row_value(row, "文章来源", "source", "来源")) or "东方财富"
|
||||
url = str(_pick_row_value(row, "新闻链接", "url", "链接") or "")
|
||||
summary = content[:240] if content else title
|
||||
items.append(NewsItem(
|
||||
title=f"{name}: {title}" if name and name not in title else title,
|
||||
content=content or title,
|
||||
summary=summary,
|
||||
source=f"akshare:{source}",
|
||||
url=url,
|
||||
published_at=published_at,
|
||||
))
|
||||
if len(items) >= limit:
|
||||
break
|
||||
return items
|
||||
|
||||
|
||||
def _pick_row_value(row, *keys: str):
|
||||
for key in keys:
|
||||
try:
|
||||
value = row.get(key)
|
||||
except Exception:
|
||||
value = None
|
||||
if value is not None and str(value).strip() and str(value).lower() != "nan":
|
||||
return value
|
||||
return ""
|
||||
|
||||
|
||||
def _split_csv(value: str) -> list[str]:
|
||||
return [item.strip() for item in (value or "").split(",") if item.strip()]
|
||||
|
||||
|
||||
def _parse_rss_sources(value: str) -> list[tuple[str, str]]:
|
||||
return _parse_named_url_sources(value)
|
||||
|
||||
|
||||
def _parse_named_url_sources(value: str) -> list[tuple[str, str]]:
|
||||
result: list[tuple[str, str]] = []
|
||||
for chunk in _split_csv(value):
|
||||
if "|" not in chunk:
|
||||
@ -130,21 +284,100 @@ def _parse_rss_sources(value: str) -> list[tuple[str, str]]:
|
||||
|
||||
|
||||
def _select_tushare_sources_for_run() -> list[str]:
|
||||
"""Tushare news 免费/低权限账号通常限制 1 次/分钟,每轮只取少量源。"""
|
||||
"""Tushare news 默认关闭,仅在显式配置时少量使用。"""
|
||||
global _tushare_source_cursor
|
||||
|
||||
if not settings.news_tushare_enabled:
|
||||
return []
|
||||
|
||||
sources = _split_csv(settings.news_tushare_sources)
|
||||
if not sources:
|
||||
return []
|
||||
|
||||
limit = max(1, min(int(settings.news_tushare_sources_per_run or 1), len(sources)))
|
||||
remaining_quota = _remaining_tushare_quota()
|
||||
if remaining_quota <= 0:
|
||||
logger.info("Tushare 新闻日额度已用尽,跳过本轮采集")
|
||||
return []
|
||||
|
||||
limit = max(1, min(int(settings.news_tushare_sources_per_run or 1), len(sources), remaining_quota))
|
||||
selected: list[str] = []
|
||||
for offset in range(limit):
|
||||
selected.append(sources[(_tushare_source_cursor + offset) % len(sources)])
|
||||
_tushare_source_cursor = (_tushare_source_cursor + limit) % len(sources)
|
||||
_consume_tushare_quota(len(selected))
|
||||
return selected
|
||||
|
||||
|
||||
def _remaining_tushare_quota() -> int:
|
||||
quota = int(settings.news_tushare_daily_quota or 0)
|
||||
if quota <= 0:
|
||||
return 0
|
||||
today = date.today().isoformat()
|
||||
return max(0, quota - int(_tushare_calls_by_day.get(today, 0)))
|
||||
|
||||
|
||||
def _consume_tushare_quota(count: int) -> None:
|
||||
if count <= 0:
|
||||
return
|
||||
today = date.today().isoformat()
|
||||
_tushare_calls_by_day[today] = int(_tushare_calls_by_day.get(today, 0)) + count
|
||||
|
||||
|
||||
def _parse_eastmoney_roll(text: str, limit: int) -> list[NewsItem]:
|
||||
result: list[NewsItem] = []
|
||||
pattern = re.compile(
|
||||
r'<a[^>]+href=["\'](?P<url>https?://(?:finance|stock|kuaixun)\.eastmoney\.com/[^"\']+)["\'][^>]*>(?P<title>.*?)</a>',
|
||||
re.I | re.S,
|
||||
)
|
||||
for match in pattern.finditer(text or ""):
|
||||
title = _clean_text(match.group("title"))
|
||||
url = _clean_text(match.group("url"))
|
||||
if not _is_useful_title(title):
|
||||
continue
|
||||
result.append(NewsItem(
|
||||
title=title,
|
||||
content=title,
|
||||
summary=title,
|
||||
source="web:eastmoney_roll",
|
||||
url=url,
|
||||
published_at=None,
|
||||
))
|
||||
if len(result) >= limit:
|
||||
break
|
||||
return result
|
||||
|
||||
|
||||
def _parse_generic_links(text: str, source: str, limit: int) -> list[NewsItem]:
|
||||
result: list[NewsItem] = []
|
||||
pattern = re.compile(r'<a[^>]+href=["\'](?P<url>https?://[^"\']+)["\'][^>]*>(?P<title>.*?)</a>', re.I | re.S)
|
||||
for match in pattern.finditer(text or ""):
|
||||
title = _clean_text(match.group("title"))
|
||||
if not _is_useful_title(title):
|
||||
continue
|
||||
result.append(NewsItem(
|
||||
title=title,
|
||||
content=title,
|
||||
summary=title,
|
||||
source=f"web:{source}",
|
||||
url=_clean_text(match.group("url")),
|
||||
published_at=None,
|
||||
))
|
||||
if len(result) >= limit:
|
||||
break
|
||||
return result
|
||||
|
||||
|
||||
def _is_useful_title(title: str) -> bool:
|
||||
if len(title) < settings.news_min_title_length:
|
||||
return False
|
||||
return not any(token in title for token in ("广告", "下载APP", "扫一扫", "关于我们", "联系我们"))
|
||||
|
||||
|
||||
def _strip_market(ts_code: str) -> str:
|
||||
text = str(ts_code or "").strip()
|
||||
return text.split(".", 1)[0] if "." in text else text
|
||||
|
||||
|
||||
def _xml_text(item: ET.Element, tag: str) -> str:
|
||||
node = item.find(tag)
|
||||
return node.text if node is not None and node.text else ""
|
||||
|
||||
@ -6,6 +6,7 @@ sqlalchemy==2.0.36
|
||||
aiosqlite==0.20.0
|
||||
greenlet==3.3.2
|
||||
tushare==1.4.20
|
||||
akshare==1.18.40
|
||||
pandas==2.2.3
|
||||
numpy==2.2.1
|
||||
apscheduler==3.10.4
|
||||
|
||||
Loading…
Reference in New Issue
Block a user