49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
#!/usr/bin/env python
|
|
"""
|
|
批量下载小红书笔记
|
|
|
|
用法:
|
|
python batch_download.py [links_file]
|
|
|
|
默认读取当前目录的 links.md 文件
|
|
"""
|
|
import asyncio
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
try:
|
|
from source import XHS
|
|
except ImportError:
|
|
print("错误: 请在 XHS-Downloader 项目目录下运行此脚本")
|
|
print("或安装依赖: pip install -e /path/to/XHS-Downloader")
|
|
sys.exit(1)
|
|
|
|
|
|
async def main():
|
|
# 读取链接文件
|
|
links_file = Path(sys.argv[1]) if len(sys.argv) > 1 else Path("links.md")
|
|
|
|
if not links_file.exists():
|
|
print(f"错误: 链接文件不存在: {links_file}")
|
|
print("用法: python batch_download.py [links_file]")
|
|
sys.exit(1)
|
|
|
|
links = links_file.read_text().strip()
|
|
link_count = len([l for l in links.split() if l.startswith("http")])
|
|
|
|
print(f"开始下载，共 {link_count} 个链接...")
|
|
|
|
async with XHS(
|
|
work_path="./Volume",
|
|
folder_name="Download",
|
|
record_data=True, # 记录作品数据到数据库
|
|
download_record=True, # 跳过已下载
|
|
author_archive=True, # 按作者分文件夹
|
|
) as xhs:
|
|
result = await xhs.extract(links, download=True)
|
|
print(f"完成！处理了 {len(result)} 个作品")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|