1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-20 07:55:09 +00:00

Make HTTPClient.request use orjson if it exists

This commit is contained in:
RGBCube 2022-06-28 15:41:07 +03:00
parent 4723f0d23d
commit 0a2efb58cd

View file

@ -14,6 +14,14 @@ from aiohttp import ClientSession, __version__ as aiohttp_version
from ..errors import error_from_request
from ..utils import human_readable_time_until
try:
import orjson
except ImportError:
import json
json_loads = json.loads
else:
json_loads = orjson.loads
if TYPE_CHECKING:
from aiohttp import BasicAuth
from typing_extensions import Self
@ -194,10 +202,12 @@ class HTTPClient:
)
if 200 <= response.status <= 299:
if response.headers["Content-Type"] == "application/json":
return await response.json()
data = await response.text(encoding='utf-8')
return await response.text(encoding="utf-8")
if response.headers["Content-Type"] == "application/json":
return json_loads(data)
return data
raise error_from_request(response)