diff --git a/crawlers/tiktok/app/app_crawler.py b/crawlers/tiktok/app/app_crawler.py index ffb6ecb..21cae73 100644 --- a/crawlers/tiktok/app/app_crawler.py +++ b/crawlers/tiktok/app/app_crawler.py @@ -43,6 +43,9 @@ from crawlers.base_crawler import BaseCrawler from crawlers.tiktok.app.endpoints import TikTokAPIEndpoints from crawlers.utils.utils import model_to_query_string +# 重试机制 +from tenacity import * + # TikTok接口数据请求模型 from crawlers.tiktok.app.models import ( BaseRequestModel, FeedVideoDetail @@ -79,6 +82,7 @@ class TikTokAPPCrawler: # 获取单个作品数据 # @deprecated("TikTok APP fetch_one_video is deprecated and will be removed in a future release. Use Web API instead. | TikTok APP fetch_one_video 已弃用,将在将来的版本中删除。请改用Web API。") + @retry(stop=stop_after_attempt(10), wait=wait_fixed(1)) async def fetch_one_video(self, aweme_id: str): # 获取TikTok的实时Cookie kwargs = await self.get_tiktok_headers() diff --git a/requirements.txt b/requirements.txt index 77f07f6..bdd5079 100644 --- a/requirements.txt +++ b/requirements.txt @@ -34,4 +34,5 @@ ua-parser==0.18.0 user-agents==2.2.0 uvicorn==0.29.0 websockets==12.0 -gmssl==3.2.2 \ No newline at end of file +gmssl==3.2.2 +tenacity~=9.0.0 \ No newline at end of file