refactor: Use a single source of truth for built-in capabilities

meltano · Jun 25, 2024 · b743009 · b743009
1 parent 9e051aa
commit b743009
Show file tree

Hide file tree

Showing 15 changed files with 642 additions and 503 deletions.
diff --git a/singer_sdk/connectors/sql.py b/singer_sdk/connectors/sql.py
@@ -17,7 +17,7 @@
 from singer_sdk import typing as th
 from singer_sdk._singerlib import CatalogEntry, MetadataMapping, Schema
 from singer_sdk.exceptions import ConfigValidationError
-from singer_sdk.helpers.capabilities import TargetLoadMethods
+from singer_sdk.helpers import capabilities
 
 if t.TYPE_CHECKING:
     from sqlalchemy.engine import Engine
@@ -779,7 +779,7 @@ def prepare_table(
                 as_temp_table=as_temp_table,
             )
             return
-        if self.config["load_method"] == TargetLoadMethods.OVERWRITE:
+        if self.config["load_method"] == capabilities.TargetLoadMethods.OVERWRITE:
             self.get_table(full_table_name=full_table_name).drop(self._engine)
             self.create_empty_table(
                 full_table_name=full_table_name,

diff --git a/singer_sdk/helpers/capabilities.py b/singer_sdk/helpers/capabilities.py