1#!/usr/bin/env python
2"""
3从 XHS-Downloader 数据库导出笔记到单个 Markdown 文件
4 
5用法:
6    python export_memory.py [db_path] [output_file]
7 
8默认:
9    db_path: Volume/Download/ExploreData.db
10    output_file: xhs_memory.md
11"""
12import sqlite3
13import sys
14from pathlib import Path
15from datetime import datetime
16 
17 
18def export_memory(db_path: Path = None, output_file: Path = None):
19    db_path = db_path or Path("Volume/Download/ExploreData.db")
20    output_file = output_file or Path("xhs_memory.md")
21 
22    if not db_path.exists():
23        print(f"错误: 数据库不存在: {db_path}")
24        return False
25 
26    conn = sqlite3.connect(db_path)
27    cursor = conn.cursor()
28 
29    # 查询所有作品
30    cursor.execute("""
31        SELECT 作品标题, 发布时间, 作品链接, 作品描述, 作者昵称, 作品标签
32        FROM explore_data
33        ORDER BY 发布时间 DESC
34    """)
35 
36    rows = cursor.fetchall()
37    conn.close()
38 
39    if not rows:
40        print("数据库为空")
41        return False
42 
43    # 生成 Markdown
44    output = f"# 小红书收藏/点赞笔记 Memory\n\n"
45    output += f"> 导出时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n"
46    output += f"> 共 {len(rows)} 条笔记\n\n---\n\n"
47 
48    for i, (title, time, link, desc, author, tags) in enumerate(rows, 1):
49        output += f"## {i}. {title or '无标题'}\n\n"
50        output += f"- **作者**: {author or '未知'}\n"
51        output += f"- **时间**: {time or '未知'}\n"
52        output += f"- **链接**: {link or '无'}\n"
53        if tags:
54            output += f"- **标签**: {tags}\n"
55        output += f"\n### 内容\n\n{desc or '无内容'}\n\n---\n\n"
56 
57    # 保存文件
58    output_file.write_text(output, encoding="utf-8")
59    print(f"导出完成: {output_file.absolute()}")
60    print(f"共 {len(rows)} 条笔记")
61    return True
62 
63 
64if __name__ == "__main__":
65    db_path = Path(sys.argv[1]) if len(sys.argv) > 1 else None
66    output_file = Path(sys.argv[2]) if len(sys.argv) > 2 else None
67    export_memory(db_path, output_file)
68