mirror of
https://github.com/JoeanAmier/XHS-Downloader.git
synced 2025-12-26 04:48:05 +08:00
110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
from textual.app import App
|
||
from textual.app import ComposeResult
|
||
from textual.binding import Binding
|
||
from textual.containers import HorizontalScroll
|
||
from textual.widgets import Button
|
||
from textual.widgets import Footer
|
||
from textual.widgets import Header
|
||
from textual.widgets import Input
|
||
from textual.widgets import Label
|
||
from textual.widgets import Log
|
||
|
||
from source import Batch
|
||
from source import Settings
|
||
from source import XHS
|
||
|
||
|
||
def example():
|
||
"""通过代码设置参数，适合二次开发"""
|
||
# 测试链接
|
||
error_demo = "https://github.com/JoeanAmier/XHS_Downloader"
|
||
image_demo = "https://www.xiaohongshu.com/explore/63b275a30000000019020185"
|
||
video_demo = "https://www.xiaohongshu.com/explore/64edb460000000001f03cadc"
|
||
# 实例对象
|
||
path = "" # 作品下载储存根路径，默认值：当前路径
|
||
folder = "Download" # 作品下载文件夹名称（自动创建），默认值：Download
|
||
cookie = "" # 小红书网页版 Cookie
|
||
proxies = None # 网络代理
|
||
timeout = 5 # 网络请求超时限制，默认值：10
|
||
chunk = 1024 * 1024 # 下载文件时，每次从服务器获取的数据块大小，单位字节
|
||
xhs = XHS(
|
||
path=path,
|
||
folder=folder,
|
||
cookie=cookie,
|
||
proxies=proxies,
|
||
timeout=timeout,
|
||
chunk=chunk, ) # 使用自定义参数
|
||
# xhs = XHS() # 使用默认参数
|
||
download = True # 是否下载作品文件
|
||
# 返回作品详细信息，包括下载地址
|
||
print(xhs.extract(error_demo)) # 获取数据失败时返回空字典
|
||
print(xhs.extract(image_demo, download=download))
|
||
print(xhs.extract(video_demo, download=download))
|
||
|
||
|
||
def program():
|
||
"""读取并应用配置文件设置的参数，适合一般作品文件下载需求"""
|
||
print("如果采集数据失败，请尝试使用手动获取的 Cookie 运行程序！")
|
||
xhs = XHS(**Settings().run())
|
||
if ids := Batch().read_txt():
|
||
for i in ids:
|
||
print(f"当前作品链接: {i}")
|
||
xhs.extract(i, download=True)
|
||
else:
|
||
while True:
|
||
if url := input("请输入小红书作品链接："):
|
||
xhs.extract(url, download=True)
|
||
else:
|
||
break
|
||
|
||
|
||
class XHSDownloader(App):
|
||
CSS_PATH = "static/XHS_Downloader.tcss"
|
||
BINDINGS = [
|
||
Binding(key="q", action="quit", description="退出程序"),
|
||
("d", "toggle_dark", "切换主题"),
|
||
]
|
||
APP = XHS(**Settings().run())
|
||
Batch = Batch()
|
||
|
||
def compose(self) -> ComposeResult:
|
||
yield Header()
|
||
yield Label("请输入小红书图文/视频作品链接：")
|
||
yield Input(placeholder="URL")
|
||
yield HorizontalScroll(Button("下载无水印图片/视频", id="solo", variant="success"),
|
||
Button("读取文件并开始批量下载作品", id="batch", variant="success"),
|
||
Button("清空输入框", id="reset", variant="error"), )
|
||
yield Log(auto_scroll=True)
|
||
yield Footer()
|
||
|
||
def on_mount(self) -> None:
|
||
self.title = "小红书作品采集工具"
|
||
|
||
def on_button_pressed(self, event: Button.Pressed) -> None:
|
||
if event.button.id == "solo":
|
||
self.solo()
|
||
elif event.button.id == "batch":
|
||
self.batch()
|
||
elif event.button.id == "reset":
|
||
self.query_one(Input).value = ""
|
||
|
||
def solo(self):
|
||
url = self.query_one(Input).value
|
||
self.APP.extract(url, True, self.query_one(Log))
|
||
|
||
def batch(self):
|
||
urls = self.Batch.read_txt()
|
||
log = self.query_one(Log)
|
||
if not urls:
|
||
log.write_line("未检测到 xhs.txt 文件 或者 该文件为空！")
|
||
for url in urls:
|
||
log.write_line(f"当前作品链接: {url}")
|
||
self.APP.extract(url, True, log)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# example()
|
||
# program()
|
||
app = XHSDownloader()
|
||
app.run()
|