dbt-adapters 1.16.5__py3-none-any.whl → 1.16.6__py3-none-any.whl

dbt/adapters/__about__.py

@@ -1 +1 @@
-version = "1.16.5"
+version = "1.16.6"

dbt/adapters/base/impl.py

@@ -30,6 +30,7 @@ from dbt.adapters.record.base import (
     AdapterGetPartitionsMetadataRecord,
     AdapterConvertTypeRecord,
     AdapterStandardizeGrantsDictRecord,
+    AdapterListRelationsWithoutCachingRecord,
 )
 from dbt_common.behavior_flags import Behavior, BehaviorFlag
 from dbt_common.clients.jinja import CallableMacroGenerator
@@ -772,6 +773,12 @@ class BaseAdapter(metaclass=AdapterMeta):
             "`expand_target_column_types` is not implemented for this adapter!"
         )
 
+    @record_function(
+        AdapterListRelationsWithoutCachingRecord,
+        method=True,
+        index_on_thread_id=True,
+        id_field_name="thread_id",
+    )
     @abc.abstractmethod
     def list_relations_without_caching(self, schema_relation: BaseRelation) -> List[BaseRelation]:
         """List relations in the given schema, bypassing the cache.

dbt/adapters/catalogs/__init__.py

@@ -3,6 +3,7 @@ from dbt.adapters.catalogs._exceptions import (
     DbtCatalogIntegrationAlreadyExistsError,
     DbtCatalogIntegrationNotFoundError,
     DbtCatalogIntegrationNotSupportedError,
+    InvalidCatalogIntegrationConfigError,
 )
 from dbt.adapters.catalogs._integration import (
     CatalogIntegration,

dbt/adapters/catalogs/_exceptions.py

@@ -30,3 +30,10 @@ class DbtCatalogIntegrationNotSupportedError(DbtConfigError):
             f"Expected one of: {', '.join(supported_catalog_types)}"
         )
         super().__init__(msg)
+
+
+class InvalidCatalogIntegrationConfigError(DbtConfigError):
+    def __init__(self, catalog_name: str, msg: str) -> None:
+        self.catalog_name = catalog_name
+        msg = f"Invalid catalog integration config: {self.catalog_name}. {msg}"
+        super().__init__(msg)

dbt/adapters/exceptions/compilation.py

@@ -255,3 +255,15 @@ class RelationWrongTypeError(CompilationError):
         )
 
         return msg
+
+
+class InvalidRelationConfigError(CompilationError):
+    def __init__(self, relation, config, msg):
+        self.relation = relation
+        self.config = config
+        self.msg = msg
+        super().__init__(msg=self.get_message())
+
+    def get_message(self) -> str:
+        msg = f"Invalid relation config: {self.config}"
+        return msg

dbt/adapters/record/base.py

@@ -2,14 +2,15 @@
 
 import dataclasses
 
-from typing import Optional, Tuple, Dict, Any, TYPE_CHECKING
+from typing import Optional, Tuple, Dict, Any, TYPE_CHECKING, List
 
 from dbt.adapters.contracts.connection import AdapterResponse
-from dbt.adapters.record.serialization import serialize_agate_table
+from dbt.adapters.record.serialization import serialize_agate_table, serialize_bindings
 from dbt_common.record import Record, Recorder
 
 if TYPE_CHECKING:
     from agate import Table
+    from dbt.adapters.base.relation import BaseRelation
 
 
 @dataclasses.dataclass
@@ -161,3 +162,85 @@ class AdapterStandardizeGrantsDictRecord(Record):
     params_cls = AdapterStandardizeGrantsDictParams
     result_cls = AdapterStandardizeGrantsDictResult
     group = "Available"
+
+
+@dataclasses.dataclass
+class AdapterAddQueryParams:
+    thread_id: str
+    sql: str
+    auto_begin: bool = True
+    bindings: Optional[Any] = None
+    abridge_sql_log: bool = False
+
+    def _to_dict(self):
+        return {
+            "thread_id": self.thread_id,
+            "sql": self.sql,
+            "auto_begin": self.auto_begin,
+            "bindings": serialize_bindings(self.bindings),
+            "abridge_sql_log": self.abridge_sql_log,
+        }
+
+
+@dataclasses.dataclass
+class AdapterAddQueryResult:
+    return_val: Tuple[str, str]
+
+    def _to_dict(self):
+        return {
+            "return_val": {
+                "conn": "conn",
+                "cursor": "cursor",
+            }
+        }
+
+
+@Recorder.register_record_type
+class AdapterAddQueryRecord(Record):
+    params_cls = AdapterAddQueryParams
+    result_cls = AdapterAddQueryResult
+    group = "Available"
+
+
+@dataclasses.dataclass
+class AdapterListRelationsWithoutCachingParams:
+    thread_id: str
+    schema_relation: "BaseRelation"
+
+    def _to_dict(self):
+        from dbt.adapters.record.serialization import serialize_base_relation
+
+        return {
+            "thread_id": self.thread_id,
+            "schema_relation": serialize_base_relation(self.schema_relation),
+        }
+
+    def _from_dict(self, data: Dict[str, Any]):
+        from dbt.adapters.record.serialization import deserialize_base_relation
+
+        self.thread_id = data["thread_id"]
+        self.schema_relation = deserialize_base_relation(data["schema_relation"])
+
+
+@dataclasses.dataclass
+class AdapterListRelationsWithoutCachingResult:
+    return_val: List["BaseRelation"]
+
+    def _to_dict(self):
+        from dbt.adapters.record.serialization import serialize_base_relation_list
+
+        return {"return_val": serialize_base_relation_list(self.return_val)}
+
+    def _from_dict(self, data: Dict[str, Any]):
+        from dbt.adapters.record.serialization import deserialize_base_relation_list
+
+        self.return_val = deserialize_base_relation_list(data["return_val"])
+
+
+@Recorder.register_record_type
+class AdapterListRelationsWithoutCachingRecord(Record):
+    """Implements record/replay support for the BaseAdapter.list_relations_without_caching() method."""
+
+    params_cls = AdapterListRelationsWithoutCachingParams
+    result_cls = AdapterListRelationsWithoutCachingResult
+    group = "Available"

dbt/adapters/record/serialization.py

@@ -1,9 +1,10 @@
 from datetime import datetime, date
 from decimal import Decimal
-from typing import Any, Dict, TYPE_CHECKING
+from typing import Any, Dict, TYPE_CHECKING, List, Union
 
 if TYPE_CHECKING:
     from agate import Table
+    from dbt.adapters.base.relation import BaseRelation
 
 
 def _column_filter(val: Any) -> Any:
@@ -29,3 +30,34 @@ def serialize_agate_table(table: "Table") -> Dict[str, Any]:
         "column_types": [t.__class__.__name__ for t in table.column_types],
         "rows": rows,
     }
+
+
+def serialize_bindings(bindings: Any) -> Union[None, List[Any], str]:
+    if bindings is None:
+        return None
+    elif isinstance(bindings, list):
+        return list(map(_column_filter, bindings))
+    else:
+        return "bindings"
+
+
+def serialize_base_relation(relation: "BaseRelation") -> Dict[str, Any]:
+    """Serialize a BaseRelation object for recording."""
+    return relation.to_dict(omit_none=True)
+
+
+def serialize_base_relation_list(relations: List["BaseRelation"]) -> List[Dict[str, Any]]:
+    """Serialize a list of BaseRelation objects for recording."""
+    return [serialize_base_relation(relation) for relation in relations]
+
+
+def deserialize_base_relation(relation_dict: Dict[str, Any]) -> "BaseRelation":
+    """Deserialize a BaseRelation object from a dictionary."""
+    from dbt.adapters.base.relation import BaseRelation
+
+    return BaseRelation.from_dict(relation_dict)
+
+
+def deserialize_base_relation_list(relations_data: List[Dict[str, Any]]) -> List["BaseRelation"]:
+    """Deserialize a list of BaseRelation objects from dictionaries."""
+    return [deserialize_base_relation(relation_dict) for relation_dict in relations_data]

dbt/adapters/sql/impl.py

@@ -8,7 +8,7 @@ from dbt.adapters.cache import _make_ref_key_dict
 from dbt.adapters.contracts.connection import AdapterResponse, Connection
 from dbt.adapters.events.types import ColTypeChange, SchemaCreation, SchemaDrop
 from dbt.adapters.exceptions import RelationTypeNullError
-from dbt.adapters.record.base import AdapterTestSqlRecord
+from dbt.adapters.record.base import AdapterTestSqlRecord, AdapterAddQueryRecord
 from dbt.adapters.sql.connections import SQLConnectionManager
 
 LIST_RELATIONS_MACRO_NAME = "list_relations_without_caching"
@@ -49,6 +49,9 @@ class SQLAdapter(BaseAdapter):
     connections: SQLConnectionManager
 
     @available.parse(lambda *a, **k: (None, None))
+    @record_function(
+        AdapterAddQueryRecord, method=True, index_on_thread_id=True, id_field_name="thread_id"
+    )
     def add_query(
         self,
         sql: str,

{dbt_adapters-1.16.5.dist-info → dbt_adapters-1.16.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbt-adapters
-Version: 1.16.5
+Version: 1.16.6
 Summary: The set of adapter protocols and base functionality that supports integration with dbt-core
 Project-URL: Homepage, https://github.com/dbt-labs/dbt-adapters/tree/main/dbt-adapters
 Project-URL: Documentation, https://docs.getdbt.com

{dbt_adapters-1.16.5.dist-info → dbt_adapters-1.16.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-dbt/adapters/__about__.py,sha256=Qz0BE08lu2_u7MixE7GM_8vIJjUC10H2IxTPA0I0kWU,19
+dbt/adapters/__about__.py,sha256=yVGsgH-1PnNg3ANhU0Ki30jCFRQx1ac_0O7ZJ8deZa8,19
 dbt/adapters/__init__.py,sha256=3noHsg-64qI0_Pw6OR9F7l1vU2_qrJvinq8POTtuaZM,252
 dbt/adapters/cache.py,sha256=WGy4ewnz-J13LverTACBW2iFhGswrWLgm-wiBrQnMzo,20084
 dbt/adapters/capability.py,sha256=M3FkC9veKnNB7a7uQyl7EHX_AGNXPChbHAkcY4cgXCY,2534
@@ -11,15 +11,15 @@ dbt/adapters/base/README.md,sha256=muHQntC07Lh6L1XfVgwKhV5RltOPBLYPdQqd8_7l34c,5
 dbt/adapters/base/__init__.py,sha256=Nc8lQVkOzAqdcxk4cw4E_raxN9CAWMwhQx4STdiicxg,456
 dbt/adapters/base/column.py,sha256=Uj20UixoxCn2rlv4QDNONyys6CDkDFyG3anCXKf0T2c,5350
 dbt/adapters/base/connections.py,sha256=-C5dOwGgMKH8n_v6wjwOxV7chBdS0GjOGwNQCUbhhWc,16951
-dbt/adapters/base/impl.py,sha256=_azN61jgWam6SWFhTJGYTI7H3ME9IQDUjv4A2dyf-mc,78968
+dbt/adapters/base/impl.py,sha256=uACwbMYJL9KykHdckf-9Egz4MBGecwOsSrYtU72OEx8,79181
 dbt/adapters/base/meta.py,sha256=c5j0qeGec1cAi-IlVV_JkhMk01p5XqbtGj02uxGP1S4,5686
 dbt/adapters/base/plugin.py,sha256=rm0GjNHnWM2mn0GJOjciZLwn-02xlzWCoMT9u-epwP0,1076
 dbt/adapters/base/query_headers.py,sha256=rHgux9b_vPp_xRsg8QU3NE202Hho4i4zO9u9Gx_BCF4,3651
 dbt/adapters/base/relation.py,sha256=eePneHZQ-bb0ZFaEpC_1xzsxJF38e6k7b3FUxD8PK0I,19383
-dbt/adapters/catalogs/__init__.py,sha256=9Lsz26jcljxwJLBgfHHQMqOyoKNahrqxSst0YN9WAy8,456
+dbt/adapters/catalogs/__init__.py,sha256=jiZb5IchAHnHSHz5J44l3Wzc0-oHeYXv4_7SBKOL7Pw,498
 dbt/adapters/catalogs/_client.py,sha256=bIXZxEL1-4kSXHFPnMw3egIOx1fKZmjxKiO1Wv-KuHg,2258
 dbt/adapters/catalogs/_constants.py,sha256=8O5wLK4oVqirpC9ap7PIXodq8XtkQhHJ5W0OU1xtwjk,55
-dbt/adapters/catalogs/_exceptions.py,sha256=5zkSmjUG5y_Dzv0AAUptsJpNDHnmb8D0mUb_HIQCj_I,1133
+dbt/adapters/catalogs/_exceptions.py,sha256=exanVSndLgMlpCyVIg5p1AUkQO2HiI4bkFtS3ADFVUQ,1407
 dbt/adapters/catalogs/_integration.py,sha256=0acuF_Awoe-GJrCdNwBQldtgRmBtpqYpZkTPRVJFhys,6594
 dbt/adapters/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt/adapters/clients/jinja.py,sha256=NsZOiBpOLunS46hRL5OcX1MpY3Ih6_87Vgz4qd_PNbc,768
@@ -36,13 +36,13 @@ dbt/adapters/events/types.py,sha256=cxGUlR2WceSOis81FwHS7rwfFpxCmy31ur5Yp8Pz4o0,
 dbt/adapters/exceptions/__init__.py,sha256=LxRkxHARMzrPegrjha5tQ8vQi1PnL_ooq1SVqm9aiZs,1268
 dbt/adapters/exceptions/alias.py,sha256=QlCd2jvatfndec1DQLMMBJ-C_7w8yAySuAFHK2ga1g8,798
 dbt/adapters/exceptions/cache.py,sha256=u720DQQKm8pyf_9loD9HGA9WgaeZAlg0sBn0sGc-rhA,2492
-dbt/adapters/exceptions/compilation.py,sha256=2QsAX-z_1K4q8jGtvbC1JM6dEh7lyOF7rTWkV6RXC8o,8720
+dbt/adapters/exceptions/compilation.py,sha256=ylUvVLwlzeKqFr16pGhSqHBbE3Z-lneDppUQtNmrE-o,9065
 dbt/adapters/exceptions/connection.py,sha256=x82j2Ix242Slm6Ima8Tol3GLOB9yZYH5lq6IV1WKq54,445
 dbt/adapters/exceptions/database.py,sha256=nIXJdQyPQOZaiKvCkQ3MoKlKOiaN58rtDZflw3CSkug,1618
 dbt/adapters/record/__init__.py,sha256=u_0eJzN5RL90oL-RNoP2wWGMCGp0kG0lZ0uVWnnGdxs,123
-dbt/adapters/record/base.py,sha256=oqBeddunk9bAyNZBC2pWQohYtD6_ju_tf_74I6NAN04,4277
+dbt/adapters/record/base.py,sha256=4HAGpxUX-9JoWgrWctM29xX_EBfVUD-FbLGukQH33kM,6714
 dbt/adapters/record/handle.py,sha256=hCQpDmZGKqJXrSqgYcb-F542TZi069uz1MY9fS15INo,1907
-dbt/adapters/record/serialization.py,sha256=ehcLioi4J7TujP58fMNGcePuMxx-2G4Dj1KoYm6j6FU,756
+dbt/adapters/record/serialization.py,sha256=aysH908v3qURycAPZ4a-TwDa5UQZrcmaeu1Zj_RFTvE,1987
 dbt/adapters/record/cursor/cursor.py,sha256=AW0B1g0j5mn7pUnvsE7zd4_Zi7cSFOsyvPbE2HV0CPI,2967
 dbt/adapters/record/cursor/description.py,sha256=5LNebP2AF2aicPWfM9FsD5r7SAmdac8TX_4NZfJQgPk,1060
 dbt/adapters/record/cursor/execute.py,sha256=f8Yc1MKExj7QLbFnK8WnnwsoGESuYMWnFWhytbcAPjc,1271
@@ -57,7 +57,7 @@ dbt/adapters/relation_configs/config_change.py,sha256=hf6fDWbZpKvZdM6z-OtY-Gveip
 dbt/adapters/relation_configs/config_validation.py,sha256=wlJUMwOEPhYFch-LRtEWfLNJMq8jL1tRhOUHmNX8nFw,1978
 dbt/adapters/sql/__init__.py,sha256=WLWZJfqc8pr1N1BMVe9gM-KQ4URJIeKfLqTuJBD1VN0,107
 dbt/adapters/sql/connections.py,sha256=N7Mmg9ZS5ztKQCbJgbb1jwVhaBeMBqnR_3SWxEytxzY,8453
-dbt/adapters/sql/impl.py,sha256=dhpvlxc3mZWdcP42xMzW8_nM2zJGMMt1_hLUUfXx3Dw,10944
+dbt/adapters/sql/impl.py,sha256=mTzK7vK-Nf6PdJwh8lzMnqOP5I6663s1ZMc3LomsYh4,11090
 dbt/include/__init__.py,sha256=qEFeq3yuf3lQKVseALmL8aPM8fpCS54B_5pry00M3hk,76
 dbt/include/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt/include/global_project/__init__.py,sha256=-0HL5OkeJSrxglm1Y-UltTiBPY2BbWx8ZpTiJ7ypSvw,73
@@ -163,7 +163,7 @@ dbt/include/global_project/macros/utils/right.sql,sha256=EwNG98CAFIwNDmarwopf7Rk
 dbt/include/global_project/macros/utils/safe_cast.sql,sha256=1mswwkDACmIi1I99JKb_-vq3kjMe4HhMRV70mW8Bt4Y,298
 dbt/include/global_project/macros/utils/split_part.sql,sha256=fXEIS0oIiYR7-4lYbb0QbZdG-q2TpV63AFd1ky4I5UM,714
 dbt/include/global_project/tests/generic/builtin.sql,sha256=p94xdyPwb2TlxgLBqCfrcRfJ1QNgsjPvBm8f0Q5eqZM,1022
-dbt_adapters-1.16.5.dist-info/METADATA,sha256=otMClzeh3gNK273cCp3pI2qGK6GYjumhFN6RYt_ka9k,4534
-dbt_adapters-1.16.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dbt_adapters-1.16.5.dist-info/licenses/LICENSE,sha256=9yjigiJhWcCZvQjdagGKDwrRph58QWc5P2bVSQwXo6s,11344
-dbt_adapters-1.16.5.dist-info/RECORD,,
+dbt_adapters-1.16.6.dist-info/METADATA,sha256=vmRysouwZKtOTcNN_y7mnp8m1zKkYdPtrk-9PpKvwW0,4534
+dbt_adapters-1.16.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dbt_adapters-1.16.6.dist-info/licenses/LICENSE,sha256=9yjigiJhWcCZvQjdagGKDwrRph58QWc5P2bVSQwXo6s,11344
+dbt_adapters-1.16.6.dist-info/RECORD,,