dblift 2.4.2__tar.gz → 2.5.1__tar.gz

{dblift-2.4.2 → dblift-2.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dblift
-Version: 2.4.2
+Version: 2.5.1
 Summary: A database migration tool that supports multiple databases with native Python drivers
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
@@ -83,6 +83,11 @@ Provides-Extra: timescaledb
 Requires-Dist: psycopg[binary]>=3.1; extra == "timescaledb"
 Provides-Extra: citus
 Requires-Dist: psycopg[binary]>=3.1; extra == "citus"
+Provides-Extra: cockroachdb
+Requires-Dist: psycopg[binary]>=3.1; extra == "cockroachdb"
+Provides-Extra: redshift
+Requires-Dist: sqlalchemy-redshift>=1.0.0; extra == "redshift"
+Requires-Dist: redshift-connector>=2.1.0; extra == "redshift"
 Provides-Extra: fastapi
 Requires-Dist: fastapi>=0.100; extra == "fastapi"
 Provides-Extra: flask
@@ -727,6 +732,8 @@ DBLift works with these databases:
 | Database | Connection URL Example | Driver extra |
 |----------|------------------------|--------------|
 | PostgreSQL | `postgresql+psycopg://localhost:5432/mydb` | `dblift[postgresql]` |
+| CockroachDB | `postgresql+psycopg://localhost:26257/mydb` | `dblift[cockroachdb]` |
+| Redshift | `postgresql+psycopg://cluster.example.com:5439/dev` | `dblift[redshift]` |
 | SQL Server | `mssql+pymssql://localhost:1433/mydb` | `dblift[sqlserver]` |
 | Oracle | `oracle+oracledb://localhost:1521?sid=SID` | `dblift[oracle]` |
 | MySQL | `mysql+pymysql://localhost:3306/mydb` | `dblift[mysql]` |

{dblift-2.4.2 → dblift-2.5.1}/README.md

@@ -630,6 +630,8 @@ DBLift works with these databases:
 | Database | Connection URL Example | Driver extra |
 |----------|------------------------|--------------|
 | PostgreSQL | `postgresql+psycopg://localhost:5432/mydb` | `dblift[postgresql]` |
+| CockroachDB | `postgresql+psycopg://localhost:26257/mydb` | `dblift[cockroachdb]` |
+| Redshift | `postgresql+psycopg://cluster.example.com:5439/dev` | `dblift[redshift]` |
 | SQL Server | `mssql+pymssql://localhost:1433/mydb` | `dblift[sqlserver]` |
 | Oracle | `oracle+oracledb://localhost:1521?sid=SID` | `dblift[oracle]` |
 | MySQL | `mysql+pymysql://localhost:3306/mydb` | `dblift[mysql]` |

dblift-2.5.1/db/plugins/cockroachdb/__init__.py

@@ -0,0 +1,12 @@
+"""CockroachDB database provider plugin (PostgreSQL-compatible)."""
+
+__plugin_name__ = "cockroachdb"
+__plugin_version__ = "1.0.0"
+__plugin_description__ = "CockroachDB database provider"
+__plugin_dialects__ = ["cockroachdb"]
+__plugin_transport__ = "native"
+__plugin_class__ = "CockroachdbProvider"
+
+from .provider import CockroachdbProvider
+
+__all__ = ["CockroachdbProvider"]

dblift-2.5.1/db/plugins/cockroachdb/plugin.py

@@ -0,0 +1,28 @@
+"""Entry-point declaration for the CockroachDB plugin.
+
+CockroachDB is PostgreSQL-compatible: this plugin reuses PostgreSQL's config
+class (``config_dialect="postgresql"``), SQLAlchemy URL builder, and
+``psycopg`` driver, attaching only a distinct identity + quirks subclass. Users
+keep their ``postgresql://`` connection string and select this engine via
+``type: cockroachdb``.
+"""
+
+from __future__ import annotations
+
+from db.plugins.cockroachdb.provider import CockroachdbProvider
+from db.plugins.cockroachdb.quirks import CockroachdbQuirks
+from db.plugins.postgresql.sqlalchemy_url import build_sqlalchemy_url
+from db.provider_registry import PluginInfo
+
+PLUGIN: PluginInfo = PluginInfo(
+    name="cockroachdb",
+    version="1.0.0",
+    description="CockroachDB database provider",
+    dialects=["cockroachdb"],
+    provider_class=CockroachdbProvider,
+    transport="native",
+    quirks_class=CockroachdbQuirks,
+    config_dialect="postgresql",
+    sqlalchemy_url_builder=build_sqlalchemy_url,
+    native_driver_module="psycopg",
+)

dblift-2.5.1/db/plugins/cockroachdb/provider.py

@@ -0,0 +1,79 @@
+"""CockroachDB native provider (PostgreSQL-compatible)."""
+
+from __future__ import annotations
+
+import time
+
+from db.plugins.postgresql.provider import PostgreSqlProvider
+
+
+def _is_lock_contention_error(error: Exception) -> bool:
+    message = str(error).lower()
+    markers = (
+        "duplicate key",
+        "unique constraint",
+        "violates unique",
+    )
+    return any(marker in message for marker in markers)
+
+
+class CockroachdbProvider(PostgreSqlProvider):
+    """CockroachDB provider with table-backed migration locking."""
+
+    canonical_dialect_key = "cockroachdb"
+
+    def acquire_migration_lock(
+        self,
+        schema: str,
+        wait_timeout_seconds: int = 60,
+    ) -> bool:
+        """Acquire a CockroachDB migration lock using a committed lock row."""
+        self.create_migration_lock_table_if_not_exists(schema)
+        lock_table = self.MIGRATION_LOCK_TABLE
+        qualified_table = self.get_schema_qualified_name(schema, lock_table)
+
+        deadline = time.monotonic() + max(0, int(wait_timeout_seconds))
+        while True:
+            try:
+                self.execute_statement(
+                    f"""
+                    INSERT INTO {qualified_table} (lock_name, locked_at)
+                    VALUES (?, CURRENT_TIMESTAMP)
+                    """,
+                    params=["migration"],
+                )
+                return True
+            except Exception as exc:
+                self._rollback_failed_lock_attempt()
+                if not _is_lock_contention_error(exc):
+                    raise
+                if time.monotonic() >= deadline:
+                    return False
+                time.sleep(min(0.2, max(0, deadline - time.monotonic())))
+
+    def release_migration_lock(self, schema: str) -> bool:
+        """Release the CockroachDB migration lock row."""
+        if not self.table_exists(schema, self.MIGRATION_LOCK_TABLE):
+            return True
+        lock_table = self.MIGRATION_LOCK_TABLE
+        qualified_table = self.get_schema_qualified_name(schema, lock_table)
+        affected = self.execute_statement(
+            f"""
+            DELETE FROM {qualified_table}
+            WHERE lock_name = ?
+            """,
+            params=["migration"],
+        )
+        return affected > 0
+
+    def _rollback_failed_lock_attempt(self) -> None:
+        connection = getattr(self, "_connection", None)
+        if connection is not None:
+            try:
+                connection.rollback()
+            except Exception as exc:
+                message = "Could not rollback failed CockroachDB lock attempt"
+                raise RuntimeError(message) from exc
+
+
+__all__ = ["CockroachdbProvider"]

dblift-2.5.1/db/plugins/cockroachdb/quirks.py

@@ -0,0 +1,18 @@
+"""CockroachDB :class:`DialectQuirks` - inherits PostgreSQL behavior."""
+
+from __future__ import annotations
+
+from db.plugins.postgresql.quirks import PostgresqlQuirks
+
+
+class CockroachdbQuirks(PostgresqlQuirks):
+    """CockroachDB quirks, inheriting PostgreSQL behavior."""
+
+    is_ansi_reference_dialect = False
+    is_default_sqlglot_read_fallback = False
+
+    def __init__(self, dialect_name: str = "cockroachdb") -> None:
+        super().__init__(dialect_name=dialect_name)
+
+
+__all__ = ["CockroachdbQuirks"]

{dblift-2.4.2 → dblift-2.5.1}/db/plugins/postgresql/config.py

@@ -30,6 +30,8 @@ class PostgreSqlConfig(BaseDatabaseConfig):
             "yugabytedb",
             "timescaledb",
             "citus",
+            "cockroachdb",
+            "redshift",
         }
     )
 

