Update

polymarket/api/data_client.py

@@ -0,0 +1,88 @@
+"""
+Polymarket Data API Client
+
+Provides positions, trade history, and portfolio data.
+API Documentation: https://docs.polymarket.com/developers/misc-endpoints/data-api-get-positions
+"""
+
+import requests
+from typing import Dict, Optional, List
+from datetime import datetime
+
+
+class DataClient:
+    """Client for Polymarket Data API - Positions and history"""
+    
+    BASE_URL = "https://data-api.polymarket.com"
+    
+    def __init__(self, api_key: Optional[str] = None, timeout: int = 30):
+        """
+        Initialize Data API client.
+        
+        Args:
+            api_key: Optional API key for authenticated requests
+            timeout: Request timeout in seconds
+        """
+        self.timeout = timeout
+        self.api_key = api_key
+        self.session = requests.Session()
+        self.session.headers.update({
+            'Accept': 'application/json',
+            'User-Agent': 'Polymarket-Trading-Framework/1.0'
+        })
+        
+        if api_key:
+            self.session.headers['Authorization'] = f'Bearer {api_key}'
+    
+    def _get(self, endpoint: str, params: Optional[Dict] = None) -> Dict:
+        """Make GET request with error handling"""
+        url = f"{self.BASE_URL}{endpoint}"
+        try:
+            response = self.session.get(url, params=params, timeout=self.timeout)
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.RequestException as e:
+            raise Exception(f"Data API error: {e}")
+    
+    def get_positions(self, user_address: str) -> List[Dict]:
+        """
+        Get user positions.
+        
+        Args:
+            user_address: User wallet address
+        
+        Returns:
+            List of position dictionaries
+        """
+        params = {'user': user_address}
+        return self._get('/positions', params=params)
+    
+    def get_trades(self, user_address: str, limit: int = 100) -> List[Dict]:
+        """
+        Get user trade history.
+        
+        Args:
+            user_address: User wallet address
+            limit: Maximum number of trades to return
+        
+        Returns:
+            List of trade dictionaries
+        """
+        params = {
+            'user': user_address,
+            'limit': limit
+        }
+        return self._get('/trades', params=params)
+    
+    def get_portfolio(self, user_address: str) -> Dict:
+        """
+        Get user portfolio summary.
+        
+        Args:
+            user_address: User wallet address
+        
+        Returns:
+            Portfolio dictionary with balances, positions, etc.
+        """
+        params = {'user': user_address}
+        return self._get('/portfolio', params=params)