user defined exceptions no longer inherit from KeyError

Author: amorison

src/stagpy/_caching.py

@@ -8,6 +8,7 @@
 from functools import cached_property
 
 from . import phyvars, stagyyparsers
+from .error import InvalidSnapshotError
 
 if typing.TYPE_CHECKING:
     from collections.abc import Mapping
@@ -169,7 +170,7 @@ def isnap(self, *, istep: int) -> int | None:
                 isnap += 1
                 try:
                     istep_try = self.sdat.snaps[isnap].istep
-                except KeyError:
+                except InvalidSnapshotError:
                     pass
                 # all intermediate istep could have their isnap to None
                 self._snap_to_step[isnap] = istep_try

src/stagpy/error.py

@@ -99,7 +99,7 @@ def __init__(self, file: PathLike, msg: str):
         super().__init__(file, msg)
 
 
-class InvalidTimestepError(StagpyError, KeyError):
+class InvalidTimestepError(StagpyError):
     """Raised when invalid time step is requested.
 
     Attributes:
@@ -115,7 +115,7 @@ def __init__(self, sdat: StagyyData, istep: int, msg: str):
         super().__init__(sdat, istep, msg)
 
 
-class InvalidSnapshotError(StagpyError, KeyError):
+class InvalidSnapshotError(StagpyError):
     """Raised when invalid snapshot is requested.
 
     Attributes:
@@ -167,11 +167,11 @@ def __init__(self, zoom: float):
         super().__init__(f"Zoom angle should be in [0,360] (received {zoom})")
 
 
-class MissingDataError(StagpyError, KeyError):
+class MissingDataError(StagpyError):
     """Raised when requested data is not present in output."""
 
 
-class UnknownVarError(StagpyError, KeyError):
+class UnknownVarError(StagpyError):
     """Raised when invalid var is requested.
 
     Attributes:

src/stagpy/stagyydata.py

@@ -568,7 +568,7 @@ def _pass(self, item: int) -> bool:
         """Check whether an item passes the filters."""
         try:
             step = self.over[item]
-        except KeyError:
+        except (error.InvalidTimestepError, error.InvalidSnapshotError):
             return False
         return self.filters.passes(step)