GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-01-31 18:11:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

send error in streaming response

Browse Source
This commit is contained in:
Maxim Devaev
2021-07-30 07:30:22 +03:00
parent 7d7f4965de
commit 4a1f51bc30
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
kvmd/apps/kvmd/api/msd.py
View File

@@ -106,9 +106,9 @@ class MsdApi:
size = written = 0
response: Optional[StreamResponse] = None
async def stream_write_info() -> None:
async def stream_write_info(err: Optional[Exception]=None) -> None:
assert response is not None
await stream_json(response, self.__make_write_info(name, size, written))
await stream_json(response, self.__make_write_info(name, size, written), err)
try:
async with htclient.download(
@@ -141,7 +141,7 @@ class MsdApi:
except Exception as err:
if response is not None:
await stream_write_info()
await stream_write_info(err)
elif isinstance(err, aiohttp.ClientError):
return make_json_exception(err, 400)
raise