halib 0.2.20__tar.gz → 0.2.21__tar.gz

{halib-0.2.20 → halib-0.2.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.2.20
+Version: 0.2.21
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -56,7 +56,9 @@ Dynamic: summary
 
 ## v0.2.x (Experiment & Core Updates)
 
-### **v0.2.20**
+### **v0.2.21**
+- ✨ **New Feature:** Added `common.common.pprint_stack_trace` to print stack traces with optional custom messages and force stop capability.
+
 - 🚀 **Improvement:**  `exp.perf.profiler` - allow to export *report dict* as csv files for further analysis
 
 ### **v0.2.19**

{halib-0.2.20 → halib-0.2.21}/README.md

@@ -2,7 +2,9 @@
 
 ## v0.2.x (Experiment & Core Updates)
 
-### **v0.2.20**
+### **v0.2.21**
+- ✨ **New Feature:** Added `common.common.pprint_stack_trace` to print stack traces with optional custom messages and force stop capability.
+
 - 🚀 **Improvement:**  `exp.perf.profiler` - allow to export *report dict* as csv files for further analysis
 
 ### **v0.2.19**

{halib-0.2.20 → halib-0.2.21}/halib/__init__.py

@@ -19,11 +19,11 @@ __all__ = [
     "os",
     "pd",
     "plt",
-    "pprint",
     "pprint_box",
     "pprint_local_path",
+    "pprint_stack_trace",
+    "pprint",
     "px",
-    "pprint_local_path",
     "rcolor_all_str",
     "rcolor_palette_all",
     "rcolor_palette",
@@ -32,10 +32,10 @@ __all__ = [
     "rprint",
     "sns",
     "tcuda",
+    "time",
     "timebudget",
     "tqdm",
     "warnings",
-    "time",
 ]
 import warnings
 
@@ -64,7 +64,8 @@ from .common.common import (
     norm_str,
     pprint_box,
     pprint_local_path,
-    tcuda
+    pprint_stack_trace,
+    tcuda,
 )
 
 # for log
@@ -76,7 +77,12 @@ from timebudget import timebudget
 import omegaconf
 from omegaconf import OmegaConf
 from omegaconf.dictconfig import DictConfig
-from .common.rich_color import rcolor_str, rcolor_palette, rcolor_palette_all, rcolor_all_str
+from .common.rich_color import (
+    rcolor_str,
+    rcolor_palette,
+    rcolor_palette_all,
+    rcolor_all_str,
+)
 
 # for visualization
 import seaborn as sns

{halib-0.2.20 → halib-0.2.21}/halib/common/common.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import re
 import arrow
 import importlib
@@ -10,6 +11,8 @@ from rich.console import Console
 from rich.pretty import pprint, Pretty
 
 from pathlib import Path, PureWindowsPath
+from typing import Optional
+from loguru import logger
 
 
 console = Console()
@@ -114,6 +117,32 @@ def linux_to_wins_path(path: str) -> str:
     return str(PureWindowsPath(win_path))
 
 
+DEFAULT_STACK_TRACE_MSG = "Caused by halib.common.common.pprint_stack_trace"
+
+
+def pprint_stack_trace(
+    msg: str = DEFAULT_STACK_TRACE_MSG,
+    e: Optional[Exception] = None,
+    force_stop: bool = False,
+):
+    """
+    Print the current stack trace or the stack trace of an exception.
+    """
+    try:
+        if e is not None:
+            raise e
+        else:
+            raise Exception("pprint_stack_trace called")
+    except Exception as e:
+        # attach the exception trace to a warning
+        global DEFAULT_STACK_TRACE_MSG
+        if len(msg.strip()) == 0:
+            msg = DEFAULT_STACK_TRACE_MSG
+        logger.opt(exception=e).warning(msg)
+        if force_stop:
+            console.rule("[red]Force Stop Triggered in <halib.common.pprint_stack_trace>[/red]")
+            sys.exit(1)
+
 def pprint_local_path(
     local_path: str, get_wins_path: bool = False, tag: str = ""
 ) -> str:

{halib-0.2.20 → halib-0.2.21}/halib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.2.20
+Version: 0.2.21
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -56,7 +56,9 @@ Dynamic: summary
 
 ## v0.2.x (Experiment & Core Updates)
 
-### **v0.2.20**
+### **v0.2.21**
+- ✨ **New Feature:** Added `common.common.pprint_stack_trace` to print stack traces with optional custom messages and force stop capability.
+
 - 🚀 **Improvement:**  `exp.perf.profiler` - allow to export *report dict* as csv files for further analysis
 
 ### **v0.2.19**

{halib-0.2.20 → halib-0.2.21}/setup.py

@@ -8,7 +8,7 @@ with open("requirements.txt", "r", encoding="utf-8") as f:
 
 setuptools.setup(
     name="halib",
-    version="0.2.20",
+    version="0.2.21",
     author="Hoang Van Ha",
     author_email="hoangvanhauit@gmail.com",
     description="Small library for common tasks",