Upload ml/02_seed_events.py with huggingface_hub

ml/02_seed_events.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+"""
+Stage 2: Seed the historical_events table from config.
+
+Idempotent — uses ON CONFLICT to upsert.
+"""
+
+import json
+import logging
+import sys
+from datetime import date
+
+from config import HISTORICAL_EVENTS
+from db import get_conn
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s  %(levelname)-8s  %(message)s",
+    handlers=[logging.StreamHandler(sys.stdout)],
+)
+log = logging.getLogger(__name__)
+
+
+def seed_events():
+    conn = get_conn()
+    with conn.cursor() as cur:
+        for event in HISTORICAL_EVENTS:
+            cur.execute(
+                """INSERT INTO historical_events
+                   (event_name, start_date, end_date, category, keywords)
+                   VALUES (%s, %s, %s, %s, %s)
+                   ON CONFLICT (event_name) DO UPDATE SET
+                     start_date = EXCLUDED.start_date,
+                     end_date = EXCLUDED.end_date,
+                     category = EXCLUDED.category,
+                     keywords = EXCLUDED.keywords
+                """,
+                (
+                    event["name"],
+                    date.fromisoformat(event["start"]),
+                    date.fromisoformat(event["end"]) if event.get("end") else None,
+                    event.get("category"),
+                    json.dumps(event.get("keywords", [])),
+                ),
+            )
+            log.info(f"  Seeded: {event['name']}")
+
+    conn.commit()
+    conn.close()
+    log.info(f"Seeded {len(HISTORICAL_EVENTS)} historical events.")
+
+
+if __name__ == "__main__":
+    seed_events()