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

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

TABLES = frozenset({
    "ap_actor_profiles", "ap_activities", "ap_followers", "ap_inbox_items",
    "ap_anchors", "ipfs_pins", "ap_remote_actors", "ap_following",
    "ap_remote_posts", "ap_local_posts", "ap_interactions",
    "ap_notifications", "ap_delivery_log",
})

run_alembic(context.config, MODELS, TABLES)