feat: 小红书增加手机号自动登录模式

media_platform/xhs/client.py

@@ -163,13 +163,15 @@ class XHSClient:
         }
         return await self.get(uri, params)
 
-    async def get_note_all_comments(self, note_id: str, crawl_interval: int = 1):
-        """get note all comments include sub comments
-
-        :param crawl_interval:
-        :param note_id: note id you want to fetch
-        :type note_id: str
+    async def get_note_all_comments(self, note_id: str, crawl_interval: float = 1.0, is_fetch_sub_comments=False):
         """
+        get note all comments include sub comments
+        :param note_id:
+        :param crawl_interval:
+        :param is_fetch_sub_comments:
+        :return:
+        """
+
         result = []
         comments_has_more = True
         comments_cursor = ""
@@ -178,6 +180,10 @@ class XHSClient:
             comments_has_more = comments_res.get("has_more", False)
             comments_cursor = comments_res.get("cursor", "")
             comments = comments_res["comments"]
+            if not is_fetch_sub_comments:
+                result.extend(comments)
+                continue
+            # handle get sub comments
             for comment in comments:
                 result.append(comment)
                 cur_sub_comment_count = int(comment["sub_comment_count"])