mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-07 00:16:57 +08:00
fix(db): repair episode_priority column type mismatch on PostgreSQL (#5892)
This commit is contained in:
@@ -22,14 +22,31 @@ def _has_column(inspector: sa.Inspector, table_name: str, column_name: str) -> b
|
||||
return any(column["name"] == column_name for column in inspector.get_columns(table_name))
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
inspector = sa.inspect(op.get_bind())
|
||||
if _has_column(inspector, "subscribe", "episode_priority") is False:
|
||||
op.add_column("subscribe", sa.Column("episode_priority", sa.JSON(), nullable=True))
|
||||
def _ensure_json_column(bind, table_name: str, column_name: str) -> None:
|
||||
"""Add the column as JSON, or fix it if it already exists with the wrong type (PostgreSQL only)."""
|
||||
inspector = sa.inspect(bind)
|
||||
if not _has_column(inspector, table_name, column_name):
|
||||
op.add_column(table_name, sa.Column(column_name, sa.JSON(), nullable=True))
|
||||
return
|
||||
if bind.dialect.name != "postgresql":
|
||||
return
|
||||
for col in inspector.get_columns(table_name):
|
||||
if col["name"] != column_name:
|
||||
continue
|
||||
type_name = type(col["type"]).__name__.upper()
|
||||
if type_name in ("JSON", "JSONB"):
|
||||
return
|
||||
# Column exists with wrong type (e.g., INTEGER from an intermediate build); replace it.
|
||||
# Existing values are non-functional garbage so data loss is acceptable.
|
||||
op.drop_column(table_name, column_name)
|
||||
op.add_column(table_name, sa.Column(column_name, sa.JSON(), nullable=True))
|
||||
return
|
||||
|
||||
inspector = sa.inspect(op.get_bind())
|
||||
if _has_column(inspector, "subscribehistory", "episode_priority") is False:
|
||||
op.add_column("subscribehistory", sa.Column("episode_priority", sa.JSON(), nullable=True))
|
||||
|
||||
def upgrade() -> None:
|
||||
bind = op.get_bind()
|
||||
_ensure_json_column(bind, "subscribe", "episode_priority")
|
||||
_ensure_json_column(bind, "subscribehistory", "episode_priority")
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
|
||||
Reference in New Issue
Block a user