dblift-2.5.1/db/plugins/redshift/__init__.py

@@ -0,0 +1,12 @@
+"""Redshift database provider plugin (PostgreSQL-compatible)."""
+
+__plugin_name__ = "redshift"
+__plugin_version__ = "1.0.0"
+__plugin_description__ = "Redshift database provider"
+__plugin_dialects__ = ["redshift"]
+__plugin_transport__ = "native"
+__plugin_class__ = "RedshiftProvider"
+
+from .provider import RedshiftProvider
+
+__all__ = ["RedshiftProvider"]

dblift-2.5.1/db/plugins/redshift/plugin.py

@@ -0,0 +1,26 @@
+"""Entry-point declaration for the Redshift plugin.
+
+Redshift is PostgreSQL-wire-compatible, but it needs the Redshift SQLAlchemy
+dialect/driver instead of the standard PostgreSQL dialect because Redshift does
+not implement every PostgreSQL connection bootstrap query.
+"""
+
+from __future__ import annotations
+
+from db.plugins.redshift.provider import RedshiftProvider
+from db.plugins.redshift.quirks import RedshiftQuirks
+from db.plugins.redshift.sqlalchemy_url import build_sqlalchemy_url
+from db.provider_registry import PluginInfo
+
+PLUGIN: PluginInfo = PluginInfo(
+    name="redshift",
+    version="1.0.0",
+    description="Redshift database provider",
+    dialects=["redshift"],
+    provider_class=RedshiftProvider,
+    transport="native",
+    quirks_class=RedshiftQuirks,
+    config_dialect="postgresql",
+    sqlalchemy_url_builder=build_sqlalchemy_url,
+    native_driver_module="redshift_connector",
+)

