refactor:: only import when needed

kraanzu · Mar 26, 2024 · 7d45130 · 7d45130
1 parent 052d971
commit 7d45130
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/smassh/__init__.py b/smassh/__init__.py
@@ -1,6 +1,4 @@
 import click
-from smassh.src.plugins.add_language import AddLanguage
-from smassh.ui.tui import Smassh
 
 PKG_VERSION = "3.1.3"
 
@@ -21,12 +19,16 @@ def main(ctx, version: bool) -> None:
         return print(f"smassh - v{PKG_VERSION}")
 
     if ctx.invoked_subcommand is None:
+        from smassh.ui.tui import Smassh
+
         Smassh().run()
 
 
 @main.command(help="Add a language to smassh")
 @click.argument("name")
 def add(name: str) -> None:
+    from smassh.src.plugins.add_language import AddLanguage
+
     AddLanguage().add(name)
 
 

diff --git a/smassh/__main__.py b/smassh/__main__.py
@@ -1,6 +1,4 @@
 import click
-from smassh.src.plugins.add_language import AddLanguage
-from smassh.ui.tui import Smassh
 
 PKG_VERSION = "3.1.3"
 
@@ -21,12 +19,16 @@ def main(ctx, version: bool) -> None:
         return print(f"smassh - v{PKG_VERSION}")
 
     if ctx.invoked_subcommand is None:
+        from smassh.ui.tui import Smassh
+
         Smassh().run()
 
 
 @main.command(help="Add a language to smassh")
 @click.argument("name")
 def add(name: str) -> None:
+    from smassh.src.plugins.add_language import AddLanguage
+
     AddLanguage().add(name)