diff options
| -rw-r--r-- | gemato/cli.py | 19 |
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)) |