0006_rename_gravity_columns.py
file-level
1
files
1
commits
0
hotspots
0
🧊 dead
0
💥 blast risk
| 1 | """rename_gravity_columns |
| 2 | |
| 3 | Renames 3 gravity columns in musehub_symbol_intel to match |
| 4 | muse code gravity --json field names exactly: |
| 5 | |
| 6 | gravity_direct → gravity_direct_dependents |
| 7 | gravity_transitive → gravity_transitive_dependents |
| 8 | gravity_depth_dist → gravity_depth_distribution |
| 9 | |
| 10 | Revision ID: 0006 |
| 11 | Revises: 0005 |
| 12 | Create Date: 2026-05-03 00:00:00.000000+00:00 |
| 13 | """ |
| 14 | from __future__ import annotations |
| 15 | |
| 16 | from typing import Sequence, Union |
| 17 | |
| 18 | from alembic import op |
| 19 | |
| 20 | revision: str = '0006' |
| 21 | down_revision: Union[str, None] = '0005' |
| 22 | branch_labels: Union[str, Sequence[str], None] = None |
| 23 | depends_on: Union[str, Sequence[str], None] = None |
| 24 | |
| 25 | |
| 26 | def upgrade() -> None: |
| 27 | op.alter_column('musehub_symbol_intel', 'gravity_direct', |
| 28 | new_column_name='gravity_direct_dependents') |
| 29 | op.alter_column('musehub_symbol_intel', 'gravity_transitive', |
| 30 | new_column_name='gravity_transitive_dependents') |
| 31 | op.alter_column('musehub_symbol_intel', 'gravity_depth_dist', |
| 32 | new_column_name='gravity_depth_distribution') |
| 33 | |
| 34 | |
| 35 | def downgrade() -> None: |
| 36 | op.alter_column('musehub_symbol_intel', 'gravity_direct_dependents', |
| 37 | new_column_name='gravity_direct') |
| 38 | op.alter_column('musehub_symbol_intel', 'gravity_transitive_dependents', |
| 39 | new_column_name='gravity_transitive') |
| 40 | op.alter_column('musehub_symbol_intel', 'gravity_depth_distribution', |
| 41 | new_column_name='gravity_depth_dist') |