update

toolbox/douyin/live/live_recording.py

@@ -159,6 +159,11 @@ class LiveRecording(FollowManager):
         return result
 
     @async_cache_decorator(600)
+    @retry(
+        wait=wait_fixed(60),
+        stop=stop_after_attempt(3),
+        before_sleep=before_sleep_log(logger, logging.ERROR),
+    )
     async def get_live_info_by_follow(self, room_id: str):
         result = None
         live_info_list: List[dict] = await self.get_living_list()

toolbox/porter/tasks/douyin_live_record_task.py

@@ -174,7 +174,7 @@ class DouyinLiveRecordTask(BaseTask):
             return None
 
         try:
-            live_info_list: List[dict] = await self.douyin_client.get_living_list(self.room_id)
+            living_list: List[dict] = await self.douyin_client.get_living_list()
         except Exception as error:
             logger.error(f"get_living_list failed; error type: {type(error)}, error text: {str(error)}, traceback: {traceback.format_exc()}")
             self.success_rate_of_get_live_info_by_follow.append(0)
@@ -185,10 +185,10 @@ class DouyinLiveRecordTask(BaseTask):
             "title": "",
             "stream_data": None,
         }
-        for live_info in live_info_list:
-            room_id_ = live_info["room_id"]
+        for live_data in living_list:
+            room_id_ = live_data["room_id"]
             if room_id_ == self.room_id:
-                result = live_info
+                result = live_data
                 break
         self.success_rate_of_get_live_info_by_follow.append(1)
         return result

toolbox/porter/tasks/douyin_live_to_bilibili_live_task.py

@@ -106,7 +106,7 @@ class DouyinLiveToBilibiliLiveTask(BaseTask):
             return None
 
         try:
-            live_info_list: List[dict] = await self.douyin_client.get_living_list(self.room_id)
+            living_list: List[dict] = await self.douyin_client.get_living_list()
         except Exception as error:
             logger.error(f"get_living_list failed; error type: {type(error)}, error text: {str(error)}, traceback: {traceback.format_exc()}")
             self.success_rate_of_get_live_info_by_follow.append(0)
@@ -117,10 +117,10 @@ class DouyinLiveToBilibiliLiveTask(BaseTask):
             "title": "",
             "stream_data": None,
         }
-        for live_info in live_info_list:
-            room_id_ = live_info["room_id"]
+        for live_data in living_list:
+            room_id_ = live_data["room_id"]
             if room_id_ == self.room_id:
-                result = live_info
+                result = live_data
                 break
         self.success_rate_of_get_live_info_by_follow.append(1)
         return result