xyvs/wakapi-readme-stats
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 1 Wiki Activity

fix: status code 202 error

Browse Source
This commit is contained in:
anmol singh
2023-11-22 11:36:38 +05:30
parent 02827d6adc
commit 7063c39f20
3 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
sources/manager_download.py
View File

@@ -166,7 +166,7 @@ class DownloadManager:
await resource
@staticmethod
async def _get_remote_resource(resource: str, convertor: Optional[Callable[[bytes], Dict]]) -> Dict:
async def _get_remote_resource(resource: str, convertor: Optional[Callable[[bytes], Dict]]) -> Dict or None:
"""
Receive execution result of static query, wait for it if necessary.
If the query wasn't cached previously, cache it.
@@ -174,7 +174,7 @@ class DownloadManager:
:param resource: Static query identifier.
:param convertor: Optional function to convert `response.contents` to dict.
By default `response.json()` is used.
:return: Response dictionary.
:return: Response dictionary or None.
"""
DBM.i(f"\tMaking a remote API query named '{resource}'...")
if isinstance(DownloadManager._REMOTE_RESOURCES_CACHE[resource], Awaitable):
@@ -184,16 +184,22 @@ class DownloadManager:
else:
res = DownloadManager._REMOTE_RESOURCES_CACHE[resource]
DBM.g(f"\tQuery '{resource}' loaded from cache!")
if res.status_code == 200 or res.status_code == 202:
if res.status_code == 200:
if convertor is None:
return res.json()
else:
return convertor(res.content)
elif res.status_code == 201:
DBM.w(f"\tQuery '{resource}' returned 201 status code")
return None
elif res.status_code == 202:
DBM.w(f"\tQuery '{resource}' returned 202 status code")
return None
else:
raise Exception(f"Query '{res.url}' failed to run by returning code of {res.status_code}: {res.json()}")
@staticmethod
async def get_remote_json(resource: str) -> Dict:
async def get_remote_json(resource: str) -> Dict or None:
"""
Shortcut for `_get_remote_resource` to return JSON response data.
:param resource: Static query identifier.
@@ -202,7 +208,7 @@ class DownloadManager:
return await DownloadManager._get_remote_resource(resource, None)
@staticmethod
async def get_remote_yaml(resource: str) -> Dict:
async def get_remote_yaml(resource: str) -> Dict or None:
"""
Shortcut for `_get_remote_resource` to return YAML response data.
:param resource: Static query identifier.