From 5f48d6ce06ea0729577b1f4405a0759bfc9cec3b Mon Sep 17 00:00:00 2001 From: z3z1ma Date: Mon, 3 Feb 2025 09:19:23 -0700 Subject: [PATCH] chore: run task format --- src/dbt_osmosis/core/osmosis.py | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/src/dbt_osmosis/core/osmosis.py b/src/dbt_osmosis/core/osmosis.py index 400dd1d..b614094 100644 --- a/src/dbt_osmosis/core/osmosis.py +++ b/src/dbt_osmosis/core/osmosis.py @@ -52,8 +52,8 @@ ) from dbt.mp_context import get_mp_context from dbt.node_types import NodeType -from dbt.parser.models import ModelParser from dbt.parser.manifest import ManifestLoader, process_node +from dbt.parser.models import ModelParser from dbt.parser.sql import SqlBlockParser, SqlMacroParser from dbt.task.docs.generate import Catalog from dbt.task.sql import SqlCompileRunner @@ -251,6 +251,7 @@ def manifest_mutex(self) -> threading.Lock: """Return the manifest mutex for thread safety.""" return self._manifest_mutex + def _add_cross_project_references(manifest, dbt_loom, project_name): """Add cross-project references to the dbt manifest from dbt-loom defined manifests.""" loomnodes = [] @@ -263,12 +264,14 @@ def _add_cross_project_references(manifest, dbt_loom, project_name): for _, node in loom_manifest_nodes.items(): if node.get("access"): node_access = node.get("access") - if node_access!="protected": - if node.get("resource_type")=="model": + if node_access != "protected": + if node.get("resource_type") == "model": loomnodes.append(ModelParser.parse_from_dict(None, node)) for node in loomnodes: manifest.nodes[node.unique_id] = node - logger.info(f":arrows_counterclockwise: added {len(loomnodes)} exposed nodes from {name} to the dbt manifest!") + logger.info( + f":arrows_counterclockwise: added {len(loomnodes)} exposed nodes from {name} to the dbt manifest!" + ) return manifest @@ -317,7 +320,6 @@ def create_dbt_project_context(config: DbtConfiguration) -> DbtProjectContext: setattr(runtime_cfg, "adapter", adapter) adapter.set_macro_resolver(manifest) - sql_parser = SqlBlockParser(runtime_cfg, manifest, runtime_cfg) macro_parser = SqlMacroParser(runtime_cfg, manifest) @@ -790,9 +792,12 @@ def f(node: ResultNode, include_external: bool = False) -> bool: if node.resource_type == NodeType.Model and node.config.materialized == "ephemeral": return False if context.settings.models: - if not _is_file_match( - node, context.settings.models, context.project.runtime_cfg.project_root - ) and not include_external: + if ( + not _is_file_match( + node, context.settings.models, context.project.runtime_cfg.project_root + ) + and not include_external + ): return False if context.settings.fqn: if not _is_fqn_match(node, context.settings.fqn):