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

MODELS = [
    "shared.models.sx_pub",
]

TABLES = frozenset({
    "sx_pub_actor",
    "sx_pub_collections",
    "sx_pub_documents",
    "sx_pub_activities",
    "sx_pub_followers",
    "sx_pub_following",
})

run_alembic(context.config, MODELS, TABLES)