"""rename_gravity_columns

Renames 3 gravity columns in musehub_symbol_intel to match
muse code gravity --json field names exactly:

    gravity_direct     → gravity_direct_dependents
    gravity_transitive → gravity_transitive_dependents
    gravity_depth_dist → gravity_depth_distribution

Revision ID: 0006
Revises: 0005
Create Date: 2026-05-03 00:00:00.000000+00:00
"""
from __future__ import annotations

from typing import Sequence, Union

from alembic import op

revision: str = '0006'
down_revision: Union[str, None] = '0005'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.alter_column('musehub_symbol_intel', 'gravity_direct',
                    new_column_name='gravity_direct_dependents')
    op.alter_column('musehub_symbol_intel', 'gravity_transitive',
                    new_column_name='gravity_transitive_dependents')
    op.alter_column('musehub_symbol_intel', 'gravity_depth_dist',
                    new_column_name='gravity_depth_distribution')


def downgrade() -> None:
    op.alter_column('musehub_symbol_intel', 'gravity_direct_dependents',
                    new_column_name='gravity_direct')
    op.alter_column('musehub_symbol_intel', 'gravity_transitive_dependents',
                    new_column_name='gravity_transitive')
    op.alter_column('musehub_symbol_intel', 'gravity_depth_distribution',
                    new_column_name='gravity_depth_dist')