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

MODELS = [
    "shared.models.user",
    "shared.models.ghost_membership_entities",
    "shared.models.magic_link",
    "shared.models.oauth_code",
    "shared.models.oauth_grant",
]

TABLES = frozenset({
    "users", "user_labels", "user_newsletters",
    "magic_links", "oauth_codes", "oauth_grants",
    "ghost_labels", "ghost_newsletters", "ghost_tiers", "ghost_subscriptions",
})

run_alembic(context.config, MODELS, TABLES)