1#!/usr/bin/env python
2"""
3批量下载小红书笔记
4 
5用法:
6    python batch_download.py [links_file]
7 
8默认读取当前目录的 links.md 文件
9"""
10import asyncio
11import sys
12from pathlib import Path
13 
14try:
15    from source import XHS
16except ImportError:
17    print("错误: 请在 XHS-Downloader 项目目录下运行此脚本")
18    print("或安装依赖: pip install -e /path/to/XHS-Downloader")
19    sys.exit(1)
20 
21 
22async def main():
23    # 读取链接文件
24    links_file = Path(sys.argv[1]) if len(sys.argv) > 1 else Path("links.md")
25 
26    if not links_file.exists():
27        print(f"错误: 链接文件不存在: {links_file}")
28        print("用法: python batch_download.py [links_file]")
29        sys.exit(1)
30 
31    links = links_file.read_text().strip()
32    link_count = len([l for l in links.split() if l.startswith("http")])
33 
34    print(f"开始下载，共 {link_count} 个链接...")
35 
36    async with XHS(
37        work_path="./Volume",
38        folder_name="Download",
39        record_data=True,       # 记录作品数据到数据库
40        download_record=True,   # 跳过已下载
41        author_archive=True,    # 按作者分文件夹
42    ) as xhs:
43        result = await xhs.extract(links, download=True)
44        print(f"完成！处理了 {len(result)} 个作品")
45 
46 
47if __name__ == "__main__":
48    asyncio.run(main())
49