Add color to error message

Author: WyattBlue

auto_editor/__main__.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 import sys
+from os import environ
 
 import auto_editor
 from auto_editor.edit import edit_media
@@ -281,9 +282,14 @@ def main() -> None:
         obj.main(sys.argv[2:])
         return
 
+    ff_color = "AV_LOG_FORCE_NOCOLOR"
+    no_color = bool(environ.get("NO_COLOR")) or bool(environ.get(ff_color))
+    log = Log(no_color=no_color)
+
     args = main_options(ArgumentParser("Auto-Editor")).parse_args(
         Args,
         sys.argv[1:],
+        log,
         macros=[
             ({"--frame-margin"}, ["--margin"]),
             ({"--export-to-premiere", "-exp"}, ["--export", "premiere"]),
@@ -311,12 +317,11 @@ def main() -> None:
         print(f"Auto-Editor Version: {auto_editor.version}")
         return
 
-    log = Log(args.debug, args.quiet)
     if not args.input:
         log.error("You need to give auto-editor an input file.")
 
     temp = setup_tempdir(args.temp_dir, log)
-    log = Log(args.debug, args.quiet, temp, args.progress == "machine")
+    log = Log(args.debug, args.quiet, temp, args.progress == "machine", no_color)
     log.debug(f"Temp Directory: {temp}")
 
     ffmpeg = FFmpeg(

auto_editor/utils/log.py

@@ -8,19 +8,21 @@
 
 
 class Log:
-    __slots__ = ("is_debug", "quiet", "temp", "machine", "start_time")
+    __slots__ = ("is_debug", "quiet", "temp", "machine", "start_time", "no_color")
 
     def __init__(
         self,
         is_debug: bool = False,
         quiet: bool = False,
         temp: str | None = None,
         machine: bool = False,
+        no_color: bool = True,
     ):
         self.is_debug = is_debug
         self.quiet = quiet
         self.temp = temp
         self.machine = machine
+        self.no_color = no_color
         self.start_time = 0 if self.quiet or self.machine else perf_counter()
 
     def debug(self, message: object) -> None:
@@ -74,7 +76,11 @@ def error(self, message: str | Exception) -> NoReturn:
             raise message
 
         self.conwrite("")
-        sys.stderr.write(f"Error! {message}\n")
+        if self.no_color:
+            sys.stderr.write(f"Error! {message}\n")
+        else:
+            sys.stderr.write(f"\033[31;40mError! {message}\033[0m\n")
+
         self.cleanup()
         from platform import system
 

auto_editor/vanparse.py

@@ -160,25 +160,6 @@ def get_option(name: str, options: list[Options]) -> Options | None:
     return None
 
 
-def parse_value(option: Options | Required, val: str | None) -> Any:
-    if val is None and option.nargs == 1:
-        Log().error(f"{option.names[0]} needs argument.")
-
-    try:
-        value = option.type(val)
-    except CoerceError as e:
-        Log().error(e)
-
-    if option.choices is not None and value not in option.choices:
-        choices = ", ".join(option.choices)
-
-        Log().error(
-            f"{value} is not a choice for {option.names[0]}\nchoices are:\n  {choices}"
-        )
-
-    return value
-
-
 class ArgumentParser:
     def __init__(self, program_name: str | None):
         self.program_name = program_name
@@ -201,6 +182,7 @@ def parse_args(
         self,
         ns_obj: type[T],
         sys_args: list[str],
+        log_: Log | None = None,
         macros: list[tuple[set[str], list[str]]] | None = None,
     ) -> T:
         if not sys_args and self.program_name is not None:
@@ -219,9 +201,26 @@ def parse_args(
             del _macros
         del macros
 
+        log = Log() if log_ is None else log_
         ns = ns_obj()
         option_names: list[str] = []
 
+        def parse_value(option: Options | Required, val: str | None) -> Any:
+            if val is None and option.nargs == 1:
+                log.error(f"{option.names[0]} needs argument.")
+
+            try:
+                value = option.type(val)
+            except CoerceError as e:
+                log.error(e)
+
+            if option.choices is not None and value not in option.choices:
+                log.error(
+                    f"{value} is not a choice for {option.names[0]}\n"
+                    f"choices are:\n  {', '.join(option.choices)}"
+                )
+            return value
+
         program_name = self.program_name
         requireds = self.requireds
         options = self.options
@@ -256,7 +255,7 @@ def parse_args(
                         val = oplist_coerce(arg)
                         ns.__setattr__(oplist_name, getattr(ns, oplist_name) + [val])
                     except (CoerceError, ValueError) as e:
-                        Log().error(e)
+                        log.error(e)
                 elif requireds and not arg.startswith("--"):
                     if requireds[0].nargs == 1:
                         ns.__setattr__(req_list_name, parse_value(requireds[0], arg))
@@ -268,19 +267,19 @@ def parse_args(
 
                     # 'Did you mean' message might appear that options need a comma.
                     if arg.replace(",", "") in option_names:
-                        Log().error(f"Option '{arg}' has an unnecessary comma.")
+                        log.error(f"Option '{arg}' has an unnecessary comma.")
 
                     close_matches = difflib.get_close_matches(arg, option_names)
                     if close_matches:
-                        Log().error(
+                        log.error(
                             f"Unknown {label}: {arg}\n\n    Did you mean:\n        "
                             + ", ".join(close_matches)
                         )
-                    Log().error(f"Unknown {label}: {arg}")
+                    log.error(f"Unknown {label}: {arg}")
             else:
                 if option.nargs != "*":
                     if option in used_options:
-                        Log().error(
+                        log.error(
                             f"Option {option.names[0]} may not be used more than once."
                         )
                     used_options.append(option)