Delay importing locust-plugins to prevent circular import.

locustio · Aug 5, 2023 · 8efc77c · 8efc77c
1 parent 5e67d05
commit 8efc77c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/locust/__init__.py b/locust/__init__.py
@@ -25,12 +25,6 @@
 from .debug import run_single_user
 from .event import Events
 
-try:
-    # import locust_plugins if it is installed, to allow it to register custom arguments etc
-    import locust_plugins  # pyright: ignore[reportMissingImports]
-except ModuleNotFoundError:
-    pass
-
 events = Events()
 
 __all__ = (

diff --git a/locust/main.py b/locust/main.py
@@ -58,6 +58,12 @@ def create_environment(
 
 
 def main():
+    try:
+        # import locust_plugins if it is installed, to allow it to register custom arguments etc
+        import locust_plugins  # pyright: ignore[reportMissingImports]
+    except ModuleNotFoundError:
+        pass
+
     # find specified locustfile(s) and make sure it exists, using a very simplified
     # command line parser that is only used to parse the -f option.
     locustfiles = parse_locustfile_option()