From bfadac4f791255c2089457ea262514b9f2315f68 Mon Sep 17 00:00:00 2001 From: scdhh Date: Wed, 15 Nov 2023 00:43:12 +0800 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=E6=B7=BB=E5=8A=A0=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E9=A1=B9=20=E8=AF=B7=E6=B1=82=E8=B6=85=E6=97=B6?= =?UTF-8?q?=E6=97=B6=E9=97=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nonebot_plugin_tetris_stats/config/config.py | 2 ++ nonebot_plugin_tetris_stats/utils/request.py | 5 +++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/nonebot_plugin_tetris_stats/config/config.py b/nonebot_plugin_tetris_stats/config/config.py index cab7c97..c35d67d 100644 --- a/nonebot_plugin_tetris_stats/config/config.py +++ b/nonebot_plugin_tetris_stats/config/config.py @@ -10,3 +10,5 @@ CACHE_PATH: Path = get_cache_dir('nonebot_plugin_tetris_stats') class Config(BaseModel): """配置类""" + + tetris_req_timeout: float = 30.0 diff --git a/nonebot_plugin_tetris_stats/utils/request.py b/nonebot_plugin_tetris_stats/utils/request.py index 2ce3bdf..5919d1d 100644 --- a/nonebot_plugin_tetris_stats/utils/request.py +++ b/nonebot_plugin_tetris_stats/utils/request.py @@ -7,11 +7,12 @@ from nonebot.log import logger from playwright.async_api import Response from ujson import JSONDecodeError, dumps, loads -from ..config.config import CACHE_PATH +from ..config.config import CACHE_PATH, Config from .browser import BrowserManager from .exception import RequestError driver = get_driver() +config = Config.parse_obj(driver.config) @driver.on_startup @@ -115,7 +116,7 @@ class Request: async def request(cls, url: str, *, is_json: bool = True) -> bytes: """请求api""" try: - async with AsyncClient(cookies=cls._cookies) as session: + async with AsyncClient(cookies=cls._cookies, timeout=config.tetris_req_timeout) as session: response = await session.get(url, headers=cls._headers) if is_json: loads(response.content)