dblift-2.5.1/db/plugins/redshift/provider.py

@@ -0,0 +1,95 @@
+"""Redshift native provider (PostgreSQL-compatible)."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from db.plugins.postgresql.provider import PostgreSqlProvider
+
+
+def _is_statement_timeout_error(error: Exception) -> bool:
+    message = str(error).lower()
+    return "statement_timeout" in message or "statement timeout" in message
+
+
+class RedshiftProvider(PostgreSqlProvider):
+    """Redshift provider with Redshift-specific history and locking SQL."""
+
+    canonical_dialect_key = "redshift"
+    _migration_lock_connection: Any | None = None
+    _migration_lock_transaction: Any | None = None
+
+    def acquire_migration_lock(
+        self,
+        schema: str,
+        wait_timeout_seconds: int = 60,
+    ) -> bool:
+        """Acquire a Redshift table lock on a dedicated transaction."""
+        if getattr(self, "_migration_lock_transaction", None) is not None:
+            return True
+
+        self.create_migration_lock_table_if_not_exists(schema)
+        connection = self.engine.connect()
+        transaction = connection.begin()
+        try:
+            timeout_ms = max(1, int(wait_timeout_seconds) * 1000)
+            connection.exec_driver_sql(f"SET statement_timeout = {timeout_ms}")
+            table = self.MIGRATION_LOCK_TABLE
+            qualified_table = self.get_schema_qualified_name(schema, table)
+            connection.exec_driver_sql(f"LOCK {qualified_table}")
+        except Exception as exc:
+            try:
+                transaction.rollback()
+            finally:
+                connection.close()
+            if _is_statement_timeout_error(exc):
+                return False
+            raise
+
+        self._migration_lock_connection = connection
+        self._migration_lock_transaction = transaction
+        return True
+
+    def release_migration_lock(self, schema: str) -> bool:
+        """Release the Redshift migration lock by ending its transaction."""
+        transaction = getattr(self, "_migration_lock_transaction", None)
+        connection = getattr(self, "_migration_lock_connection", None)
+        if transaction is None or connection is None:
+            return True
+
+        try:
+            transaction.commit()
+            return True
+        except Exception:
+            try:
+                transaction.rollback()
+            except Exception:
+                pass
+            return False
+        finally:
+            try:
+                connection.close()
+            finally:
+                self._migration_lock_connection = None
+                self._migration_lock_transaction = None
+
+    def create_history_table(self, schema: str, table_name: str) -> str:
+        """Return SQL for the Redshift migration history table."""
+        qualified_table = self.get_schema_qualified_name(schema, table_name)
+        return f"""
+            CREATE TABLE IF NOT EXISTS {qualified_table} (
+                installed_rank INTEGER IDENTITY(1,1) PRIMARY KEY,
+                version VARCHAR(50),
+                description VARCHAR(200) NOT NULL,
+                type VARCHAR(20) NOT NULL,
+                script VARCHAR(1000) NOT NULL,
+                checksum VARCHAR(64),
+                installed_by VARCHAR(100) NOT NULL,
+                installed_on TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                execution_time INTEGER NOT NULL,
+                success BOOLEAN NOT NULL
+            )
+        """
+
+
+__all__ = ["RedshiftProvider"]

dblift-2.5.1/db/plugins/redshift/quirks.py

@@ -0,0 +1,33 @@
+"""Redshift :class:`DialectQuirks` - inherits PostgreSQL behavior."""
+
+from __future__ import annotations
+
+from db.plugins.postgresql.quirks import PostgresqlQuirks
+
+
+class RedshiftQuirks(PostgresqlQuirks):
+    """Redshift quirks, inheriting PostgreSQL behavior."""
+
+    is_ansi_reference_dialect = False
+    is_default_sqlglot_read_fallback = False
+
+    def __init__(self, dialect_name: str = "redshift") -> None:
+        super().__init__(dialect_name=dialect_name)
+
+    def build_snapshot_table_ddl(
+        self,
+        qualified_table: str,
+        snapshot_id_size: int,
+        checksum_size: int,
+    ) -> str:
+        """Render snapshot storage with Redshift's widest VARCHAR payload column."""
+        return (
+            f"CREATE TABLE {qualified_table} ("
+            f"snapshot_id VARCHAR({snapshot_id_size}) PRIMARY KEY, "
+            f"captured_at VARCHAR({snapshot_id_size}) NOT NULL, "
+            f"checksum VARCHAR({checksum_size}) NOT NULL, "
+            f"model_data VARCHAR(MAX) NOT NULL)"
+        )
+
+
+__all__ = ["RedshiftQuirks"]

