PaiGramTeam/HonkaiStarRailWikiDataParser
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/HonkaiStarRailWikiDataParser.git synced 2024-09-28 00:35:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix: timeout error

Browse Source
This commit is contained in:
xtaodada 2023-04-25 22:53:37 +08:00
parent 6ea54ee5e0
commit 0c9f1f8b86
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
modules/apihelper/client/base/httpxrequest.py
View File

@ -5,11 +5,18 @@ from typing import Optional, Type
import httpx
__all__ = ("HTTPXRequest",)
timeout = httpx.Timeout(
timeout=120.0,
read=120.0,
write=120.0,
connect=120.0,
pool=120.0,
)
class HTTPXRequest(AbstractAsyncContextManager):
def __init__(self, *args, headers=None, **kwargs):
self._client = httpx.AsyncClient(headers=headers, timeout=httpx.Timeout(120.0), *args, **kwargs)
self._client = httpx.AsyncClient(headers=headers, timeout=timeout, *args, **kwargs)
async def __aenter__(self):
try:
@ -20,13 +27,14 @@ class HTTPXRequest(AbstractAsyncContextManager):
raise exc
async def __aexit__(
self, exc_type: Optional[Type[BaseException]], exc_val: Optional[BaseException], exc_tb: Optional[TracebackType]
self, exc_type: Optional[Type[BaseException]], exc_val: Optional[BaseException],
exc_tb: Optional[TracebackType]
) -> None:
await self.initialize()
async def initialize(self):
if self._client.is_closed:
self._client = httpx.AsyncClient()
self._client = httpx.AsyncClient(timeout=timeout)
async def shutdown(self):
if self._client.is_closed:

2
modules/apihelper/client/components/hyperion.py
View File

@ -123,7 +123,7 @@ class Hyperion:
_, file_extension = os.path.splitext(filename)
is_image = bool(file_extension in ".jpg" or file_extension in ".png")
response = await self.client.get(
url, params=self.get_images_params(resize=2000) if is_image else None, timeout=10, de_json=False
url, params=self.get_images_params(resize=2000) if is_image else None, de_json=False
)
return ArtworkImage(
art_id=art_id, page=page, file_name=filename, file_extension=url.split(".")[-1], data=response.content