zstdlib 0.10.0__tar.gz → 0.10.2__tar.gz

{zstdlib-0.10.0 → zstdlib-0.10.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zstdlib
-Version: 0.10.0
+Version: 0.10.2
 Summary: A set of useful python utilities
 License: GPLv3
 Project-URL: Homepage, https://github.com/zwimer/zstdlib

{zstdlib-0.10.0 → zstdlib-0.10.2}/tests/test_exc_capture.py

@@ -19,6 +19,14 @@ class TestLazy(unittest.TestCase):
         with self.assertRaises(ValueError), rt.capture():
             raise ValueError()
         self.assertEqual(len(rt.errors), 1)
+        # previous
+        with rt.capture():
+            pass
+        self.assertIs(rt.previous, None)
+        test_err = RuntimeError("test")
+        with rt.capture():
+            raise test_err
+        self.assertEqual(rt.previous, test_err)
         # capture's exc_t
         rt = ExcCapture(RuntimeError)
         with self.assertRaises(RuntimeError), rt.capture(ValueError):
@@ -64,6 +72,19 @@ class TestLazy(unittest.TestCase):
             with capture():
                 raise RuntimeError()
             raise ValueError()
+        # msg
+        _ = ExcDelay("")
+        with self.assertRaises(TypeError):
+            ExcDelay(RuntimeError)  # type: ignore[arg-type]
+        # armed
+        with ExcDelay("", RuntimeError, armed=False) as capture, capture():
+            raise RuntimeError()
+        ed = ExcDelay("", RuntimeError, armed=True)
+        with ed as capture, capture():
+            ed.armed = False
+            raise RuntimeError()
+        with self.assertRaises(ExceptionGroup), ExcDelay("", RuntimeError, armed=True) as capture, capture():
+            raise RuntimeError()
 
 
 if __name__ == "__main__":

{zstdlib-0.10.0 → zstdlib-0.10.2}/zstdlib/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.10.0"
+__version__ = "0.10.2"
 
 from .exc_capture import ExcCapture, ExcDelay
 from .not_set import NotSetType, NotSet

{zstdlib-0.10.0 → zstdlib-0.10.2}/zstdlib/exc_capture.py

@@ -17,7 +17,7 @@ class CaptureFType(Protocol):
 class ExcCapture:
     """A class that exposes a context manager which captures and suppresses any raised exception of a given type / tuple of types"""
 
-    __slots__ = ("errors", "exc_t")
+    __slots__ = ("errors", "exc_t", "previous")
 
     def __init__(self, exc_t: EXC_T | None = None) -> None:
         """
@@ -25,38 +25,50 @@ class ExcCapture:
             msg: The exception group message
             exc_t: The exception type the capture function should capture, if None it must be passed to the capture function directly
         """
-        self.exc_t: EXC_T | None = exc_t
+        self.previous: BaseException | None = None  # The result of the last capture() invocation
         self.errors: list[BaseException] = []
+        self.exc_t: EXC_T | None = exc_t
 
     @contextmanager
     def capture(self, exc_t: EXC_T | None = None) -> Iterator[None]:
-        """Capture and suppress any exception if type/tuple of types exc_t (self.exc_t if None)"""
+        """
+        Capture and suppress any exception if type/tuple of types exc_t (self.exc_t if None)
+        This function sets self.previous on exit
+        """
         exc_t = self.exc_t if exc_t is None else exc_t
         if exc_t is None:
             raise ValueError("No exception type provided here nor during construction")
         try:
             yield
+            self.previous = None
         except exc_t as e:
+            self.previous = e
             self.errors.append(e)
 
 
 class ExcDelay(ExcCapture):
     """A context manager subclass of ExcCapture that raises on exit if errors are captured"""
 
-    __slots__ = ("_msg",)
+    __slots__ = ("_msg", "armed")
 
     def __init__(
         self,
         msg: str,
         exc_t: type[BaseException] | None = None,
+        *,
+        armed: bool = True,
     ) -> None:
         """
         Args:
-            exc_t: The exception type the capture function should capture, if None it must be passed to the capture function directly
             msg: The exception group message
+            exc_t: The exception type the capture function should capture, if None it must be passed to the capture function directly
+            armed: If False, do not raise on context exit if self.errors
         """
+        if not isinstance(msg, str):
+            raise TypeError(f"Invalid message type: {type(msg)}")
+        self.armed: bool = armed
+        self._msg: str = msg
         super().__init__(exc_t)
-        self._msg = msg
 
     def __enter__(self) -> CaptureFType:
         """Return the capture function to use"""
@@ -64,6 +76,6 @@ class ExcDelay(ExcCapture):
 
     def __exit__(self, type_: type[BaseException] | None, *_: object) -> None:
         """Raise an ExceptionGroup (or BaseExceptionGroup depending on exc_t) if self.errors"""
-        if type_ is None and self.errors:
+        if type_ is None and self.errors and self.armed:
             # Auto-converts to ExceptionGroup if all items are Exceptions
             raise BaseExceptionGroup(self._msg, self.errors)

{zstdlib-0.10.0 → zstdlib-0.10.2}/zstdlib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zstdlib
-Version: 0.10.0
+Version: 0.10.2
 Summary: A set of useful python utilities
 License: GPLv3
 Project-URL: Homepage, https://github.com/zwimer/zstdlib