Fix prober, color config

src/tools/interop/idt/capture/config.py

@@ -19,3 +19,4 @@
 
 log_level = logging.INFO
 async_timeout = 45.0
+enable_color = True

src/tools/interop/idt/capture/ecosystem/play_services/prober.py

@@ -18,14 +18,16 @@
 import os
 
 from capture.utils.shell import Bash
+from capture.utils import log
+logger = log.get_logger(__file__)
 
 
 class PlayServicesProber:
 
     def __init__(self, platform):
         self.platform = platform
         self.artifact_dir = self.platform.artifact_dir
-        self.logger = log_format.get_logger(__file__)
+        self.logger = logger
 
     async def _probe_foyer(self) -> None:
         self.logger.info("probing remote services")

src/tools/interop/idt/capture/utils/log.py

@@ -18,7 +18,7 @@
 import logging
 from typing import TextIO
 
-from capture import config
+from .. import config
 
 _CONFIG_LEVEL = config.log_level
 
@@ -33,8 +33,10 @@
 
 reset = "\x1b[0m"
 
-format_pre = cyan + "%(asctime)s %(levelname)s {%(module)s} [%(funcName)s] " + reset
+format_pre = "%(asctime)s %(levelname)s {%(module)s} [%(funcName)s] " + reset
+format_pre = cyan + format_pre if config.enable_color else format_pre
 format_post = "%(message)s" + reset
+format_no_color = format_pre+format_post
 
 FORMATS = {
     logging.DEBUG: format_pre + blue + format_post,
@@ -48,7 +50,7 @@
 class LoggingFormatter(logging.Formatter):
 
     def format(self, record):
-        log_fmt = FORMATS.get(record.levelno)
+        log_fmt = FORMATS.get(record.levelno) if config.enable_color else format_no_color
         formatter = logging.Formatter(log_fmt)
         return formatter.format(record)
 
@@ -68,11 +70,19 @@ def border_print(to_print: str, important: bool = False) -> None:
     len_borders = len(to_print)
     border = f"\n{'_' * len_borders}\n"
     i_border = f"\n{'!' * len_borders}\n" if important else ""
-    print(f"\x1b[35;1m{border}{i_border}{to_print}{i_border}{border}\x1b[0m")
+    if config.enable_color:
+        print("\x1b[35;1m")
+    print(f"{border}{i_border}{to_print}{i_border}{border}")
+    if config.enable_color:
+        print("\x1b[0m")
 
 
 def print_and_write(to_print: str, file: TextIO) -> None:
-    print(f"\x1b[32;1m{to_print}\x1b[0m")
+    if config.enable_color:
+        print("\x1b[32;1m")
+    print(f"{to_print}")
+    if config.enable_color:
+        print("\x1b[0m")
     file.write(to_print)