excs

akatrevorjay · Dec 10, 2018 · 61178b9 · 61178b9
1 parent 21080a8
commit 61178b9
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/pytutils/excs.py b/pytutils/excs.py
@@ -0,0 +1,16 @@
+from contextlib import contextmanager
+
+
+@contextmanager
+def ok(*exceptions):
+    """Context manager to pass exceptions.
+    :param exceptions: Exceptions to pass
+    """
+    try:
+        yield
+    except Exception as e:
+        if isinstance(e, exceptions):
+            pass
+        else:
+            raise e
+