gabriel / musehub public
0005_gravity_columns.py python
60 lines 2.3 KB
Raw
sha256:1ccb8409daa5aabe577bd26d11b56ed12f3376d64011d0e75a247e81211a66ee docs(mwp4/phase5): tick Phase 5 checkboxes, close musehub#109 Sonnet 4.6 21 days ago
1 """gravity_columns
2
3 Extends musehub_symbol_intel with 6 new columns that power /intel/gravity:
4
5 gravity_pct FLOAT — gravity_pct from muse code gravity
6 gravity_direct INTEGER — direct_dependents count
7 gravity_transitive INTEGER — transitive_dependents count
8 gravity_max_depth SMALLINT — deepest dependency chain
9 gravity_depth_dist JSONB — {depth_level: count} for the sparkline
10 symbol_kind VARCHAR(64) — method/function/class/async_method
11
12 New index: ix_symbol_intel_repo_gravity_pct on (repo_id, gravity_pct DESC)
13 — primary sort key for the /intel/gravity page query.
14
15 Revision ID: 0005
16 Revises: 0004
17 Create Date: 2026-05-03 00:00:00.000000+00:00
18 """
19 from __future__ import annotations
20
21 from typing import Sequence, Union
22
23 import sqlalchemy as sa
24 from alembic import op
25
26 revision: str = '0005'
27 down_revision: Union[str, None] = '0004'
28 branch_labels: Union[str, Sequence[str], None] = None
29 depends_on: Union[str, Sequence[str], None] = None
30
31
32 def upgrade() -> None:
33 op.add_column('musehub_symbol_intel',
34 sa.Column('gravity_pct', sa.Float, nullable=True))
35 op.add_column('musehub_symbol_intel',
36 sa.Column('gravity_direct', sa.Integer, nullable=True))
37 op.add_column('musehub_symbol_intel',
38 sa.Column('gravity_transitive', sa.Integer, nullable=True))
39 op.add_column('musehub_symbol_intel',
40 sa.Column('gravity_max_depth', sa.SmallInteger, nullable=True))
41 op.add_column('musehub_symbol_intel',
42 sa.Column('gravity_depth_dist', sa.JSON, nullable=True))
43 op.add_column('musehub_symbol_intel',
44 sa.Column('symbol_kind', sa.String(64), nullable=True))
45
46 op.create_index(
47 'ix_symbol_intel_repo_gravity_pct',
48 'musehub_symbol_intel',
49 ['repo_id', sa.text('gravity_pct DESC NULLS LAST')],
50 )
51
52
53 def downgrade() -> None:
54 op.drop_index('ix_symbol_intel_repo_gravity_pct', table_name='musehub_symbol_intel')
55 op.drop_column('musehub_symbol_intel', 'symbol_kind')
56 op.drop_column('musehub_symbol_intel', 'gravity_depth_dist')
57 op.drop_column('musehub_symbol_intel', 'gravity_max_depth')
58 op.drop_column('musehub_symbol_intel', 'gravity_transitive')
59 op.drop_column('musehub_symbol_intel', 'gravity_direct')
60 op.drop_column('musehub_symbol_intel', 'gravity_pct')
File History 2 commits
sha256:1ccb8409daa5aabe577bd26d11b56ed12f3376d64011d0e75a247e81211a66ee docs(mwp4/phase5): tick Phase 5 checkboxes, close musehub#109 Sonnet 4.6 21 days ago
sha256:2c523da45351334b5c4dbefed4dc3dd553b3faa8737a4e6caf301e5dc82141be test(mwp4): Phase 0 RED reproduction tests for RC-4 ordering race Sonnet 4.6 21 days ago