diff options
| -rw-r--r-- | kvmd/apps/kvmd/api/msd.py | 2 | ||||
| -rw-r--r-- | kvmd/htclient.py | 10 |
2 files changed, 1 insertions, 11 deletions
diff --git a/kvmd/apps/kvmd/api/msd.py b/kvmd/apps/kvmd/api/msd.py index 1f1afd1e..a8d44e5e 100644 --- a/kvmd/apps/kvmd/api/msd.py +++ b/kvmd/apps/kvmd/api/msd.py @@ -123,7 +123,7 @@ class MsdApi: name = htclient.get_filename(remote) 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) async with self.__msd.write_image(name, size) as chunk_size: diff --git a/kvmd/htclient.py b/kvmd/htclient.py index 1927ede2..208f94b5 100644 --- a/kvmd/htclient.py +++ b/kvmd/htclient.py @@ -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: try: disp = response.headers["Content-Disposition"] |