dblift-2.5.1/db/plugins/redshift/sqlalchemy_url.py

@@ -0,0 +1,50 @@
+"""SQLAlchemy URL construction for the Redshift plugin."""
+
+from typing import Any, Dict, cast
+
+from sqlalchemy.engine import URL, make_url
+
+
+def _string_mapping(values: Any) -> Dict[str, str]:
+    if not isinstance(values, dict):
+        return {}
+    return {str(key): str(value) for key, value in values.items()}
+
+
+def _query_mapping(database_config: Any, base_query: Any = None) -> Dict[str, str]:
+    query = _string_mapping(base_query)
+    query.update(_string_mapping(getattr(database_config, "extra_params", None)))
+    query.update(_string_mapping(getattr(database_config, "options", None)))
+
+    return query
+
+
+def build_sqlalchemy_url(database_config: Any) -> str:
+    """Build a Redshift SQLAlchemy URL using the Amazon Redshift connector."""
+    raw_url = getattr(database_config, "url", None)
+    if isinstance(raw_url, str) and raw_url:
+        url = make_url(raw_url)
+        if url.drivername.startswith(("postgresql", "postgres", "redshift")):
+            url = url.set(drivername="redshift+redshift_connector")
+            username = getattr(database_config, "username", None) or url.username or None
+            password = getattr(database_config, "password", None) or url.password or None
+            if username != url.username or password != url.password:
+                url = url.set(username=username, password=password)
+            query = _query_mapping(database_config, url.query)
+            if query != dict(url.query):
+                url = url.set(query=query)
+            return cast(str, url.render_as_string(hide_password=False))
+        raise ValueError("Redshift native connections require a SQLAlchemy URL")
+
+    return cast(
+        str,
+        URL.create(
+            "redshift+redshift_connector",
+            username=getattr(database_config, "username", None) or None,
+            password=getattr(database_config, "password", None) or None,
+            host=getattr(database_config, "host", None) or "localhost",
+            port=getattr(database_config, "port", None),
+            database=getattr(database_config, "database", None),
+            query=_query_mapping(database_config),
+        ).render_as_string(hide_password=False),
+    )

