feat: implement CNN Fear & Greed Index collector

2 files changed, 111 insertions, 0 deletions

diff --git a/services/news-collector/src/news_collector/collectors/fear_greed.py b/services/news-collector/src/news_collector/collectors/fear_greed.py
new file mode 100644
index 0000000..305d416
--- /dev/null
+++ b/services/news-collector/src/news_collector/collectors/fear_greed.py
@@ -0,0 +1,62 @@
+"""CNN Fear & Greed Index collector."""
+
+import logging
+from dataclasses import dataclass
+from typing import Optional
+
+import aiohttp
+
+from news_collector.collectors.base import BaseCollector
+from shared.models import NewsItem
+
+logger = logging.getLogger(__name__)
+
+FEAR_GREED_URL = "https://production.dataviz.cnn.io/index/fearandgreed/graphdata"
+
+
+@dataclass
+class FearGreedResult:
+    fear_greed: int
+    fear_greed_label: str
+
+
+class FearGreedCollector(BaseCollector):
+    name = "fear_greed"
+    poll_interval = 3600  # 1 hour
+
+    async def is_available(self) -> bool:
+        return True
+
+    async def _fetch_index(self) -> Optional[dict]:
+        headers = {"User-Agent": "Mozilla/5.0"}
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.get(FEAR_GREED_URL, headers=headers, timeout=aiohttp.ClientTimeout(total=10)) as resp:
+                    if resp.status != 200:
+                        return None
+                    return await resp.json()
+        except Exception:
+            return None
+
+    def _classify(self, score: int) -> str:
+        if score <= 20:
+            return "Extreme Fear"
+        if score <= 40:
+            return "Fear"
+        if score <= 60:
+            return "Neutral"
+        if score <= 80:
+            return "Greed"
+        return "Extreme Greed"
+
+    async def collect(self) -> Optional[FearGreedResult]:
+        data = await self._fetch_index()
+        if data is None:
+            return None
+        try:
+            fg = data["fear_and_greed"]
+            score = int(fg["score"])
+            label = fg.get("rating", self._classify(score))
+            return FearGreedResult(fear_greed=score, fear_greed_label=label)
+        except (KeyError, ValueError, TypeError):
+            return None
diff --git a/services/news-collector/tests/test_fear_greed.py b/services/news-collector/tests/test_fear_greed.py
new file mode 100644
index 0000000..d483aa6
--- /dev/null
+++ b/services/news-collector/tests/test_fear_greed.py
@@ -0,0 +1,49 @@
+"""Tests for CNN Fear & Greed Index collector."""
+
+import pytest
+from unittest.mock import AsyncMock, patch
+
+from news_collector.collectors.fear_greed import FearGreedCollector
+
+
+@pytest.fixture
+def collector():
+    return FearGreedCollector()
+
+
+def test_collector_name(collector):
+    assert collector.name == "fear_greed"
+    assert collector.poll_interval == 3600
+
+
+async def test_is_available(collector):
+    assert await collector.is_available() is True
+
+
+async def test_collect_parses_api_response(collector):
+    mock_data = {
+        "fear_and_greed": {
+            "score": 45.0,
+            "rating": "Fear",
+            "timestamp": "2026-04-02T12:00:00+00:00",
+        }
+    }
+    with patch.object(collector, "_fetch_index", new_callable=AsyncMock, return_value=mock_data):
+        result = await collector.collect()
+    assert result.fear_greed == 45
+    assert result.fear_greed_label == "Fear"
+
+
+async def test_collect_returns_none_on_failure(collector):
+    with patch.object(collector, "_fetch_index", new_callable=AsyncMock, return_value=None):
+        result = await collector.collect()
+    assert result is None
+
+
+def test_classify_label():
+    c = FearGreedCollector()
+    assert c._classify(10) == "Extreme Fear"
+    assert c._classify(30) == "Fear"
+    assert c._classify(50) == "Neutral"
+    assert c._classify(70) == "Greed"
+    assert c._classify(85) == "Extreme Greed"