98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
|
"""
|
|||
|
|
新闻总结生成模块
|
|||
|
|
使用规则引擎生成中文总结（不调用外部 AI API）
|
|||
|
|
"""
|
|||
|
|
import logging
|
|||
|
|
from datetime import datetime, timedelta
|
|||
|
|
from collections import defaultdict
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_summary(news_list: list, period: str = "全天") -> str:
|
|||
|
|
"""
|
|||
|
|
生成新闻总结
|
|||
|
|
period: "最近1小时" / "上午" / "全天" / "昨晚到今早"
|
|||
|
|
"""
|
|||
|
|
if not news_list:
|
|||
|
|
return f"📭 {period}暂无重要新闻"
|
|||
|
|
|
|||
|
|
# 按评分排序，高分在前
|
|||
|
|
sorted_news = sorted(news_list, key=lambda x: x.get("score", 0), reverse=True)
|
|||
|
|
|
|||
|
|
# 按来源分组统计
|
|||
|
|
source_count = defaultdict(int)
|
|||
|
|
for n in sorted_news:
|
|||
|
|
source_count[n.get("source_name", "未知")] += 1
|
|||
|
|
|
|||
|
|
# 构建总结
|
|||
|
|
lines = []
|
|||
|
|
lines.append(f"📊 *{period}新闻总结*")
|
|||
|
|
lines.append(f"━━━━━━━━━━━━━━━")
|
|||
|
|
lines.append(f"📰 共 {len(sorted_news)} 条快讯")
|
|||
|
|
lines.append("")
|
|||
|
|
|
|||
|
|
# 🔴 重磅新闻（评分 >= 8）
|
|||
|
|
hot_news = [n for n in sorted_news if n.get("score", 0) >= 8]
|
|||
|
|
if hot_news:
|
|||
|
|
lines.append("🔴 *重磅快讯*")
|
|||
|
|
for n in hot_news[:5]:
|
|||
|
|
emoji = _score_emoji(n.get("score", 0))
|
|||
|
|
lines.append(f" {emoji} {n['title']}")
|
|||
|
|
lines.append("")
|
|||
|
|
|
|||
|
|
# 🟡 值得关注（评分 6-7）
|
|||
|
|
mid_news = [n for n in sorted_news if 6 <= n.get("score", 0) <= 7]
|
|||
|
|
if mid_news:
|
|||
|
|
lines.append("🟡 *值得关注*")
|
|||
|
|
for n in mid_news[:8]:
|
|||
|
|
lines.append(f" • {n['title']}")
|
|||
|
|
lines.append("")
|
|||
|
|
|
|||
|
|
# 来源统计
|
|||
|
|
lines.append("📡 *来源分布*")
|
|||
|
|
for src, cnt in sorted(source_count.items(), key=lambda x: x[1], reverse=True):
|
|||
|
|
lines.append(f" {src}: {cnt}条")
|
|||
|
|
|
|||
|
|
lines.append("")
|
|||
|
|
lines.append(f"⏰ 生成时间: {datetime.now().strftime('%H:%M')}")
|
|||
|
|
|
|||
|
|
return "\n".join(lines)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _score_emoji(score: int) -> str:
|
|||
|
|
"""根据评分返回 emoji"""
|
|||
|
|
if score >= 9:
|
|||
|
|
return "🔥"
|
|||
|
|
elif score >= 8:
|
|||
|
|
return "⚡"
|
|||
|
|
elif score >= 7:
|
|||
|
|
return "📌"
|
|||
|
|
elif score >= 6:
|
|||
|
|
return "📎"
|
|||
|
|
else:
|
|||
|
|
return "•"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_period_range(period: str) -> tuple:
|
|||
|
|
"""
|
|||
|
|
根据时间段名称返回 (start_ts, end_ts)
|
|||
|
|
"""
|
|||
|
|
now = datetime.now()
|
|||
|
|
today_start = now.replace(hour=0, minute=0, second=0, microsecond=0)
|
|||
|
|
|
|||
|
|
if period == "最近1小时":
|
|||
|
|
start = now - timedelta(hours=1)
|
|||
|
|
elif period == "上午":
|
|||
|
|
start = today_start.replace(hour=6)
|
|||
|
|
now = min(now, today_start.replace(hour=12))
|
|||
|
|
elif period == "全天":
|
|||
|
|
start = today_start
|
|||
|
|
elif period == "昨晚到今早":
|
|||
|
|
start = (today_start - timedelta(days=1)).replace(hour=20)
|
|||
|
|
now = today_start.replace(hour=8)
|
|||
|
|
else:
|
|||
|
|
start = now - timedelta(hours=3)
|
|||
|
|
|
|||
|
|
return int(start.timestamp()), int(now.timestamp())
|