docs: 更新tiktok代码片段 (#197)

Johnserf-Seed · Dec 1, 2024 · 3b4a008 · 3b4a008
1 parent d95eca6
commit 3b4a008
Show file tree

Hide file tree

Showing 6 changed files with 43 additions and 25 deletions.
diff --git a/docs/snippets/tiktok/user-collect.py b/docs/snippets/tiktok/user-collect.py
@@ -18,15 +18,15 @@ async def main():
         "YOUR_HOME_PAGE"
     )  # 替换开放收藏列表的用户主页
 
-    async for aweme_data_list in TiktokHandler(kwargs).fetch_user_collect_videos(
+    async for collect_list in TiktokHandler(kwargs).fetch_user_collect_videos(
         secUid, 0, 10, 20
     ):
         print("=================_to_raw================")
-        print(aweme_data_list._to_raw())
+        print(collect_list._to_raw())
         # print("=================_to_dict===============")
-        # print(aweme_data_list._to_dict())
+        # print(collect_list._to_dict())
         # print("=================_to_list===============")
-        # print(aweme_data_list._to_list())
+        # print(collect_list._to_list())
 
 
 if __name__ == "__main__":

diff --git a/docs/snippets/tiktok/user-like.py b/docs/snippets/tiktok/user-like.py
@@ -18,15 +18,15 @@ async def main():
         "YOUR_HOME_PAGE"
     )  # 替换开放喜欢列表的用户主页
 
-    async for aweme_data_list in TiktokHandler(kwargs).fetch_user_like_videos(
+    async for user_like_list in TiktokHandler(kwargs).fetch_user_like_videos(
         secUid, 0, 10, 20
     ):
         print("=================_to_raw================")
-        print(aweme_data_list._to_raw())
+        print(user_like_list._to_raw())
         # print("=================_to_dict===============")
-        # print(aweme_data_list._to_dict())
+        # print(user_like_list._to_dict())
         # print("=================_to_list===============")
-        # print(aweme_data_list._to_list())
+        # print(user_like_list._to_list())
 
 
 if __name__ == "__main__":

diff --git a/docs/snippets/tiktok/user-live-im-fetch.py b/docs/snippets/tiktok/user-live-im-fetch.py
@@ -1,4 +1,5 @@
 import asyncio
+from f2.log.logger import logger
 from f2.apps.tiktok.handler import TiktokHandler
 
 
@@ -27,8 +28,15 @@
 
 
 async def main():
+    room_id = "7404848324131638062"
+
+    room = await TiktokHandler(kwargs).fetch_check_live_alive(room_id)
+    if not room.is_alive[0]:
+        logger.info("直播间：{0} 未开播".format(room_id))
+        return
+
     # 通过该接口获取wss所需的cursor和internal_ext
-    live_im = await TiktokHandler(kwargs).fetch_live_im(room_id="7404848324131638062")
+    live_im = await TiktokHandler(kwargs).fetch_live_im(room_id=room_id)
     # logger.info(
     #     "直播间IM页码：",
     #     live_im.cursor,
@@ -38,13 +46,14 @@ async def main():
     #     live_im.routeParams.wrss,
     # )
 
-    # # 获取直播弹幕
-    await TiktokHandler(kwargs2).fetch_live_danmaku(
-        room_id="7404848324131638062",
-        internal_ext=live_im.internalExt,
-        cursor=live_im.cursor,
-        wrss=live_im.routeParams.wrss,
-    )
+    if live_im:
+        # 获取直播间信息
+        await TiktokHandler(kwargs2).fetch_live_danmaku(
+            room_id=room_id,
+            internal_ext=live_im.internalExt,
+            cursor=live_im.cursor,
+            wrss=live_im.routeParams.wrss,
+        )
 
 
 if __name__ == "__main__":

diff --git a/docs/snippets/tiktok/user-mix.py b/docs/snippets/tiktok/user-mix.py
@@ -17,16 +17,25 @@
 
 async def main():
     secUid = await SecUserIdFetcher.get_secuid("https://www.tiktok.com/@vantoan___")
-    playlist = await TiktokHandler(kwargs).fetch_play_list(secUid)
-
-    for mixId in playlist.get("mixId", []):
-        async for aweme_data_list in TiktokHandler(kwargs).fetch_user_mix_videos(mixId):
+    playlist = await TiktokHandler(kwargs).fetch_play_list(
+        secUid,
+        cursor=0,
+        page_counts=20,
+    )
+
+    for mixId in playlist._to_dict().get("mixId", []):
+        async for user_mix_list in TiktokHandler(kwargs).fetch_user_mix_videos(
+            mixId,
+            cursor=0,
+            page_counts=20,
+            max_counts=20,
+        ):
             print("=================_to_raw================")
-            print(aweme_data_list._to_raw())
+            print(user_mix_list._to_raw())
             # print("=================_to_dict===============")
-            # print(aweme_data_list._to_dict())
+            # print(user_mix_list._to_dict())
             # print("=================_to_list===============")
-            # print(aweme_data_list._to_list())
+            # print(user_mix_list._to_list())
 
 
 if __name__ == "__main__":

diff --git a/docs/snippets/tiktok/user-post.py b/docs/snippets/tiktok/user-post.py
@@ -18,7 +18,7 @@ async def main():
     secUid = await SecUserIdFetcher.get_secuid("https://www.tiktok.com/@vantoan___")
 
     async for aweme_data_list in TiktokHandler(kwargs).fetch_user_post_videos(
-        secUid, 0, 0, 10, None
+        secUid, 0, 0, 10, 20
     ):
         print("=================_to_raw================")
         print(aweme_data_list._to_raw())

diff --git a/docs/snippets/tiktok/user-profile.py b/docs/snippets/tiktok/user-profile.py
@@ -7,8 +7,8 @@
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36 Edg/130.0.0.0",
         "Referer": "https://www.tiktok.com/",
     },
-    "cookie": "YOUR_COOKIE_HERE",
     "proxies": {"http://": None, "https://": None},
+    "cookie": "YOUR_COOKIE_HERE",
 }