{dblift-2.4.2 → dblift-2.5.1}/dblift.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dblift
-Version: 2.4.2
+Version: 2.5.1
 Summary: A database migration tool that supports multiple databases with native Python drivers
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
@@ -83,6 +83,11 @@ Provides-Extra: timescaledb
 Requires-Dist: psycopg[binary]>=3.1; extra == "timescaledb"
 Provides-Extra: citus
 Requires-Dist: psycopg[binary]>=3.1; extra == "citus"
+Provides-Extra: cockroachdb
+Requires-Dist: psycopg[binary]>=3.1; extra == "cockroachdb"
+Provides-Extra: redshift
+Requires-Dist: sqlalchemy-redshift>=1.0.0; extra == "redshift"
+Requires-Dist: redshift-connector>=2.1.0; extra == "redshift"
 Provides-Extra: fastapi
 Requires-Dist: fastapi>=0.100; extra == "fastapi"
 Provides-Extra: flask
@@ -727,6 +732,8 @@ DBLift works with these databases:
 | Database | Connection URL Example | Driver extra |
 |----------|------------------------|--------------|
 | PostgreSQL | `postgresql+psycopg://localhost:5432/mydb` | `dblift[postgresql]` |
+| CockroachDB | `postgresql+psycopg://localhost:26257/mydb` | `dblift[cockroachdb]` |
+| Redshift | `postgresql+psycopg://cluster.example.com:5439/dev` | `dblift[redshift]` |
 | SQL Server | `mssql+pymssql://localhost:1433/mydb` | `dblift[sqlserver]` |
 | Oracle | `oracle+oracledb://localhost:1521?sid=SID` | `dblift[oracle]` |
 | MySQL | `mysql+pymysql://localhost:3306/mydb` | `dblift[mysql]` |

{dblift-2.4.2 → dblift-2.5.1}/dblift.egg-info/SOURCES.txt

@@ -291,6 +291,10 @@ db/plugins/citus/__init__.py
 db/plugins/citus/plugin.py
 db/plugins/citus/provider.py
 db/plugins/citus/quirks.py
+db/plugins/cockroachdb/__init__.py
+db/plugins/cockroachdb/plugin.py
+db/plugins/cockroachdb/provider.py
+db/plugins/cockroachdb/quirks.py
 db/plugins/cosmosdb/__init__.py
 db/plugins/cosmosdb/config.py
 db/plugins/cosmosdb/plugin.py
@@ -401,6 +405,11 @@ db/plugins/postgresql/postgresql/__init__.py
 db/plugins/postgresql/postgresql/history_manager.py
 db/plugins/postgresql/postgresql/locking_manager.py
 db/plugins/postgresql/postgresql/schema_operations.py
+db/plugins/redshift/__init__.py
+db/plugins/redshift/plugin.py
+db/plugins/redshift/provider.py
+db/plugins/redshift/quirks.py
+db/plugins/redshift/sqlalchemy_url.py
 db/plugins/sqlite/__init__.py
 db/plugins/sqlite/config.py
 db/plugins/sqlite/plugin.py

{dblift-2.4.2 → dblift-2.5.1}/dblift.egg-info/entry_points.txt

@@ -5,6 +5,7 @@ dblift = cli.main:main
 alloydb = db.plugins.alloydb.plugin:PLUGIN
 aurora-postgresql = db.plugins.aurora_postgresql.plugin:PLUGIN
 citus = db.plugins.citus.plugin:PLUGIN
+cockroachdb = db.plugins.cockroachdb.plugin:PLUGIN
 cosmosdb = db.plugins.cosmosdb.plugin:PLUGIN
 db2 = db.plugins.db2.plugin:PLUGIN
 duckdb = db.plugins.duckdb.plugin:PLUGIN
