dbt-firebolt 1.7.0__py3-none-any.whl → 1.8.1__py3-none-any.whl

dbt/adapters/firebolt/__init__.py

@@ -7,7 +7,7 @@ from dbt.adapters.firebolt.connections import (
 from dbt.adapters.firebolt.impl import FireboltAdapter
 from dbt.include import firebolt
 
-__version__ = "1.7.0"
+__version__ = "1.8.1"
 
 Plugin = AdapterPlugin(
     adapter=FireboltAdapter,  # type: ignore

dbt/adapters/firebolt/connections.py

@@ -1,19 +1,23 @@
 from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import datetime
+from multiprocessing.context import SpawnContext
 from typing import Generator, Optional, Tuple, Union
 
-import dbt.exceptions
-from dbt.adapters.base import Credentials
-from dbt.adapters.sql import SQLConnectionManager
-from dbt.contracts.connection import (
+from dbt.adapters.contracts.connection import (
     AdapterRequiredConfig,
     AdapterResponse,
     Connection,
+    Credentials,
     QueryComment,
 )
-from dbt.events import AdapterLogger  # type: ignore
-from dbt.exceptions import DbtRuntimeError
+from dbt.adapters.events.logging import AdapterLogger
+from dbt.adapters.sql.connections import SQLConnectionManager
+from dbt_common.exceptions import (
+    DbtConfigError,
+    DbtRuntimeError,
+    NotImplementedError,
+)
 from firebolt.client import DEFAULT_API_URL
 from firebolt.client.auth import Auth, ClientCredentials, UsernamePassword
 from firebolt.db import ARRAY, DECIMAL
@@ -40,18 +44,22 @@ class FireboltCredentials(Credentials):
     account_name: Optional[str] = None
     retries: int = 1
 
+    _ALIASES = {
+        'host': 'api_endpoint',
+    }
+
     def __post_init__(self) -> None:
         # If user and password are not provided, assume client_id and client_secret
         # are provided instead
         if not self.user and not self.password:
             if not self.client_id or not self.client_secret:
-                raise dbt.exceptions.DbtProfileError(
+                raise DbtConfigError(
                     'Either user and password or client_id and client_secret'
                     ' must be provided'
                 )
         else:
             if self.client_id or self.client_secret:
-                raise dbt.exceptions.DbtProfileError(
+                raise DbtConfigError(
                     'Either user and password or client_id and client_secret'
                     ' must be provided'
                 )
@@ -99,13 +107,13 @@ class FireboltConnectionManager(SQLConnectionManager):
 
     TYPE = 'firebolt'
 
-    def __init__(self, profile: AdapterRequiredConfig):
+    def __init__(self, profile: AdapterRequiredConfig, mp_context: SpawnContext):
         # Query comment in appent mode only
         # This allows clearer view of queries in query_history
         if not hasattr(profile, 'query_comment'):
             setattr(profile, 'query_comment', QueryComment())
         profile.query_comment.append = True
-        super().__init__(profile)
+        super().__init__(profile, mp_context)
 
     def __str__(self) -> str:
         return 'FireboltConnectionManager()'
@@ -181,9 +189,7 @@ class FireboltConnectionManager(SQLConnectionManager):
 
     def cancel(self, connection: Connection) -> None:
         """Cancel the last query on the given connection."""
-        raise dbt.exceptions.NotImplementedError(
-            '`cancel` is not implemented for this adapter!'
-        )
+        raise NotImplementedError('`cancel` is not implemented for this adapter!')
 
     @classmethod
     def data_type_code_to_name(  # type: ignore[override] # FIR-29423

dbt/adapters/firebolt/impl.py

@@ -5,9 +5,8 @@ from datetime import datetime
 from typing import Any, List, Mapping, Optional, Union
 
 import agate
-from dbt.adapters.base import available  # type: ignore
-from dbt.adapters.base.impl import AdapterConfig  # type: ignore
 from dbt.adapters.base.impl import ConstraintSupport
+from dbt.adapters.base.meta import available
 from dbt.adapters.base.relation import BaseRelation
 from dbt.adapters.capability import (
     Capability,
@@ -15,10 +14,11 @@ from dbt.adapters.capability import (
     CapabilitySupport,
     Support,
 )
-from dbt.adapters.sql import SQLAdapter  # type: ignore
-from dbt.contracts.graph.nodes import ConstraintType
-from dbt.dataclass_schema import ValidationError, dbtClassMixin
-from dbt.exceptions import (
+from dbt.adapters.protocol import AdapterConfig
+from dbt.adapters.sql.impl import SQLAdapter
+from dbt_common.contracts.constraints import ConstraintType
+from dbt_common.dataclass_schema import ValidationError, dbtClassMixin
+from dbt_common.exceptions import (
     CompilationError,
     DbtRuntimeError,
     NotImplementedError,

dbt/adapters/firebolt/relation.py

@@ -1,10 +1,10 @@
 from dataclasses import dataclass, field
 from typing import Dict, FrozenSet, Optional
 
-from dbt.adapters.base import RelationType  # type: ignore
-from dbt.adapters.base.relation import BaseRelation, Policy  # type: ignore
-from dbt.adapters.relation_configs import RelationConfigBase  # type: ignore
-from dbt.exceptions import DbtRuntimeError
+from dbt.adapters.base.relation import BaseRelation
+from dbt.adapters.contracts.relation import Policy, RelationType
+from dbt.adapters.relation_configs.config_base import RelationConfigBase
+from dbt_common.exceptions import DbtRuntimeError
 
 
 @dataclass

{dbt_firebolt-1.7.0.dist-info → dbt_firebolt-1.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbt-firebolt
-Version: 1.7.0
+Version: 1.8.1
 Summary: The Firebolt adapter plugin for dbt (data build tool)
 Home-page: https://github.com/firebolt-db/dbt-firebolt
 Author: Firebolt
@@ -16,8 +16,9 @@ Classifier: Programming Language :: Python :: 3.9
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: dbt-core <1.8,~=1.6
-Requires-Dist: firebolt-sdk >=1.1.0
+Requires-Dist: dbt-adapters <2.0,>=1.0
+Requires-Dist: dbt-core >=1.8.0
+Requires-Dist: firebolt-sdk >=1.5.0
 Requires-Dist: pydantic >=0.23
 Provides-Extra: dev
 Requires-Dist: allure-pytest ==2.* ; extra == 'dev'

{dbt_firebolt-1.7.0.dist-info → dbt_firebolt-1.8.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-dbt/adapters/firebolt/__init__.py,sha256=ifGIav8ks_A2VpLQpLvT6xkMMV0n0LOPE767T1DtlLI,411
+dbt/adapters/firebolt/__init__.py,sha256=9aBbFLagxZS-jN6NXO9V4GuNfFOMir0711l55MIer6w,411
 dbt/adapters/firebolt/__version__.py,sha256=zRlZGglif76ZVuWWSjsH_MMPgtVQqmj-SryYJW25FL4,69
 dbt/adapters/firebolt/column.py,sha256=COo_wjhCFgS3GFcPIPcoq7WAWgzN6DB2XqG-gk51WBc,539
-dbt/adapters/firebolt/connections.py,sha256=Sp2C03ynSqmRIQO9z2YGfVIimWZrygST7uKT0t5_RwU,7880
-dbt/adapters/firebolt/impl.py,sha256=dyqJvJwbs2CFouSFjOPosD8qY4uz2TufzBfRWlQmDnk,13957
-dbt/adapters/firebolt/relation.py,sha256=HdzHsjYyP1djnUxOl9TB2OGMv7LJyeRXr2kgZRzsgZM,1937
+dbt/adapters/firebolt/connections.py,sha256=r7imGPka0klw1ev-5GTSIwUbO-Df_5DE4Ss-9OfLBMk,7990
+dbt/adapters/firebolt/impl.py,sha256=vHK7AfpmMdZMCHaky3IEufl18IYFPs9mdvdmTpMrUh4,13939
+dbt/adapters/firebolt/relation.py,sha256=Xg3Nrjw3UrF_qwnuGbPT97rSXRiDP1GlIAoBF4b7QnY,1922
 dbt/adapters/firebolt/relation_configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbt/include/firebolt/__init__.py,sha256=vBGWeG-dHHkimfnX8axBJ4IgAowFw8xADmo6Auzn2xc,52
 dbt/include/firebolt/dbt_project.yml,sha256=uHJ-i0wD1D74DWSSSzsFKoKbAN1w4LBgVpnkm8e-M1g,76
@@ -53,8 +53,8 @@ dbt/include/firebolt/macros/utils/position.sql,sha256=WPo9_bhvNYJooEAsHC9OcnNAwU
 dbt/include/firebolt/macros/utils/right.sql,sha256=_mm1_2MvlOH4O6CmYhgvVxMLfDxAvgv-EMwZ8OBOq-I,254
 dbt/include/firebolt/macros/utils/split_part.sql,sha256=5dUlbx3Pt1iWWaIlxiXyYUwUqzXuLLXMB-1aGJHNk4o,464
 dbt/include/firebolt/macros/utils/timestamps.sql,sha256=22g-QpJjBuBUQOHNpQ_TuMRa-cHxaxXPv8ItEUo-vEk,397
-dbt_firebolt-1.7.0.dist-info/LICENSE,sha256=Nn0EGvW3qmoZpBV_JVM3iPukFf3RiNCIizrWe_2oTHk,11354
-dbt_firebolt-1.7.0.dist-info/METADATA,sha256=uKP4G9kpOgRj60f675-MUQSFPYynk9itGUei8HUH4oU,5034
-dbt_firebolt-1.7.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-dbt_firebolt-1.7.0.dist-info/top_level.txt,sha256=B2YH4he17ajilEWOGCKHbRcEJlCuZKwCcgFcLPntLsE,4
-dbt_firebolt-1.7.0.dist-info/RECORD,,
+dbt_firebolt-1.8.1.dist-info/LICENSE,sha256=Nn0EGvW3qmoZpBV_JVM3iPukFf3RiNCIizrWe_2oTHk,11354
+dbt_firebolt-1.8.1.dist-info/METADATA,sha256=dTjTrACpMqv6fx2vGcP7gq01LevtxZnQS6_iRTmhSq4,5070
+dbt_firebolt-1.8.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+dbt_firebolt-1.8.1.dist-info/top_level.txt,sha256=B2YH4he17ajilEWOGCKHbRcEJlCuZKwCcgFcLPntLsE,4
+dbt_firebolt-1.8.1.dist-info/RECORD,,