Remove usage of dbt.contracts.relation in dbt/adapters

.changes/unreleased/Under the Hood-20231205-120559.yaml

@@ -0,0 +1,6 @@
+kind: Under the Hood
+body: Remove usage of dbt.contracts in dbt/adapters
+time: 2023-12-05T12:05:59.936775+09:00
+custom:
+  Author: michelleark
+  Issue: "9208"

core/dbt/adapters/base/relation.py

@@ -3,7 +3,7 @@
 from typing import Optional, TypeVar, Any, Type, Dict, Iterator, Tuple, Set, Union, FrozenSet
 
 from dbt.contracts.graph.nodes import SourceDefinition, ManifestNode, ResultNode, ParsedNode
-from dbt.contracts.relation import (
+from dbt.adapters.contracts.relation import (
     RelationType,
     ComponentName,
     HasQuoting,

core/dbt/contracts/relation.py → core/dbt/adapters/contracts/relation.py

@@ -8,9 +8,8 @@
 
 from dbt.common.dataclass_schema import dbtClassMixin, StrEnum
 
-from dbt.contracts.util import Replaceable
-from dbt.common.exceptions import CompilationError
-from dbt.exceptions import DataclassNotDictError
+from dbt.common.contracts.util import Replaceable
+from dbt.common.exceptions import CompilationError, DataclassNotDictError
 from dbt.common.utils import deep_merge
 
 

core/dbt/adapters/protocol.py

@@ -14,10 +14,10 @@
 import agate
 
 from dbt.adapters.contracts.connection import Connection, AdapterRequiredConfig, AdapterResponse
+from dbt.adapters.contracts.relation import Policy, HasQuoting
 from dbt.contracts.graph.nodes import ResultNode
 from dbt.contracts.graph.model_config import BaseConfig
 from dbt.contracts.graph.manifest import Manifest
-from dbt.contracts.relation import Policy, HasQuoting
 
 
 @dataclass

core/dbt/common/exceptions/__init__.py

@@ -1,3 +1,4 @@
 from dbt.common.exceptions.base import *  # noqa
 from dbt.common.exceptions.events import *  # noqa
 from dbt.common.exceptions.macros import *  # noqa
+from dbt.common.exceptions.contracts import *  # noqa

core/dbt/common/exceptions/contracts.py

@@ -0,0 +1,17 @@
+from typing import Any
+from dbt.common.exceptions import CompilationError
+
+
+# this is part of the context and also raised in dbt.contracts.relation.py
+class DataclassNotDictError(CompilationError):
+    def __init__(self, obj: Any):
+        self.obj = obj
+        super().__init__(msg=self.get_message())
+
+    def get_message(self) -> str:
+        msg = (
+            f'The object ("{self.obj}") was used as a dictionary. This '
+            "capability has been removed from objects of this type."
+        )
+
+        return msg

core/dbt/config/runtime.py

@@ -15,13 +15,13 @@
     Type,
 )
 
-from dbt.flags import get_flags
 from dbt.adapters.factory import get_include_paths, get_relation_class_by_name
-from dbt.config.project import load_raw_project
 from dbt.adapters.contracts.connection import AdapterRequiredConfig, Credentials, HasCredentials
+from dbt.adapters.contracts.relation import ComponentName
+from dbt.flags import get_flags
+from dbt.config.project import load_raw_project
 from dbt.contracts.graph.manifest import ManifestMetadata
 from dbt.contracts.project import Configuration, UserConfig
-from dbt.contracts.relation import ComponentName
 from dbt.common.dataclass_schema import ValidationError
 from dbt.common.events.functions import warn_or_error
 from dbt.common.events.types import UnusedResourceConfigPath

core/dbt/context/exceptions_jinja.py

@@ -4,7 +4,12 @@
 from dbt.common.events.functions import warn_or_error
 from dbt.common.events.types import JinjaLogWarning
 
-from dbt.common.exceptions import DbtRuntimeError, NotImplementedError, DbtDatabaseError
+from dbt.common.exceptions import (
+    DbtRuntimeError,
+    NotImplementedError,
+    DbtDatabaseError,
+    DataclassNotDictError,
+)
 from dbt.adapters.exceptions import (
     MissingConfigError,
     ColumnTypeMissingError,
@@ -15,7 +20,6 @@
     MissingRelationError,
     AmbiguousAliasError,
     AmbiguousCatalogMatchError,
-    DataclassNotDictError,
     CompilationError,
     DependencyNotFoundError,
     DependencyError,

core/dbt/contracts/util.py

@@ -10,6 +10,7 @@
 )
 from dbt.version import __version__
 
+from dbt.common.contracts.util import Replaceable
 from dbt.common.events.functions import get_metadata_vars
 from dbt.common.invocation import get_invocation_id
 from dbt.common.dataclass_schema import dbtClassMixin
@@ -41,11 +42,6 @@ class Foo:
     return []
 
 
-class Replaceable:
-    def replace(self, **kwargs):
-        return dataclasses.replace(self, **kwargs)
-
-
 class Mergeable(Replaceable):
     def merged(self, *args):
         """Perform a shallow merge, where the last non-None write wins. This is

core/dbt/exceptions.py

@@ -1363,21 +1363,6 @@ def get_message(self) -> str:
         return msg
 
 
-# this is part of the context and also raised in dbt.contracts.relation.py
-class DataclassNotDictError(CompilationError):
-    def __init__(self, obj: Any):
-        self.obj = obj
-        super().__init__(msg=self.get_message())
-
-    def get_message(self) -> str:
-        msg = (
-            f'The object ("{self.obj}") was used as a dictionary. This '
-            "capability has been removed from objects of this type."
-        )
-
-        return msg
-
-
 class DependencyNotFoundError(CompilationError):
     def __init__(self, node, node_description, required_pkg):
         self.node = node

plugins/postgres/dbt/adapters/postgres/relation.py

@@ -7,7 +7,7 @@
     RelationResults,
 )
 from dbt.context.providers import RuntimeConfigObject
-from dbt.contracts.relation import RelationType
+from dbt.adapters.contracts.relation import RelationType
 from dbt.common.exceptions import DbtRuntimeError
 
 from dbt.adapters.postgres.relation_configs import (

tests/adapter/dbt/tests/adapter/materialized_view/basic.py

@@ -3,7 +3,7 @@
 import pytest
 
 from dbt.adapters.base.relation import BaseRelation
-from dbt.contracts.relation import RelationType
+from dbt.adapters.contracts.relation import RelationType
 from dbt.tests.util import (
     assert_message_in_logs,
     get_model_file,

tests/adapter/dbt/tests/adapter/materialized_view/changes.py

@@ -4,7 +4,7 @@
 
 from dbt.adapters.base.relation import BaseRelation
 from dbt.contracts.graph.model_config import OnConfigurationChangeOption
-from dbt.contracts.relation import RelationType
+from dbt.adapters.contracts.relation import RelationType
 from dbt.tests.util import (
     assert_message_in_logs,
     get_model_file,

tests/unit/test_postgres_adapter.py

@@ -7,7 +7,7 @@
 from unittest import mock
 
 from dbt.adapters.base import BaseRelation
-from dbt.contracts.relation import Path
+from dbt.adapters.contracts.relation import Path
 from dbt.task.debug import DebugTask
 
 from dbt.adapters.base.query_headers import MacroQueryStringSetter

tests/unit/test_relation.py

@@ -3,7 +3,7 @@
 import pytest
 
 from dbt.adapters.base import BaseRelation
-from dbt.contracts.relation import RelationType
+from dbt.adapters.contracts.relation import RelationType
 
 
 @pytest.mark.parametrize(