GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-12 17:20:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Maxim Devaev 2021-07-28 15:10:07 +03:00
parent 9aa3998924
commit da2e86ae72
2 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -123,7 +123,7 @@ class MsdApi:
name = htclient.get_filename(remote) name = htclient.get_filename(remote)
name = valid_msd_image_name(name) name = valid_msd_image_name(name)
size = htclient.get_content_length(remote) size = valid_int_f0(remote.content_length)
get_logger(0).info("Downloading image %r as %r to MSD ...", url, name) get_logger(0).info("Downloading image %r as %r to MSD ...", url, name)
async with self.__msd.write_image(name, size) as chunk_size: async with self.__msd.write_image(name, size) as chunk_size:

10
kvmd/htclient.py
View File

@ -51,16 +51,6 @@ def raise_not_200(response: aiohttp.ClientResponse) -> None:
) )
def get_content_length(response: aiohttp.ClientResponse) -> int:
try:
value = int(response.headers["Content-Length"])
except Exception:
raise aiohttp.ClientError("Empty or invalid Content-Length")
if value < 0:
raise aiohttp.ClientError("Negative Content-Length")
return value
def get_filename(response: aiohttp.ClientResponse) -> str: def get_filename(response: aiohttp.ClientResponse) -> str:
try: try:
disp = response.headers["Content-Disposition"] disp = response.headers["Content-Disposition"]