@@ -13,6 +14,7 @@ mysql = db.plugins.mysql.plugin:PLUGIN
 neon = db.plugins.neon.plugin:PLUGIN
 oracle = db.plugins.oracle.plugin:PLUGIN
 postgresql = db.plugins.postgresql.plugin:PLUGIN
+redshift = db.plugins.redshift.plugin:PLUGIN
 sqlite = db.plugins.sqlite.plugin:PLUGIN
 sqlserver = db.plugins.sqlserver.plugin:PLUGIN
 supabase = db.plugins.supabase.plugin:PLUGIN

{dblift-2.4.2 → dblift-2.5.1}/dblift.egg-info/requires.txt

@@ -23,6 +23,9 @@ psycopg[binary]>=3.1
 [citus]
 psycopg[binary]>=3.1
 
+[cockroachdb]
+psycopg[binary]>=3.1
+
 [db2]
 ibm_db>=3.2
 ibm_db_sa>=0.4
@@ -88,6 +91,10 @@ opentelemetry-api>=1.20
 [postgresql]
 psycopg[binary]>=3.1
 
+[redshift]
+sqlalchemy-redshift>=1.0.0
+redshift-connector>=2.1.0
+
 [sqlserver]
 pymssql>=2.3
 

{dblift-2.4.2 → dblift-2.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dblift"
-version = "2.4.2"
+version = "2.5.1"
 description = "A database migration tool that supports multiple databases with native Python drivers"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -61,6 +61,8 @@ alloydb = "db.plugins.alloydb.plugin:PLUGIN"
 yugabytedb = "db.plugins.yugabytedb.plugin:PLUGIN"
 timescaledb = "db.plugins.timescaledb.plugin:PLUGIN"
 citus = "db.plugins.citus.plugin:PLUGIN"
+cockroachdb = "db.plugins.cockroachdb.plugin:PLUGIN"
+redshift = "db.plugins.redshift.plugin:PLUGIN"
 
 # CLI command parser extensions (third-party packages can add subcommands
 # by loading a function that receives the top-level ArgumentParser).
@@ -131,6 +133,8 @@ alloydb = ["psycopg[binary]>=3.1"]
 yugabytedb = ["psycopg[binary]>=3.1"]
 timescaledb = ["psycopg[binary]>=3.1"]
 citus = ["psycopg[binary]>=3.1"]
+cockroachdb = ["psycopg[binary]>=3.1"]
+redshift = ["sqlalchemy-redshift>=1.0.0", "redshift-connector>=2.1.0"]
 fastapi = ["fastapi>=0.100"]
 flask = ["flask>=2.3"]
 django = ["Django>=4.2"]
@@ -672,9 +676,9 @@ warn_unused_ignores = true
 strict_equality = true
 
 # strict db/plugins/<pg-compatible> — thin PostgreSQL-derived plugins (Neon,
-# Supabase, Aurora PostgreSQL, AlloyDB, YugabyteDB, TimescaleDB, Citus). Each
-# subclasses the (strict) PostgreSQL provider/quirks, so disallow_subclassing_any
-# is safe.
+# Supabase, Aurora PostgreSQL, AlloyDB, YugabyteDB, TimescaleDB, Citus,
+# CockroachDB, Redshift). Each subclasses the (strict) PostgreSQL provider/
+# quirks, so disallow_subclassing_any is safe.
 [[tool.mypy.overrides]]
 module = [
     "db.plugins.neon",
@@ -705,6 +709,15 @@ module = [
     "db.plugins.citus.plugin",
     "db.plugins.citus.provider",
     "db.plugins.citus.quirks",
+    "db.plugins.cockroachdb",
+    "db.plugins.cockroachdb.plugin",
+    "db.plugins.cockroachdb.provider",
+    "db.plugins.cockroachdb.quirks",
+    "db.plugins.redshift",
+    "db.plugins.redshift.plugin",
+    "db.plugins.redshift.provider",
+    "db.plugins.redshift.quirks",
+    "db.plugins.redshift.sqlalchemy_url",
 ]
 disallow_any_generics = true
 disallow_subclassing_any = true