Merge pull request #27 from cocolato/dev

merge dev
cocolato · Mar 18, 2024 · 15b561b · 15b561b
2 parents 40f0e1c + 942ac3b
commit 15b561b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/pydumpling/helpers.py b/pydumpling/helpers.py
@@ -1,4 +1,6 @@
+import sys
 from traceback import print_tb, print_exception
+from .pydumpling import save_dumping
 
 
 def print_traceback_and_except(dumpling_result):
@@ -10,3 +12,13 @@ def print_traceback_and_except(dumpling_result):
         print_exception(exc_type, exc_value, exc_tb)
     else:
         print_tb(exc_tb)
+
+
+def catch_any_exception():
+    original_hook = sys.excepthook
+
+    def _hook(exc_type, exc_value, exc_tb):
+        save_dumping(exc_info=(exc_type, exc_value, exc_tb))
+        original_hook(exc_type, exc_value, exc_tb)  # call sys original hook
+
+    sys.excepthook = _hook
diff --git a/pydumpling/pydumpling.py b/pydumpling/pydumpling.py
@@ -11,13 +11,15 @@
 __version__ = "0.1.4"
 
 
-def save_dumping(filename=None, tb=None):
+def save_dumping(filename=None, exc_info=None):
     try:
-        if tb is None:
+        if exc_info is None:
             exc_type, exc_value, exc_tb = sys.exc_info()
+        else:
+            exc_type, exc_value, exc_tb = exc_info
 
         if filename is None:
-            filename = "%s:%d.dump" % (
+            filename = "%s-%d.dump" % (
                 exc_tb.tb_frame.f_code.co_filename, exc_tb.tb_frame.f_lineno)
 
         fake_tb = FakeTraceback(exc_tb)