101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
"""Markdown formatter -- wraps AI-generated body with frontmatter and annotations."""
|
||
|
||
from datetime import datetime
|
||
|
||
|
||
def format_daily_digest(date, ai_body, tags=None, doc_type="daily-digest"):
|
||
"""Wrap AI-generated Markdown body with frontmatter, word count and reading time."""
|
||
date_str = date.strftime("%Y-%m-%d")
|
||
weekday = ["周一", "周二", "周三", "周四", "周五", "周六", "周日"][date.weekday()]
|
||
|
||
char_count = len(ai_body.replace(" ", "").replace("\n", ""))
|
||
read_min = max(1, round(char_count / 300))
|
||
|
||
if tags is None:
|
||
tags = ["灵感收集器", "每日总结"]
|
||
tag_str = ", ".join(["'" + t + "'" for t in tags])
|
||
|
||
lines = []
|
||
lines.append("---")
|
||
lines.append("date: " + date_str)
|
||
lines.append("type: " + doc_type)
|
||
lines.append("tags: [" + tag_str + "]")
|
||
lines.append("---")
|
||
lines.append("")
|
||
lines.append("# " + date_str + " 灵感摘要 · " + weekday)
|
||
lines.append("")
|
||
lines.append("> 本文共 **" + str(char_count) + "** 字 · 预计阅读 **" + str(read_min) + "** 分钟")
|
||
lines.append("")
|
||
lines.append("## AI 分析（自动生成，请勿编辑）")
|
||
lines.append("")
|
||
lines.append(ai_body)
|
||
lines.append("")
|
||
lines.append("---")
|
||
lines.append("")
|
||
lines.append("## 我的批注")
|
||
lines.append("")
|
||
lines.append("> *在这里写下你的想法、质疑、补充*")
|
||
lines.append("")
|
||
|
||
return "\n".join(lines)
|
||
|
||
|
||
def format_weekly_trend(start_date, end_date, stats, highlights, insight):
|
||
"""Format weekly trend markdown file."""
|
||
week_start = start_date.strftime("%m/%d")
|
||
week_end = end_date.strftime("%m/%d")
|
||
week_number = start_date.isocalendar()[1]
|
||
|
||
lines = []
|
||
lines.append("---")
|
||
lines.append("date: " + str(start_date.year) + "-W" + str(week_number))
|
||
lines.append("type: weekly-trend")
|
||
lines.append("tags: [灵感收集器, 每周趋势]")
|
||
lines.append("---")
|
||
lines.append("")
|
||
lines.append("# 第 " + str(week_number) + " 周灵感趋势（" + week_start + " - " + week_end + "）")
|
||
lines.append("")
|
||
lines.append("## AI 分析（自动生成，请勿编辑）")
|
||
lines.append("")
|
||
lines.append("本周共记录 **" + str(stats.get('total', 0)) + "** 条灵感。")
|
||
lines.append("")
|
||
|
||
if stats.get("daily_counts"):
|
||
lines.append("### 每日活跃度")
|
||
lines.append("")
|
||
for day, count in stats["daily_counts"].items():
|
||
bar = "█" * count if count > 0 else "▏"
|
||
lines.append("- " + day + "：" + bar + " " + str(count) + "条")
|
||
lines.append("")
|
||
|
||
if stats.get("categories"):
|
||
lines.append("### 主题分布")
|
||
lines.append("")
|
||
total = stats.get("total", 1)
|
||
for cat, count in stats["categories"].items():
|
||
pct = count / total * 100 if total else 0
|
||
lines.append("- **" + cat + "**：" + str(count) + "条（" + str(int(pct)) + "%）")
|
||
lines.append("")
|
||
|
||
if highlights:
|
||
lines.append("### 本周亮点")
|
||
lines.append("")
|
||
for h in highlights:
|
||
lines.append("- " + h)
|
||
lines.append("")
|
||
|
||
if insight:
|
||
lines.append("### 值得关注的模式")
|
||
lines.append("")
|
||
lines.append(insight)
|
||
lines.append("")
|
||
|
||
lines.append("---")
|
||
lines.append("")
|
||
lines.append("## 我的批注")
|
||
lines.append("")
|
||
lines.append("> *有什么想补充的写在下面*")
|
||
lines.append("")
|
||
|
||
return "\n".join(lines)
|