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

MODELS = [
    "blog.models.content",
    "shared.models.kv",
    "shared.models.menu_item",
    "shared.models.menu_node",
    "shared.models.page_config",
    "blog.models.snippet",
    "blog.models.tag_group",
]

TABLES = frozenset({
    "posts", "authors", "post_authors", "tags", "post_tags",
    "post_users",
    "snippets", "tag_groups", "tag_group_tags",
    "menu_items", "menu_nodes", "kv",
    "page_configs",
})

run_alembic(context.config, MODELS, TABLES)