Enable multi project references and settings

memiiso · Dec 16, 2024 · 92d345f · 92d345f
1 parent 2d09443
commit 92d345f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/opendbt/__init__.py b/opendbt/__init__.py
@@ -3,13 +3,6 @@
 import sys
 from pathlib import Path
 
-######################
-from opendbt.dbt import patch_dbt
-
-patch_dbt()
-from opendbt.utils import Utils
-######################
-
 from dbt.cli.main import dbtRunner as DbtCliRunner
 from dbt.cli.main import dbtRunnerResult
 from dbt.config import PartialProject
@@ -18,6 +11,14 @@
 from dbt.exceptions import DbtRuntimeError
 from dbt.task.base import get_nearest_project_dir
 
+######################
+from opendbt.dbt import patch_dbt
+# patch_dbt()
+from opendbt.utils import Utils
+
+
+######################
+
 
 class OpenDbtLogger:
     _log = None
@@ -106,6 +107,7 @@ def run(args: list, callbacks: list = None) -> dbtRunnerResult:
         :return: The result of the dbt run.
         """
         callbacks = callbacks if callbacks else []
+        patch_dbt()
         # https://docs.getdbt.com/reference/programmatic-invocations
         dbtcr = DbtCliRunner(callbacks=callbacks)
         result: dbtRunnerResult = dbtcr.invoke(args)

diff --git a/opendbt/dbt/v17/config/runtime.py b/opendbt/dbt/v17/config/runtime.py
@@ -41,7 +41,7 @@ def load_dependencies(self, base_only=False) -> Mapping[str, "RuntimeConfig"]:
 
         if self.dependencies is None:
             # this sets self.dependencies variable!
-            self.dependencies = super().load_dependencies()
+            self.dependencies = super().load_dependencies(base_only=base_only)
 
             # additionally load `projects` defined in `dependencies.yml`
             for project_name, project in self.load_dependence_projects():

diff --git a/opendbt/dbt/v18/config/runtime.py b/opendbt/dbt/v18/config/runtime.py
@@ -41,7 +41,7 @@ def load_dependencies(self, base_only=False) -> Mapping[str, "RuntimeConfig"]:
 
         if self.dependencies is None:
             # this sets self.dependencies variable!
-            self.dependencies = super().load_dependencies()
+            self.dependencies = super().load_dependencies(base_only=base_only)
 
             # additionally load `projects` defined in `dependencies.yml`
             for project_name, project in self.load_dependence_projects():