raise as more generic ConnectionError, because ClientResponseError ca…

…nnot be created easily
ungarj · Dec 3, 2024 · 260fc33 · 260fc33
1 parent 054ca7e
commit 260fc33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mapchete/executor/future.py b/mapchete/executor/future.py
@@ -123,6 +123,7 @@ def skip(skip_info: Optional[Any] = None, result: Optional[Any] = None) -> MFutu
     def from_func(
         func: Callable, fargs: Optional[Tuple] = None, fkwargs: Optional[Dict] = None
     ) -> MFuture:
+        fkwargs = fkwargs or {}
         try:
             return MFuture(result=func(*fargs, **fkwargs))
         except Exception as exc:  # pragma: no cover

diff --git a/mapchete/io/_misc.py b/mapchete/io/_misc.py
@@ -1,7 +1,6 @@
 import logging
 from enum import Enum
 
-from aiohttp import ClientResponseError
 import rasterio
 from rasterio.warp import calculate_default_transform
 from shapely.errors import TopologicalError
@@ -216,5 +215,7 @@ def copy(src_path, dst_path, src_fs=None, dst_fs=None, overwrite=False):
         if repr(exception).startswith(
             'Exception("ClientResponseError'
         ):  # pragma: no cover
-            raise ClientResponseError(exception) from exception
+            raise ConnectionError(repr(exception)).with_traceback(
+                exception.__traceback__
+            ) from exception
         raise