refactor: split market data into modular components

Author: JumpiiX

tests/data/test_market_data.py

@@ -1,5 +1,4 @@
-import pytest
-from traid.data.market_data import MarketData
+from traid.data.handlers.market_data import MarketData
 
 
 def test_market_data_initialization():

traid/data/clients/__init__.py

@@ -0,0 +1,14 @@
+from typing import Dict
+import requests
+
+class KrakenClient:
+    BASE_URL = "https://api.kraken.com/0/public"
+
+    def get_ohlcv(self, symbol: str, timeframe: str) -> Dict:
+        endpoint = f"{self.BASE_URL}/OHLC"
+        params = {
+            "pair": symbol,
+            "interval": timeframe
+        }
+        response = requests.get(endpoint, params=params)
+        return response.json()

traid/data/clients/kraken_client.py

@@ -0,0 +1,20 @@
+from typing import List, Dict
+from ..clients.kraken_client import KrakenClient
+from ..models.ohlcv import OHLCV
+
+class MarketData:
+    def __init__(self, symbol: str, timeframe: str) -> None:
+        self.symbol = symbol
+        self.timeframe = timeframe
+        self._client = KrakenClient()
+
+    def get_ohlcv(self) -> List[Dict]:
+        """Fetch OHLCV data.
+        Returns:
+            List[Dict]: List of OHLCV candles
+        """
+        response = self._client.get_ohlcv(self.symbol, self.timeframe)
+        return self._parse_ohlcv_response(response)
+
+    def _parse_ohlcv_response(self, response: Dict) -> List[Dict]:
+        return []

traid/data/handlers/__init__.py

@@ -0,0 +1,21 @@
+from dataclasses import dataclass
+from typing import Dict
+
+@dataclass
+class OHLCV:
+    timestamp: int
+    open: float
+    high: float
+    low: float
+    close: float
+    volume: float
+
+    def to_dict(self) -> Dict:
+        return {
+            'timestamp': self.timestamp,
+            'open': self.open,
+            'high': self.high,
+            'low': self.low,
+            'close': self.close,
+            'volume': self.volume
+        }