jetpytools 1.6.3__tar.gz → 1.6.4__tar.gz

{jetpytools-1.6.3 → jetpytools-1.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jetpytools
-Version: 1.6.3
+Version: 1.6.4
 Summary: Collection of stuff that's useful in general python programming
 Project-URL: Source Code, https://github.com/Jaded-Encoding-Thaumaturgy/jetpytools
 Project-URL: Contact, https://discord.gg/XTpc6Fa9eB

{jetpytools-1.6.3 → jetpytools-1.6.4}/jetpytools/_metadata.py

@@ -1,6 +1,6 @@
 """Collection of stuff that's useful in general python programming"""
 
-__version__ = "1.6.3"
+__version__ = "1.6.4"
 
 __author_name__, __author_email__ = "Jaded Encoding Thaumaturgy", "jaded.encoding.thaumaturgy@gmail.com"
 __maintainer_name__, __maintainer_email__ = __author_name__, __author_email__

{jetpytools-1.6.3 → jetpytools-1.6.4}/jetpytools/exceptions/base.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
 import sys
+from contextlib import AbstractContextManager
 from copy import deepcopy
-from typing import TYPE_CHECKING, Any, TypeVar
+from types import TracebackType
+from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
 from typing_extensions import Self
 
@@ -166,12 +168,69 @@ class CustomError(ExceptionError, metaclass=CustomErrorMeta):
         else:
             reason = ""
 
-        return f"{func_header}{self.message!s}{reason}".format(**self.kwargs).strip()
+        out = f"{func_header}{self.message!s}{reason}".format(**self.kwargs).strip()
+
+        if sys.version_info < (3, 13) and hasattr(self, "__notes__"):
+            out += "\n" + "\n".join(self.__notes__)
+
+        return out
+
+    if sys.version_info < (3, 13):
+        __notes__: list[str]
+
+        def add_note(self, note: str, /) -> None:
+            if not hasattr(self, "__notes__"):
+                self.__notes__ = []
+
+            self.__notes__.append(note)
+
+    @classmethod
+    def catch(cls) -> CatchError[Self]:
+        """
+        Create a context manager that catches exceptions of this class type.
+
+        Returns:
+            CatchError[Self]: A context manager that will catch and store exceptions of type `cls`
+                when used in a `with` block.
+        """
+        return CatchError(cls)
 
 
 SelfError = TypeVar("SelfError", bound=CustomError)
 
 
+class CatchError(AbstractContextManager["CatchError[SelfError]"], Generic[SelfError]):
+    """
+    Context manager for catching a specific exception type.
+    """
+
+    error: SelfError | None
+    """The caught exception instance, if any."""
+    tb: TracebackType | None
+    """The traceback object associated with the caught exception."""
+
+    def __init__(self, error: type[SelfError]) -> None:
+        self.error = None
+        self.tb = None
+        self._to_catch_error = error
+
+    def __enter__(self) -> Self:
+        return self
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> bool | None:
+        if isinstance(exc_value, self._to_catch_error):
+            self.error = exc_value
+            self.tb = traceback
+            return True
+
+        return None
+
+
 class CustomValueError(CustomError, ValueError):
     """Thrown when a specified value is invalid."""