fix: properly load dbt relation type for get_relation()

related functions

fixes #1987

Author: newtonapple

sqlmesh/dbt/adapter.py

@@ -18,6 +18,7 @@
     from dbt.adapters.base import BaseRelation
     from dbt.adapters.base.column import Column
     from dbt.adapters.base.impl import AdapterResponse
+    from sqlmesh.core.engine_adapter.base import DataObject
     from sqlmesh.dbt.relation import Policy
 
 
@@ -259,6 +260,10 @@ def load_relation(self, relation: BaseRelation) -> t.Optional[BaseRelation]:
         if not self.engine_adapter.table_exists(mapped_table):
             return None
 
+        data_object = self.engine_adapter.get_data_object(mapped_table)
+        if data_object is not None:
+            return self._data_object_to_relation(data_object)
+
         return self._table_to_relation(mapped_table)
 
     def list_relations(self, database: t.Optional[str], schema: str) -> t.List[BaseRelation]:
@@ -269,24 +274,10 @@ def list_relations(self, database: t.Optional[str], schema: str) -> t.List[BaseR
         return self.list_relations_without_caching(self._table_to_relation(target_schema))
 
     def list_relations_without_caching(self, schema_relation: BaseRelation) -> t.List[BaseRelation]:
-        from sqlmesh.dbt.relation import RelationType
-
         schema = self._normalize(self._schema(schema_relation))
 
         relations = [
-            self.relation_type.create(
-                database=do.catalog,
-                schema=do.schema_name,
-                identifier=do.name,
-                quote_policy=self.quote_policy,
-                # DBT relation types aren't snake case and instead just one word without spaces so we remove underscores
-                type=(
-                    RelationType.External
-                    if do.type.is_unknown
-                    else RelationType(do.type.lower().replace("_", ""))
-                ),
-            )
-            for do in self.engine_adapter.get_data_objects(schema)
+            self._data_object_to_relation(do) for do in self.engine_adapter.get_data_objects(schema)
         ]
         return relations
 
@@ -381,6 +372,22 @@ def _map_table_name(self, table: exp.Table) -> exp.Table:
     def _relation_to_table(self, relation: BaseRelation) -> exp.Table:
         return exp.to_table(relation.render(), dialect=self.project_dialect)
 
+    def _data_object_to_relation(self, data_object: DataObject) -> BaseRelation:
+        from sqlmesh.dbt.relation import RelationType
+
+        return self.relation_type.create(
+            database=data_object.catalog,
+            schema=data_object.schema_name,
+            identifier=data_object.name,
+            quote_policy=self.quote_policy,
+            # DBT relation types aren't snake case and instead just one word without spaces so we remove underscores
+            type=(
+                RelationType.External
+                if data_object.type.is_unknown
+                else RelationType(data_object.type.lower().replace("_", ""))
+            ),
+        )
+
     def _table_to_relation(self, table: exp.Table) -> BaseRelation:
         return self.relation_type.create(
             database=table.catalog or None,

tests/dbt/test_adapter.py

@@ -38,6 +38,9 @@ def test_adapter_relation(sushi_test_project: Project, runtime_renderer: t.Calla
     engine_adapter.create_table(
         table_name="foo.another", columns_to_types={"col": exp.DataType.build("int")}
     )
+    engine_adapter.create_view(
+        view_name="foo.bar_view", query_or_df=parse_one("select * from foo.bar")
+    )
     engine_adapter.create_table(
         table_name="ignored.ignore", columns_to_types={"col": exp.DataType.build("int")}
     )
@@ -46,11 +49,24 @@ def test_adapter_relation(sushi_test_project: Project, runtime_renderer: t.Calla
         renderer("{{ adapter.get_relation(database=None, schema='foo', identifier='bar') }}")
         == '"memory"."foo"."bar"'
     )
+
+    assert (
+        renderer("{{ adapter.get_relation(database=None, schema='foo', identifier='bar').type }}")
+        == "table"
+    )
+
+    assert (
+        renderer(
+            "{{ adapter.get_relation(database=None, schema='foo', identifier='bar_view').type }}"
+        )
+        == "view"
+    )
+
     assert renderer(
         "{%- set relation = adapter.get_relation(database=None, schema='foo', identifier='bar') -%} {{ adapter.get_columns_in_relation(relation) }}"
     ) == str([Column.from_description(name="baz", raw_data_type="INT")])
 
-    assert renderer("{{ adapter.list_relations(database=None, schema='foo')|length }}") == "2"
+    assert renderer("{{ adapter.list_relations(database=None, schema='foo')|length }}") == "3"
 
     assert renderer(
         """