From 00789ba275d46ac21e17d966cf4974b1958b0aa1 Mon Sep 17 00:00:00 2001 From: DELL Date: Mon, 2 Mar 2026 15:13:46 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9Epost=E8=AF=B7=E6=B1=82?= =?UTF-8?q?=E8=B6=85=E6=97=B6=E6=97=B6=E9=97=B4=E4=BA=94=E5=88=86=E9=92=9F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../WebsiteSpider/WebsiteSpider/utils/selenium_utils.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/spiders/WebsiteSpider/WebsiteSpider/utils/selenium_utils.py b/spiders/WebsiteSpider/WebsiteSpider/utils/selenium_utils.py index 5193a3b..258ca43 100644 --- a/spiders/WebsiteSpider/WebsiteSpider/utils/selenium_utils.py +++ b/spiders/WebsiteSpider/WebsiteSpider/utils/selenium_utils.py @@ -3,6 +3,7 @@ import logging as logger import os import random +import requests from PIL import Image from selenium.webdriver import DesiredCapabilities from selenium.webdriver.firefox.options import Options @@ -25,10 +26,15 @@ def check_session(drive_path): api = drive_path + '/graphql' post_body = '{"query": "{ grid { maxSession, sessionCount } }"}' try: - response = http_post(api, post_body) + # 添加超时控制，5分钟 = 300秒 + response = http_post(api, post_body, timeout=300) data_body = json.loads(response.content.decode()) session_info = data_body['data']['grid'] return session_info + except requests.exceptions.Timeout as e: + logger.error("获取地址为 {} 的 Selenium 信息超时（超过5分钟）：".format(drive_path)) + logger.error(repr(e)) + return None except Exception as e: logger.warning("获取地址为 {} 的 Selenium 信息失败，错误信息是：".format(drive_path)) logger.warning(repr(e))