fix need to reset even for log output or we get interrupted by other (broken) loggers such as pytest

Signed-off-by: Qubitium <qubitium@modelcloud.ai>

Author: Qubitium

logbar/logbar.py

@@ -176,26 +176,26 @@ def _process(self, level: LEVEL, msg, *args, **kwargs):
         columns, _ = terminal_size()
         str_msg = str(msg)
 
+        if columns > 0:
+            str_msg += " " * (columns - LEVEL_MAX_LENGTH - 2 - len(str_msg))  # -2 for cursor + space between LEVEL and msg
+
         global last_pb_instance
         if isinstance(last_pb_instance, ProgressBar) and not last_pb_instance.closed:
-             buf = f'\r'
-             if columns > 0:
-                str_msg += " " * columns
-
-             print(buf,end='',flush=True)
+            print('\r',end='',flush=True)
 
         # Get the color for the log level
 
         reset = COLORS["RESET"]
         color = COLORS.get(level.value, reset)
 
         level_padding = " " * (LEVEL_MAX_LENGTH - len(level.value)) # 5 is max enum string length
-        print(f"{color}{level.value}{reset}{level_padding} {str_msg}", end='\n', flush=True)
+        print(f"\r{color}{level.value}{reset}{level_padding} {str_msg}", end='\n', flush=True)
 
         if isinstance(last_pb_instance, ProgressBar):
             if not last_pb_instance.closed:
                 # only do this for our instance
                 if self == logger:
+                    #print('\r', end='', flush=True)
                     last_pb_instance.draw()
             else:
                 last_pb_instance = None

tests/test_log.py

@@ -2,7 +2,7 @@
 
 from logbar import LogBar
 
-log = LogBar.shared()
+log = LogBar.shared(override_logger=True)
 
 class TestProgressBar(unittest.TestCase):