yuxuanhui
7f93d6b0f5
- Introduced ActiveMoneyflowSource to fetch optional active-order flow, enhancing the sector radar's data capabilities. - Updated StockFactRecord and DailyAggregateRecord to include pct_change and active_buy_net_amount_yuan for improved financial insights. - Modified the build process to incorporate active moneyflow data without invalidating main rankings on failure. - Enhanced the HTTP API to return detailed sector history and metrics, including pct_change and active buy metrics for members. - Updated tests to validate the new functionality and ensure data integrity across various scenarios.
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
"""Preserve independent stock detail facts without invalidating old publications."""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "0008_radar_detail"
|
|
down_revision: str | None = "0007_selection_pattern_scoring"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Add nullable facts and allow the optional raw moneyflow checkpoint."""
|
|
for name in ("pct_change", "active_buy_net_amount_yuan"):
|
|
op.add_column("sector_radar_stock_fact", sa.Column(name, sa.Numeric(), nullable=True))
|
|
op.create_check_constraint(
|
|
f"ck_radar_stock_{name}_finite",
|
|
"sector_radar_stock_fact",
|
|
f"{name} IS NULL OR {name} NOT IN "
|
|
"('NaN'::numeric, 'Infinity'::numeric, '-Infinity'::numeric)",
|
|
)
|
|
op.drop_constraint(
|
|
"ck_sector_radar_publication_source_group", "sector_radar_publication_source"
|
|
)
|
|
op.create_check_constraint(
|
|
"ck_sector_radar_publication_source_group",
|
|
"sector_radar_publication_source",
|
|
"source_group IN ('calendar', 'concept_indices', 'industry_indices', 'members', "
|
|
"'stock_basics', 'suspensions', 'daily', 'moneyflow_dc', 'moneyflow')",
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Drop added fact columns; retain optional checkpoint audit rows for older readers."""
|
|
for name in ("pct_change", "active_buy_net_amount_yuan"):
|
|
op.drop_column("sector_radar_stock_fact", name)
|