Merge pull request #699 from 2513502304/main

添加了对媒体资源服务器的异常处理，参见 issue #691

Author: NanmiCoder

media_platform/bilibili/client.py

@@ -202,12 +202,17 @@ async def get_video_play_url(self, aid: int, cid: int) -> Dict:
 
     async def get_video_media(self, url: str) -> Union[bytes, None]:
         async with httpx.AsyncClient(proxy=self.proxy) as client:
-            response = await client.request("GET", url, timeout=self.timeout, headers=self.headers)
-            if not response.reason_phrase == "OK":
-                utils.logger.error(f"[BilibiliClient.get_video_media] request {url} err, res:{response.text}")
+            try:
+                response = await client.request("GET", url, timeout=self.timeout, headers=self.headers)
+                response.raise_for_status()
+                if not response.reason_phrase == "OK":
+                    utils.logger.error(f"[BilibiliClient.get_video_media] request {url} err, res:{response.text}")
+                    return None
+                else:
+                    return response.content
+            except httpx.HTTPStatusError as exc:  # some wrong when call httpx.request method, such as connection error, client error or server error
+                utils.logger.error(f"[BilibiliClient.get_video_media] {exc}")
                 return None
-            else:
-                return response.content
 
     async def get_video_comments(
         self,

media_platform/bilibili/core.py

@@ -544,6 +544,7 @@ async def get_bilibili_video(self, video_item: Dict, semaphore: asyncio.Semaphor
             return
 
         content = await self.bili_client.get_video_media(video_url)
+        await asyncio.sleep(random.random())
         if content is None:
             return
         extension_file_name = f"video.mp4"

media_platform/douyin/client.py

@@ -30,7 +30,7 @@ class DouYinClient(AbstractApiClient):
 
     def __init__(
         self,
-        timeout=30,  # 若开启爬取媒体选项，抖音的短视频需要更久的超时时间
+        timeout=60,  # 若开启爬取媒体选项，抖音的短视频需要更久的超时时间
         proxy=None,
         *,
         headers: Dict,
@@ -305,17 +305,22 @@ async def get_all_user_aweme_posts(self, sec_user_id: str, callback: Optional[Ca
             posts_has_more = aweme_post_res.get("has_more", 0)
             max_cursor = aweme_post_res.get("max_cursor")
             aweme_list = aweme_post_res.get("aweme_list") if aweme_post_res.get("aweme_list") else []
-            utils.logger.info(f"[DouYinCrawler.get_all_user_aweme_posts] get sec_user_id:{sec_user_id} video len : {len(aweme_list)}")
+            utils.logger.info(f"[DouYinClient.get_all_user_aweme_posts] get sec_user_id:{sec_user_id} video len : {len(aweme_list)}")
             if callback:
                 await callback(aweme_list)
             result.extend(aweme_list)
         return result
 
     async def get_aweme_media(self, url: str) -> Union[bytes, None]:
         async with httpx.AsyncClient(proxy=self.proxy) as client:
-            response = await client.request("GET", url, timeout=self.timeout, follow_redirects=True)
-            if not response.reason_phrase == "OK":
-                utils.logger.error(f"[DouYinCrawler.get_aweme_media] request {url} err, res:{response.text}")
+            try:
+                response = await client.request("GET", url, timeout=self.timeout, follow_redirects=True)
+                response.raise_for_status()
+                if not response.reason_phrase == "OK":
+                    utils.logger.error(f"[DouYinClient.get_aweme_media] request {url} err, res:{response.text}")
+                    return None
+                else:
+                    return response.content
+            except httpx.HTTPStatusError as exc:  # some wrong when call httpx.request method, such as connection error, client error or server error
+                utils.logger.error(f"[DouYinClient.get_aweme_media] {exc}")
                 return None
-            else:
-                return response.content

media_platform/douyin/core.py

@@ -362,6 +362,7 @@ async def get_aweme_images(self, aweme_item: Dict):
             if not url:
                 continue
             content = await self.dy_client.get_aweme_media(url)
+            await asyncio.sleep(random.random())
             if content is None:
                 continue
             extension_file_name = f"{picNum:>03d}.jpeg"
@@ -385,6 +386,7 @@ async def get_aweme_video(self, aweme_item: Dict):
         if not video_download_url:
             return
         content = await self.dy_client.get_aweme_media(video_download_url)
+        await asyncio.sleep(random.random())
         if content is None:
             return
         extension_file_name = f"video.mp4"

media_platform/weibo/client.py

@@ -35,7 +35,7 @@ class WeiboClient:
 
     def __init__(
         self,
-        timeout=30,  # 若开启爬取媒体选项，weibo 的图片需要更久的超时时间
+        timeout=60,  # 若开启爬取媒体选项，weibo 的图片需要更久的超时时间
         proxy=None,
         *,
         headers: Dict[str, str],
@@ -248,12 +248,17 @@ async def get_note_image(self, image_url: str) -> bytes:
         final_uri = (f"{self._image_agent_host}"
                      f"{image_url}")
         async with httpx.AsyncClient(proxy=self.proxy) as client:
-            response = await client.request("GET", final_uri, timeout=self.timeout)
-            if not response.reason_phrase == "OK":
-                utils.logger.error(f"[WeiboClient.get_note_image] request {final_uri} err, res:{response.text}")
+            try:
+                response = await client.request("GET", final_uri, timeout=self.timeout)
+                response.raise_for_status()
+                if not response.reason_phrase == "OK":
+                    utils.logger.error(f"[WeiboClient.get_note_image] request {final_uri} err, res:{response.text}")
+                    return None
+                else:
+                    return response.content
+            except httpx.HTTPStatusError as exc:  # some wrong when call httpx.request method, such as connection error, client error or server error
+                utils.logger.error(f"[DouYinClient.get_aweme_media] {exc}")
                 return None
-            else:
-                return response.content
 
     async def get_creator_container_info(self, creator_id: str) -> Dict:
         """

media_platform/weibo/core.py

@@ -250,6 +250,7 @@ async def get_note_images(self, mblog: Dict):
             if not url:
                 continue
             content = await self.wb_client.get_note_image(url)
+            await asyncio.sleep(random.random())
             if content != None:
                 extension_file_name = url.split(".")[-1]
                 await weibo_store.update_weibo_note_image(pic["pid"], content, extension_file_name)

media_platform/xhs/client.py

@@ -32,7 +32,7 @@ class XiaoHongShuClient(AbstractApiClient):
 
     def __init__(
         self,
-        timeout=30,  # 若开启爬取媒体选项，xhs 的长视频需要更久的超时时间
+        timeout=60,  # 若开启爬取媒体选项，xhs 的长视频需要更久的超时时间
         proxy=None,
         *,
         headers: Dict[str, str],
@@ -152,12 +152,17 @@ async def post(self, uri: str, data: dict, **kwargs) -> Dict:
 
     async def get_note_media(self, url: str) -> Union[bytes, None]:
         async with httpx.AsyncClient(proxy=self.proxy) as client:
-            response = await client.request("GET", url, timeout=self.timeout)
-            if not response.reason_phrase == "OK":
-                utils.logger.error(f"[XiaoHongShuClient.get_note_media] request {url} err, res:{response.text}")
+            try:
+                response = await client.request("GET", url, timeout=self.timeout)
+                response.raise_for_status()
+                if not response.reason_phrase == "OK":
+                    utils.logger.error(f"[XiaoHongShuClient.get_note_media] request {url} err, res:{response.text}")
+                    return None
+                else:
+                    return response.content
+            except httpx.HTTPStatusError as exc:  # some wrong when call httpx.request method, such as connection error, client error or server error
+                utils.logger.error(f"[DouYinClient.get_aweme_media] {exc}")
                 return None
-            else:
-                return response.content
 
     async def pong(self) -> bool:
         """

media_platform/xhs/core.py

@@ -453,6 +453,7 @@ async def get_note_images(self, note_item: Dict):
             if not url:
                 continue
             content = await self.xhs_client.get_note_media(url)
+            await asyncio.sleep(random.random())
             if content is None:
                 continue
             extension_file_name = f"{picNum}.jpg"
@@ -476,6 +477,7 @@ async def get_notice_video(self, note_item: Dict):
         videoNum = 0
         for url in videos:
             content = await self.xhs_client.get_note_media(url)
+            await asyncio.sleep(random.random())
             if content is None:
                 continue
             extension_file_name = f"{videoNum}.mp4"