1

ai_pyppeteer.py

@@ -407,29 +407,7 @@ class ToWork:
                 logger.error(e)
                 logger.info('浏览器打开失败,正在重试')
 
-    # 初始化任务
-    async def on_start(self):
-        logger.info('on_start')
-        # 打开窗口
-        page = await self.open_browser(self.browser_item)
-        if not page:
-            # 浏览器打开失败
-            logger.error(f'{self.browser_item["num"]},窗口打开失败')
-            return
-        else:
-            logger.info(f'{self.browser_item["num"]},打开窗口')
-        while True:
-            # 获取任务
-            tasks = self.get_current_tasks()
-            if len(tasks):
-                # 开始执行任务
-                logger.info('开始执行任务')
-                status = await self.run_task(self.browser_item, tasks, page)
-            else:
-                time.sleep(3)
-                logger.info('不存在任务,等待3秒')
-
-    async def run_task(self, browser_item, tasks, page):
+    async def specified_server(self, page):
         try:
             logger.info('等待登录成功')
             # 判断页面是否加载完成
@@ -453,14 +431,45 @@ class ToWork:
                     element = await self.CheckElement(page, input_el)
                     if element:
                         logger.info('聊天窗口存在')
-                        # 输入指令
-                        await self.enter_prompt(page, tasks)
                         return True
             return False
         except Exception as e:
             logger.error(e)
             return False
 
+    # 初始化任务
+    async def on_start(self):
+        logger.info('on_start')
+        # 打开窗口
+        page = await self.open_browser(self.browser_item)
+        if not page:
+            # 浏览器打开失败
+            logger.error(f'{self.browser_item["num"]},窗口打开失败')
+            return
+        else:
+            logger.info(f'{self.browser_item["num"]},打开窗口')
+        # 进入到指定服务器
+        init_status = await self.specified_server(page)
+        if init_status:
+            while True:
+                # 获取任务
+                tasks = self.get_current_tasks()
+                if len(tasks):
+                    # 开始执行任务
+                    logger.info('开始执行任务')
+                    status = await self.run_task(self.browser_item, tasks, page)
+                else:
+                    time.sleep(2)
+                    logger.info('不存在任务,等待2秒')
+
+    async def run_task(self, browser_item, tasks, page):
+        try:
+            # 输入指令
+            await self.enter_prompt(page, tasks)
+        except Exception as e:
+            logger.error(e)
+            return False
+
 
 def process_start(browser_item, lock):
     try: