Smaller fixes

christoph2 · Sep 24, 2024 · 029e569 · 029e569
1 parent 6987ef1
commit 029e569
Show file tree

Hide file tree

Showing 13 changed files with 712 additions and 98 deletions.
diff --git a/objutils/__init__.py b/objutils/__init__.py
@@ -18,7 +18,7 @@
 __copyright__ = """
     objutils - Object file library for Python.
 
-   (C) 2010-2020 by Christoph Schueler <cpu12.gems@googlemail.com>
+   (C) 2010-2024 by Christoph Schueler <cpu12.gems@googlemail.com>
 
    All Rights Reserved
 
@@ -37,6 +37,9 @@
   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 """
 
+from rich import pretty
+from rich.console import Console
+from rich.traceback import install as tb_install
 
 import objutils.ash
 import objutils.binfile
@@ -57,6 +60,9 @@
 from objutils.section import Section  # noqa: F401
 
 
+pretty.install()
+
+
 registry.register("bin", objutils.binfile.Reader, objutils.binfile.Writer, "Plain binary format.")
 registry.register(
     "binzip",
@@ -94,6 +100,10 @@
 registry.register("shf", objutils.shf.Reader, objutils.shf.Writer, "S Hexdump Format (rfc4149).")
 
 
+console = Console()
+tb_install(show_locals=True, max_frames=3)  # Install custom exception handler.
+
+
 def load(codec_name, *args, **kws):
     """Load hex data from file.