From 7cd9858761f985f1c1db2c4d71e0a1ab6a142682 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Anders=20Fredrik=20Ki=C3=A6r?=
 <31612826+anders-kiaer@users.noreply.github.com>
Date: Wed, 23 Nov 2022 14:26:16 +0100
Subject: [PATCH] Lazy load modules in command line entrypoint (#653)

---
 CHANGELOG.md                  |  1 +
 webviz_config/command_line.py | 41 ++++++++++++++++++++++++++---------
 2 files changed, 32 insertions(+), 10 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 1a61087d..9be4dbed 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Changed
 - [#648](https://github.com/equinor/webviz-config/pull/648) - Allow `blob:` in `script-src` CSP in order to enable web worker usage in Dash components.
 - [#652](https://github.com/equinor/webviz-config/pull/652) - Enabled support for LaTeX math/equations in markdown.
+- [#653](https://github.com/equinor/webviz-config/pull/653) - Reduce time for running `webviz --help` by lazy importing top level entrypoints.
 
 ### Added
 - [#644](https://github.com/equinor/webviz-config/pull/644) - Added option to download tables in `DataTable` and `PivotTable`.
diff --git a/webviz_config/command_line.py b/webviz_config/command_line.py
index ccada925..b39862df 100644
--- a/webviz_config/command_line.py
+++ b/webviz_config/command_line.py
@@ -6,15 +6,11 @@
 import pathlib
 import subprocess
 
-from ._build_webviz import build_webviz
-from ._deployment import main_radix_deployment
-from ._docs.open_docs import open_docs
-from ._docs._create_schema import create_schema
 from ._user_data_dir import user_data_dir
 from ._user_preferences import set_user_preferences, get_user_preference
 
 
-def main() -> None:  # pylint: disable=too-many-statements
+def main() -> None:  # pylint: disable=too-many-locals,too-many-statements
 
     parser = argparse.ArgumentParser(
         prog=("Creates a Webviz Dash app from a configuration setup")
@@ -76,7 +72,14 @@ def main() -> None:  # pylint: disable=too-many-statements
         help="Path to YAML file with logging configuration.",
     )
 
-    parser_build.set_defaults(func=build_webviz)
+    def parser_build_function(args: argparse.Namespace) -> None:
+        from ._build_webviz import (  # pylint: disable=import-outside-toplevel
+            build_webviz,
+        )
+
+        build_webviz(args)
+
+    parser_build.set_defaults(func=parser_build_function)
 
     # Add "certificate" parser:
 
@@ -141,7 +144,14 @@ def main() -> None:  # pylint: disable=too-many-statements
         "Do not include this if only overwriting an existing application.",
     )
 
-    parser_radix_deploy.set_defaults(func=main_radix_deployment)
+    def parser_radix_deploy_function(args: argparse.Namespace) -> None:
+        from ._deployment import (  # pylint: disable=import-outside-toplevel
+            main_radix_deployment,
+        )
+
+        main_radix_deployment(args)
+
+    parser_radix_deploy.set_defaults(func=parser_radix_deploy_function)
 
     # Add "documentation" parser:
 
@@ -172,7 +182,14 @@ def main() -> None:  # pylint: disable=too-many-statements
         help="Skip opening the documentation automatically in browser.",
     )
 
-    parser_docs.set_defaults(func=open_docs)
+    def parser_docs_function(args: argparse.Namespace) -> None:
+        from ._docs.open_docs import (  # pylint: disable=import-outside-toplevel
+            open_docs,
+        )
+
+        open_docs(args)
+
+    parser_docs.set_defaults(func=parser_docs_function)
 
     # Add "preferences" parser:
 
@@ -223,11 +240,15 @@ def entrypoint_preferences(args: argparse.Namespace) -> None:
         "it will be stored in your Webviz application settings folder.",
     )
 
-    def entrypoint_schema(args: argparse.Namespace) -> None:
+    def parser_schema_function(args: argparse.Namespace) -> None:
+        from ._docs._create_schema import (  # pylint: disable=import-outside-toplevel
+            create_schema,
+        )
+
         args.output.write_text(json.dumps(create_schema(), indent=4))
         print(f"Schema written to {args.output}")
 
-    parser_schema.set_defaults(func=entrypoint_schema)
+    parser_schema.set_defaults(func=parser_schema_function)
 
     # Add "editor" parser: