Add identifier property support for source entity type.

z3z1ma · Oct 25, 2023 · 1a09e44 · 1a09e44
1 parent be30a85
commit 1a09e44
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/dbt_osmosis/core/osmosis.py b/src/dbt_osmosis/core/osmosis.py
@@ -287,6 +287,8 @@ def get_target_schema_path(self, node: ManifestNode) -> Path:
     @staticmethod
     def get_catalog_key(node: ManifestNode) -> CatalogKey:
         """Returns CatalogKey for a given node."""
+        if node.resource_type == NodeType.Source:
+            return CatalogKey(node.database, node.schema, getattr(node, "identifier", node.name))
         return CatalogKey(node.database, node.schema, getattr(node, "alias", node.name))
 
     def get_base_model(self, node: ManifestNode) -> Dict[str, Any]: