4c86a5b9ca
Self-hosted US Congress monitoring platform with AI policy briefs, bill/member/topic follows, ntfy + RSS + email notifications, alignment scoring, collections, and draft-letter generator. Authored by: Jack Levy
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
"""backfill bill congress_urls with proper public URLs
|
|
|
|
Bills stored before this fix have congress_url set to the API endpoint
|
|
(https://api.congress.gov/v3/bill/...) instead of the public page
|
|
(https://www.congress.gov/bill/...). This migration rebuilds all URLs
|
|
from the congress_number, bill_type, and bill_number columns which are
|
|
already stored correctly.
|
|
|
|
Revision ID: 0010
|
|
Revises: 0009
|
|
Create Date: 2026-03-01
|
|
"""
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision = "0010"
|
|
down_revision = "0009"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
_BILL_TYPE_SLUG = {
|
|
"hr": "house-bill",
|
|
"s": "senate-bill",
|
|
"hjres": "house-joint-resolution",
|
|
"sjres": "senate-joint-resolution",
|
|
"hres": "house-resolution",
|
|
"sres": "senate-resolution",
|
|
"hconres": "house-concurrent-resolution",
|
|
"sconres": "senate-concurrent-resolution",
|
|
}
|
|
|
|
|
|
def _ordinal(n: int) -> str:
|
|
if 11 <= n % 100 <= 13:
|
|
return f"{n}th"
|
|
suffixes = {1: "st", 2: "nd", 3: "rd"}
|
|
return f"{n}{suffixes.get(n % 10, 'th')}"
|
|
|
|
|
|
def upgrade():
|
|
conn = op.get_bind()
|
|
bills = conn.execute(
|
|
sa.text("SELECT bill_id, congress_number, bill_type, bill_number FROM bills")
|
|
).fetchall()
|
|
for bill in bills:
|
|
slug = _BILL_TYPE_SLUG.get(bill.bill_type, bill.bill_type)
|
|
url = f"https://www.congress.gov/bill/{_ordinal(bill.congress_number)}-congress/{slug}/{bill.bill_number}"
|
|
conn.execute(
|
|
sa.text("UPDATE bills SET congress_url = :url WHERE bill_id = :bill_id"),
|
|
{"url": url, "bill_id": bill.bill_id},
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
# Original API URLs cannot be recovered — no-op
|
|
pass
|