Reset color at end of each line (#34)

intgr · Feb 16, 2023 · 76ce293 · 76ce293
1 parent 576cb84
commit 76ce293
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/topy/topy.py b/topy/topy.py
@@ -139,10 +139,10 @@ def is_color_output_required(stream):
 def add_output_color(line):
     """Adds color to the output of the diff lines."""
     if line.startswith('+'):
-        line = f'\033[1;32m{line}'
+        line = f'\033[1;32m{line}\033[0m'
     elif line.startswith('-'):
-        line = f'\033[1;31m{line}'
-    return f'{line}\033[0m'
+        line = f'\033[1;31m{line}\033[0m'
+    return line
 
 
 def handle_file(regs, filename):