From 81b4b390ef7eff4ab7cfb5c111fccf7fdaac84cb Mon Sep 17 00:00:00 2001
From: JoeamAmier <yonglelolu@gmail.com>
Date: Sat, 9 Sep 2023 17:38:41 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=89=B9=E9=87=8F=E4=B8=8B?=
 =?UTF-8?q?=E8=BD=BD=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md          |  1 +
 main.py            |  2 +-
 source/Explore.py  | 11 ++++++-----
 source/__init__.py |  3 +++
 4 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 0f49094..b798e45 100644
--- a/README.md
+++ b/README.md
@@ -9,6 +9,7 @@
 <li>下载小红书图文/视频作品文件</li>
 <li>自动跳过已存在的作品文件</li>
 <li>作品文件完整性处理机制</li>
+<li>批量下载小红书作品文件</li>
 </ul>
 <h1>📸 程序截图</h1>
 <br>
diff --git a/main.py b/main.py
index 1a95368..afc7a5a 100644
--- a/main.py
+++ b/main.py
@@ -44,7 +44,7 @@ def main():
     xhs = XHS(**Settings().run())
     if ids := Batch().read_txt():
         for i in ids:
-            xhs.extract(i, download=True)
+            xhs.extract(i.rstrip('\n'), download=True)
     else:
         while True:
             if url := input("请输入小红书作品链接："):
diff --git a/source/Explore.py b/source/Explore.py
index 838c8e3..627e73f 100644
--- a/source/Explore.py
+++ b/source/Explore.py
@@ -20,11 +20,12 @@ class Explore:
 
     def __extract_data(self, data: dict) -> dict:
         result = {}
-        self.__extract_interact_info(result, data)
-        self.__extract_tags(result, data)
-        self.__extract_info(result, data)
-        self.__extract_time(result, data)
-        self.__extract_user(result, data)
+        if data:
+            self.__extract_interact_info(result, data)
+            self.__extract_tags(result, data)
+            self.__extract_info(result, data)
+            self.__extract_time(result, data)
+            self.__extract_user(result, data)
         return result
 
     @staticmethod
diff --git a/source/__init__.py b/source/__init__.py
index bee2aef..3a6e3ae 100644
--- a/source/__init__.py
+++ b/source/__init__.py
@@ -56,6 +56,9 @@ class XHS:
         if not html:
             return {}
         data = self.explore.run(html)
+        if not data:
+            print(f"获取作品数据失败: {url}")
+            return {}
         if data["作品类型"] == "视频":
             self.__get_video(data, html, download)
         else: