Use coloredlogs when the module is available

Author: bitoffdev

kasatk/__main__.py

@@ -386,12 +386,17 @@ def main():
     )
     args = parser.parse_args()
     log_level = logging.CRITICAL - args.verbosity * 10
-    formatter = logging.Formatter(
-        "[%(asctime)s] p%(process)s {%(pathname)s:%(lineno)d} %(levelname)s - %(message)s",
-        "%m-%d %H:%M:%S",
-    )
-    logging.basicConfig(level=log_level)
-    root = logging.getLogger()
-    hdlr = root.handlers[0]
-    hdlr.setFormatter(formatter)
+    try:
+        import coloredlogs
+
+        coloredlogs.install(level=log_level)
+    except ImportError:
+        formatter = logging.Formatter(
+            "[%(asctime)s] p%(process)s {%(pathname)s:%(lineno)d} %(levelname)s - %(message)s",
+            "%m-%d %H:%M:%S",
+        )
+        logging.basicConfig(level=log_level)
+        root = logging.getLogger()
+        hdlr = root.handlers[0]
+        hdlr.setFormatter(formatter)
     main()