A-Minos/nonebot-plugin-tetris-stats
1
0
Fork 0
mirror of https://github.com/A-Minos/nonebot-plugin-tetris-stats.git synced 2026-03-05 05:36:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ 添加状态码检查

Browse Source
This commit is contained in:
scdhh
2023-12-30 06:52:45 +08:00
parent abf4410a00
commit 4afda62782
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
nonebot_plugin_tetris_stats/utils/request.py
View File

@@ -1,3 +1,4 @@
from http import HTTPStatus
from urllib.parse import urljoin, urlparse from urllib.parse import urljoin, urlparse
from aiofiles import open from aiofiles import open
@@ -113,11 +114,15 @@ class Request:
try: try:
async with AsyncClient(cookies=cls._cookies, timeout=config.tetris_req_timeout) as session: async with AsyncClient(cookies=cls._cookies, timeout=config.tetris_req_timeout) as session:
response = await session.get(url, headers=cls._headers) response = await session.get(url, headers=cls._headers)
if response.status_code != HTTPStatus.OK:
raise RequestError(
f'请求错误 code: {response.status_code} {HTTPStatus(response.status_code).phrase}\n{response.text}'
)
if is_json: if is_json:
loads(response.content) loads(response.content)
return response.content return response.content
except HTTPError as e: except HTTPError as e:
raise RequestError(f'请求错误\n{e!r}') from e raise RequestError(f'请求错误 \n{e!r}') from e
except JSONDecodeError: except JSONDecodeError:
if urlparse(url).netloc.lower().endswith('tetr.io'): if urlparse(url).netloc.lower().endswith('tetr.io'):
return await cls._anti_cloudflare(url) return await cls._anti_cloudflare(url)