Initial public release

hydra/supplemental/events.py

@@ -0,0 +1,73 @@
+# hydra/supplemental/events.py
+import asyncio
+import json
+import httpx
+from hydra.logging.setup import get_logger
+
+logger = get_logger(__name__)
+
+_EVENTS_KEY = "hydra:events:upcoming"
+_API_URL = "https://api.coinmarketcal.com/v1/events"
+_DEFAULT_INTERVAL = 3600
+
+
+class EventCalendarPoller:
+    def __init__(self, redis_client, api_key: str = "",
+                 interval_sec: int = _DEFAULT_INTERVAL):
+        self._redis = redis_client
+        self._api_key = api_key
+        self._interval = interval_sec
+
+    async def _fetch(self) -> list[dict]:
+        if not self._api_key:
+            return []
+        try:
+            async with httpx.AsyncClient(timeout=15) as client:
+                resp = await client.get(
+                    _API_URL,
+                    headers={"x-api-key": self._api_key},
+                    params={"max": 10, "dateRangeEnd": "+24h"},
+                )
+                resp.raise_for_status()
+                body = resp.json()
+                events = []
+                for item in body.get("body", []):
+                    coins = item.get("coins") or []
+                    symbol = coins[0].get("symbol", "") if coins else ""
+                    events.append({
+                        "title": item.get("title", ""),
+                        "symbol": symbol,
+                        "date_event": item.get("date_event", ""),
+                        "source": "coinmarketcal",
+                    })
+                return events
+        except Exception as e:
+            logger.warning("events_fetch_error", error=str(e))
+            return []
+
+    async def run(self) -> None:
+        logger.info("events_poller_started", interval=self._interval,
+                    has_key=bool(self._api_key))
+        while True:
+            events = await self._fetch()
+            await self._redis.set(_EVENTS_KEY, json.dumps(events))
+            logger.debug("events_cached", count=len(events))
+            await asyncio.sleep(self._interval)
+
+
+async def main() -> None:
+    import os
+    import redis.asyncio as aioredis
+    from hydra.config.settings import get_settings
+    settings = get_settings()
+    r = aioredis.from_url(settings.redis_url, decode_responses=True)
+    api_key = os.environ.get("COINMARKETCAL_API_KEY", "")
+    poller = EventCalendarPoller(r, api_key=api_key)
+    try:
+        await poller.run()
+    finally:
+        await r.aclose()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())