from alembic import context
from shared.db.alembic_env import run_alembic

MODELS = [
    "shared.models.market",
    "shared.models.market_place",
]

TABLES = frozenset({
    "products", "product_images", "product_sections", "product_labels",
    "product_stickers", "product_attributes", "product_nutrition",
    "product_allergens",
    "market_places", "nav_tops", "nav_subs",
    "listings", "listing_items",
    "link_errors", "link_externals", "subcategory_redirects", "product_logs",
})

run_alembic(context.config, MODELS, TABLES)