chore(iast): fix import chains

Author: avara1986

ddtrace/appsec/__init__.py

@@ -4,7 +4,6 @@
 
 
 _APPSEC_TO_BE_LOADED = True
-_IAST_TO_BE_LOADED = True
 
 
 def load_appsec():
@@ -17,16 +16,6 @@ def load_appsec():
         _APPSEC_TO_BE_LOADED = False
 
 
-def load_iast():
-    """Lazily load the iast module listeners."""
-    from ddtrace.appsec._iast._listener import iast_listen
-
-    global _IAST_TO_BE_LOADED
-    if _IAST_TO_BE_LOADED:
-        iast_listen()
-        _IAST_TO_BE_LOADED = False
-
-
 def load_common_appsec_modules():
     """Lazily load the common module patches."""
     from ddtrace.settings.asm import config as asm_config

ddtrace/appsec/_iast/__init__.py

@@ -37,11 +37,15 @@ def wrapped_function(wrapped, instance, args, kwargs):
 from ddtrace.internal.module import ModuleWatchdog
 from ddtrace.settings.asm import config as asm_config
 
+from ._listener import iast_listen
 from ._overhead_control_engine import OverheadControl
 
 
 log = get_logger(__name__)
 
+_IAST_TO_BE_LOADED = True
+_iast_propagation_enabled = False
+
 oce = OverheadControl()
 
 
@@ -79,9 +83,6 @@ def ddtrace_iast_flask_patch():
     exec(compiled_code, module.__dict__)  # nosec B102
 
 
-_iast_propagation_enabled = False
-
-
 def enable_iast_propagation():
     """Add IAST AST patching in the ModuleWatchdog"""
     # DEV: These imports are here to avoid _ast.ast_patching import in the top level
@@ -141,3 +142,11 @@ def disable_iast_propagation():
     "enable_iast_propagation",
     "disable_iast_propagation",
 ]
+
+
+def load_iast():
+    """Lazily load the iast module listeners."""
+    global _IAST_TO_BE_LOADED
+    if _IAST_TO_BE_LOADED:
+        iast_listen()
+        _IAST_TO_BE_LOADED = False

ddtrace/appsec/_iast/processor.py

@@ -15,7 +15,7 @@
 @dataclass(eq=False)
 class AppSecIastSpanProcessor(SpanProcessor):
     def __post_init__(self) -> None:
-        from ddtrace.appsec import load_iast
+        from . import load_iast
 
         load_iast()