WIP

Author: z3z1ma

src/dbt_core_interface/dbt_templater/__init__.py

@@ -1,23 +1,26 @@
 """Defines the hook endpoints for the dbt templater plugin."""
+
 import logging
 
 from sqlfluff.core.plugin import hookimpl
 
 from dbt_core_interface.dbt_templater.templater import DCIDbtTemplater
 
-
 LOGGER = logging.getLogger(__name__)
 
 
 @hookimpl
 def get_templaters():
     """Get templaters."""
+
     def create_templater(**kwargs):
         import dbt_core_interface.state
-        assert dbt_core_interface.state.dbt_project_container is not None, "dbt_project_container is None"
+
+        assert dbt_core_interface.state.dbt_project_container is not None, (
+            "dbt_project_container is None"
+        )
         return DCIDbtTemplater(
-            dbt_project_container=dbt_core_interface.state.dbt_project_container,
-            **kwargs
+            dbt_project_container=dbt_core_interface.state.dbt_project_container, **kwargs
         )
 
     create_templater.name = DCIDbtTemplater.name

src/dbt_core_interface/project.py

@@ -9,7 +9,6 @@
 
     dbt.adapters.factory.get_adapter = lambda config: config.adapter  # pyright: ignore[reportUnknownLambdaType]
 
-import argparse
 import contextlib
 import json
 import logging
@@ -75,7 +74,8 @@ class DbtConfiguration:
     profiles_dir: str = DEFAULT_PROFILES_DIR
     target: str | None = None
     threads: int = 1
-    vars: dict[str, t.Any] = {}
+    vars: dict[str, t.Any] = field(default_factory=dict)
+    profile: str | None = None
 
     single_threaded: bool = True
     quiet: bool = True
@@ -84,7 +84,7 @@ class DbtConfiguration:
     partial_parse: bool = False
 
     dependencies: list[str] = field(default_factory=list)
-    which: str = "cupertino"
+    which: str = "zezima was here"
     REQUIRE_RESOURCE_NAMES_WITHOUT_SPACES: bool = field(default_factory=bool)
 
     def __post_init__(self) -> None:
@@ -237,33 +237,14 @@ def _initialize_adapter(self) -> None:
             self._adapter.debug_query()
 
         logger.debug(f"Initialized adapter for {self.project_name}")
-        self.runtime_config.adapter = self.adapter  # pyright: ignore[reportAttributeAccessIssue]
+        self.runtime_config.adapter = self._adapter  # pyright: ignore[reportAttributeAccessIssue]
 
     def _parse_project(self, write_manifest: bool = True) -> None:
         """Parse the dbt project and load manifest."""
-        set_from_args(
-            argparse.Namespace(
-                target=self._base_params.target,
-                profiles_dir=self._base_params.profiles_dir,
-                project_dir=self._base_params.project_dir,
-                threads=self._base_params.threads,
-                vars=self._base_params.vars,
-                quiet=True,
-                single_threaded=True,
-            ),
-            None,
-        )
+        set_from_args(self._base_params, None)  # pyright: ignore[reportArgumentType]
 
         with self._manifest_lock:
-            self._runtime_config = RuntimeConfig.from_args(
-                argparse.Namespace(
-                    target=self._base_params.target,
-                    profiles_dir=self._base_params.profiles_dir,
-                    project_dir=self._base_params.project_dir,
-                    threads=self._base_params.threads,
-                    vars=self._base_params.vars,
-                )
-            )
+            self._runtime_config = RuntimeConfig.from_args(self._base_params)
 
             manifest_loader = ManifestLoader(
                 self._runtime_config,
@@ -313,14 +294,17 @@ def generate_runtime_model_context(self, node: ManifestNode) -> dict[str, t.Any]
             self.manifest,
         )
 
-    def get_ref_node(
+    def ref(
         self,
         model_name: str,
         target_package: str | None = None,
         model_version: int | None = None,
         source_node: ManifestNode | None = None,
     ) -> ManifestNode | None:
-        """Get node by ref (model) name."""
+        """Look up a model node by name, package, and version.
+
+        Akin to using {{ ref() }} in SQL.
+        """
         candidates: list[str | None] = [self.project_name, None]
         if target_package:
             candidates.insert(0, target_package)
@@ -332,8 +316,11 @@ def get_ref_node(
             if node:
                 return node
 
-    def get_source_node(self, source_name: str, table_name: str) -> SourceDefinition | None:
-        """Get source node by source and table name."""
+    def source(self, source_name: str, table_name: str) -> SourceDefinition | None:
+        """Look up a source by name and table name.
+
+        Akin to using {{ source() }} in SQL.
+        """
         return self.manifest.source_lookup.find(f"{source_name}.{table_name}", None, self.manifest)
 
     def execute_sql(self, sql: str) -> ExecutionResult:
@@ -409,7 +396,7 @@ def _create_temp_node(
 
         def _cleanup() -> None:
             with contextlib.suppress(KeyError):
-                del self.manifest.nodes[node_id]
+                del self.manifest.nodes[sql_node.unique_id]
 
         return sql_node, _cleanup