添加状态码检查

This commit is contained in:
2023-12-30 06:52:45 +08:00
parent abf4410a00
commit 4afda62782

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,6 +114,10 @@ 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