un-optimize cli file; it's useless with the barrel __init__

azuline · Apr 16, 2024 · f4c5ba3 · f4c5ba3
1 parent 6b0ec50
commit f4c5ba3
Show file tree

Hide file tree

Showing 4 changed files with 66 additions and 95 deletions.
diff --git a/rose_cli/__main__.py b/rose_cli/__main__.py
@@ -2,14 +2,15 @@
 
 import click
 
-from rose_cli.cli import cli
-from rose_lib import RoseExpectedError
+from rose_cli.cli import CliExpectedError, cli
 
 
 def main() -> None:
+    from rose_lib import RoseExpectedError
+
     try:
         cli()
-    except RoseExpectedError as e:
+    except (RoseExpectedError, CliExpectedError) as e:
         click.secho(f"{e.__class__.__module__}.{e.__class__.__name__}: ", fg="red", nl=False)
         click.secho(str(e))
         sys.exit(1)