RGBCube/GitHubWrapper
1
Fork 0
mirror of https://github.com/RGBCube/GitHubWrapper synced 2025-05-21 16:35:08 +00:00
Code Issues Activity Actions

Make HTTPClient.request use orjson if it exists

Browse source
This commit is contained in:
RGBCube 2022-06-28 15:41:07 +03:00
parent 4723f0d23d
commit 0a2efb58cd
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
github/internals/http.py
View file

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