Support pretty log formatting using rich

Signed-off-by: Michał Górny <mgorny@gentoo.org>

1 files changed, 16 insertions, 3 deletions

diff --git a/gemato/cli.py b/gemato/cli.py
index 4df8ee7..382e961 100644
--- a/gemato/cli.py
+++ b/gemato/cli.py
@@ -659,8 +659,21 @@ def main(argv):
 
 
 def setuptools_main():
+    try:
+        from rich.logging import RichHandler
+    except ImportError:
+        format_args = {
+            "format": "[{levelname:>7}] {message}",
+            "style": "{",
+        }
+    else:
+        format_args = {
+            "format": "{message}",
+            "style": "{",
+            "handlers": [RichHandler(show_time=False, show_path=False)],
+        }
+
     logging.basicConfig(
-        format="[{levelname:>7}] {message}",
-        style="{",
-        level=logging.INFO)
+        level=logging.INFO,
+        **format_args)
     sys.exit(main(sys.argv))