refactoring

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"]