refactor get_url_size

EcoExtreML · Oct 22, 2024 · c24bbf7 · c24bbf7
1 parent 49637d1
commit c24bbf7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/zampy/datasets/utils.py b/src/zampy/datasets/utils.py
@@ -43,10 +43,13 @@ def download_url(url: str, fpath: Path, overwrite: bool) -> None:
         print(f"File '{fpath.name}' already exists, skipping...")
 
 
-def get_url_size(url: str) -> int:
+def get_url_size(url: str) -> int | None:
     """Return the size (bytes) of a given URL."""
     response = requests.head(url)
-    return int(response.headers["Content-Length"])
+    content_length = response.headers.get("Content-Length")
+    if content_length:
+        return int(content_length)
+    return None
 
 
 def get_file_size(fpath: Path) -> int: