fix: fix sentry error 145755

datagouv · Oct 30, 2024 · bcd85d2 · bcd85d2
1 parent 7c83764
commit bcd85d2
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/udata_hydra/utils/file.py b/udata_hydra/utils/file.py
@@ -52,15 +52,18 @@ async def download_resource(
     async with aiohttp.ClientSession(
         headers={"user-agent": config.USER_AGENT}, raise_for_status=True
     ) as session:
-        async with session.get(url, allow_redirects=True) as response:
-            async for chunk in response.content.iter_chunked(chunk_size):
-                if max_size_allowed is None or i * chunk_size < max_size_allowed:
-                    tmp_file.write(chunk)
-                else:
-                    tmp_file.close()
-                    log.warning(f"File {url} is too big, skipping")
-                    raise IOError("File too large to download")
-                i += 1
+        try:
+            async with session.get(url, allow_redirects=True) as response:
+                async for chunk in response.content.iter_chunked(chunk_size):
+                    if max_size_allowed is None or i * chunk_size < max_size_allowed:
+                        tmp_file.write(chunk)
+                    else:
+                        tmp_file.close()
+                        log.warning(f"File {url} is too big, skipping")
+                        raise IOError("File too large to download")
+                    i += 1
+        except aiohttp.ClientResponseError as e:
+            raise IOError(f"Error downloading CSV: {e}")
     tmp_file.close()
     if magic.from_file(tmp_file.name, mime=True) in [
         "application/x-gzip",