REFACT: don't import CONFIG UpFront ...

+ dropped from 17ms --> 3m `import composition`

Author: ankostis

graphtik/base.py

@@ -99,6 +99,13 @@ def __repr__(self):
 UNSET = Token("UNSET")
 
 
+def first_solid(*tristates, default=None):
+    """Utility combining multiple tri-state booleans."""
+    from boltons.iterutils import first
+
+    return first(tristates, default=default,    key=lambda i: i is not None)
+
+
 def aslist(i, argname, allowed_types=list):
     """Utility to accept singular strings as lists, and None --> []."""
     if not i:

graphtik/composition.py

@@ -11,6 +11,7 @@
 import abc
 import itertools as itt
 import logging
+import re
 import textwrap
 from collections import abc as cabc
 from functools import wraps
@@ -27,18 +28,10 @@
     Token,
     aslist,
     astuple,
+    first_solid,
     func_name,
     jetsam,
 )
-from .config import (
-    first_solid,
-    is_debug,
-    is_endure_operations,
-    is_marshal_tasks,
-    is_parallel_tasks,
-    is_reschedule_operations,
-    reset_abort,
-)
 from .modifiers import (
     dep_singularized,
     dep_stripped,
@@ -352,6 +345,15 @@ def __repr__(self):
         """
         Display more informative names for the Operation class
         """
+        from .config import (
+            is_debug,
+            is_endure_operations,
+            is_marshal_tasks,
+            is_parallel_tasks,
+            is_reschedule_operations,
+            reset_abort,
+        )
+
         needs = aslist(self.needs, "needs")
         provides = aslist(self.provides, "provides")
         aliases = aslist(self.aliases, "aliases")
@@ -386,6 +388,7 @@ def deps(self) -> Mapping[str, Collection]:
 
         if not DEBUG, all deps are converted into lists, ready to be printed.
         """
+        from .config import is_debug
 
         return {
             k: v if is_debug() else list(v)
@@ -423,6 +426,8 @@ def _prepare_match_inputs_error(
         varargs_bad: List,
         named_inputs: Mapping,
     ) -> ValueError:
+        from .config import is_debug
+
         errors = [
             f"Need({n}) failed due to: {type(nex).__name__}({nex})"
             for n, nex in enumerate(exceptions, 1)
@@ -498,6 +503,8 @@ def _match_inputs_with_fn_needs(self, named_inputs) -> Tuple[list, list, dict]:
 
     def _zip_results_with_provides(self, results) -> dict:
         """Zip results with expected "real" (without sideffects) `provides`."""
+        from .config import is_reschedule_operations
+
         fn_expected: iset = self._fn_provides
         rescheduled = first_solid(is_reschedule_operations(), self.rescheduled)
         if self.returns_dict:
@@ -936,6 +943,7 @@ def __repr__(self):
         """
         Display more informative names for the Operation class
         """
+        from .config import is_debug
         from .network import yield_ops
 
         clsname = type(self).__name__
@@ -1142,6 +1150,8 @@ def compute(
 
         See also :meth:`.Operation.compute()`.
         """
+        from .config import reset_abort
+
         try:
             if named_inputs is UNSET:
                 named_inputs = {}

graphtik/config.py

@@ -260,8 +260,3 @@ def get_execution_pool() -> "Optional[Pool]":
     a "reset" token (see :meth:`.ContextVar.set`)
 
 ."""
-
-
-def first_solid(*tristates, default=None):
-    """Utility combining multiple tri-state booleans."""
-    return first(tristates, default=default, key=lambda i: i is not None)

graphtik/execution.py

@@ -21,10 +21,11 @@
     Plottable,
     aslist,
     astuple,
+    first_solid,
     jetsam,
 )
+from .composition import Operation
 from .config import (
-    first_solid,
     get_execution_pool,
     is_abort,
     is_debug,
@@ -41,7 +42,6 @@
     yield_node_names,
     yield_ops,
 )
-from .composition import Operation
 
 #: If this logger is *eventually* DEBUG-enabled,
 #: the string-representation of network-objects (network, plan, solution)

graphtik/plot.py

@@ -42,17 +42,16 @@
 import pydot
 from boltons.iterutils import default_enter, default_exit, get_path, remap
 
-from .base import PlotArgs, func_name, func_source
+from .base import PlotArgs, first_solid, func_name, func_source
+from .composition import Operation
 from .config import (
-    first_solid,
     is_debug,
     is_endure_operations,
     is_marshal_tasks,
     is_parallel_tasks,
     is_reschedule_operations,
 )
 from .modifiers import is_mapped, is_sfx, is_sfxed
-from .composition import Operation
 
 log = logging.getLogger(__name__)
 

test/conftest.py

@@ -1,7 +1,5 @@
 import pytest
 
-from graphtik import compose, operation
-
 # Enable pytest-sphinx fixtures
 # See https://www.sphinx-doc.org/en/master/devguide.html#unit-testing
 pytest_plugins = "sphinx.testing.fixtures"
@@ -13,7 +11,7 @@
 
 @pytest.fixture
 def debug_mode():
-    from graphtik import debug_enabled
+    from graphtik.config import debug_enabled
 
     with debug_enabled(True):
         yield