mirror of
https://github.com/R0m1k3/Priceflow.git
synced 2026-10-20 13:48:16 +02:00
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
"""Add missing columns to search_sites table
|
|
|
|
Revision ID: add_search_sites_columns
|
|
Revises: add_notification_channels
|
|
Create Date: 2025-11-24
|
|
|
|
"""
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy import inspect
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "add_search_sites_columns"
|
|
down_revision: str | None = "add_notification_channels"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def column_exists(table_name: str, column_name: str) -> bool:
|
|
"""Check if a column exists in a table."""
|
|
bind = op.get_bind()
|
|
inspector = inspect(bind)
|
|
columns = [col["name"] for col in inspector.get_columns(table_name)]
|
|
return column_name in columns
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Add debug_enabled, price_selector, search_url, product_link_selector columns to search_sites."""
|
|
# Add debug_enabled column (only if not exists)
|
|
if not column_exists("search_sites", "debug_enabled"):
|
|
op.add_column(
|
|
"search_sites",
|
|
sa.Column("debug_enabled", sa.Boolean(), server_default="false", nullable=False),
|
|
)
|
|
|
|
# Add price_selector column (only if not exists)
|
|
if not column_exists("search_sites", "price_selector"):
|
|
op.add_column(
|
|
"search_sites",
|
|
sa.Column("price_selector", sa.String(), nullable=True),
|
|
)
|
|
|
|
# Add search_url column (only if not exists)
|
|
if not column_exists("search_sites", "search_url"):
|
|
op.add_column(
|
|
"search_sites",
|
|
sa.Column("search_url", sa.String(), nullable=True),
|
|
)
|
|
|
|
# Add product_link_selector column (only if not exists)
|
|
if not column_exists("search_sites", "product_link_selector"):
|
|
op.add_column(
|
|
"search_sites",
|
|
sa.Column("product_link_selector", sa.String(), nullable=True),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Remove added columns from search_sites."""
|
|
if column_exists("search_sites", "product_link_selector"):
|
|
op.drop_column("search_sites", "product_link_selector")
|
|
if column_exists("search_sites", "search_url"):
|
|
op.drop_column("search_sites", "search_url")
|
|
if column_exists("search_sites", "price_selector"):
|
|
op.drop_column("search_sites", "price_selector")
|
|
if column_exists("search_sites", "debug_enabled"):
|
|
op.drop_column("search_sites", "debug_enabled")
|