testpython.git

@@ -9,6 +9,7 @@ import sitesize.sitesize

 from typing import Callable

 from decorators import timeit, profile

 from pathlib import Path

+import traceback

 try:

     import marshmallow

     marshmallow_imported = True

@@ -35,6 +36,19 @@ def set_v_print(verbose: bool) -> None:

     v_print = print if verbose else lambda *a, **k: None

+def filenum_excepthook(exc_type, exc_value, tb):

+    """Print errors as:

+         File path/file.py:##

+       not:

+         File path/file.py, line ##"""

+    for line in traceback.format_exception(exc_type, exc_value, tb):

+        if line.startswith('  File'):

+            print(line.replace(', line ', ':').rstrip(), file=sys.stderr)

+        else:

+            print(line, end='', file=sys.stderr)

+

+sys.excepthook = filenum_excepthook

+

 # This class demonstrates decorators.

 #

 # Consider these alternatives for data classes: