检查秘塔session每日额度是否用完

spiders/ai_seo/metaso.py

@@ -8,7 +8,7 @@ from json import JSONDecodeError
 import ftfy
 import pyperclip
 from playwright.async_api import Browser, async_playwright
-
+from utils.ai_seo_api_utils import AiSeoApis
 import config
 from abs_spider import AbstractAiSeoSpider
 from domain.ai_seo import AiAnswer, AiSearchResult
@@ -31,6 +31,14 @@ class MetasoSpider(AbstractAiSeoSpider):
         self._init_data()
         await self.browser_page.goto(self.get_home_url(), timeout=600000)
         await asyncio.sleep(2)
+        info = await self.browser_page.wait_for_selector('#left-menu > div > div.LeftMenu_footer__qsJdJ > div > div > div > button', timeout=600000)
+        await info.click()
+
+        edu = await self.browser_page.wait_for_selector('body > div:nth-child(51) > div > div > div > div > div.MuiBox-root.css-o45jia > div:nth-child(2) > div.MuiListItemText-root.css-rkhw2f', timeout=600000)
+        edu_txt= await edu.text_content()
+        if edu_txt == '0':
+            await AiSeoApis.update_spider_session(self.session_info['id'], 3)
+            raise "session额度已用完!"
         # 开始操作
         chat_input_element = self.browser_page.locator("//textarea[contains(@class, 'search-consult-textarea')]")
         # 输入提问词
@@ -42,7 +50,6 @@ class MetasoSpider(AbstractAiSeoSpider):
         await self.browser_page.reload()
         # await self.completed_event.wait()
         # 等待指定元素
-
         #//*[@id="search-content-container-8626530479804592128"]/div[2]/button
         copy_button = await self.browser_page.wait_for_selector('//*[starts-with(@id, "search-content-container-")]/div[2]/button', timeout=600000)
         # 点击复制按钮