REFACTOR data service

shared/clients/sales_client.py

@@ -0,0 +1,156 @@
+# shared/clients/sales_client.py
+"""
+Sales Service Client
+Handles all API calls to the sales service
+"""
+
+import httpx
+import structlog
+from typing import Dict, Any, Optional, List, Union
+from .base_service_client import BaseServiceClient
+from shared.config.base import BaseServiceSettings
+
+logger = structlog.get_logger()
+
+
+class SalesServiceClient(BaseServiceClient):
+    """Client for communicating with the sales service"""
+    
+    def __init__(self, config: BaseServiceSettings, calling_service_name: str = "unknown"):
+        super().__init__(calling_service_name, config)
+        self.service_url = config.SALES_SERVICE_URL
+    
+    def get_service_base_path(self) -> str:
+        return "/api/v1"
+    
+    # ================================================================
+    # SALES DATA (with advanced pagination support)
+    # ================================================================
+    
+    async def get_sales_data(
+        self,
+        tenant_id: str,
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+        product_id: Optional[str] = None,
+        aggregation: str = "daily"
+    ) -> Optional[List[Dict[str, Any]]]:
+        """Get sales data for a date range"""
+        params = {"aggregation": aggregation}
+        if start_date:
+            params["start_date"] = start_date
+        if end_date:
+            params["end_date"] = end_date
+        if product_id:
+            params["product_id"] = product_id
+        
+        result = await self.get("sales", tenant_id=tenant_id, params=params)
+        return result.get("sales", []) if result else None
+    
+    async def get_all_sales_data(
+        self,
+        tenant_id: str,
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+        product_id: Optional[str] = None,
+        aggregation: str = "daily",
+        page_size: int = 1000,
+        max_pages: int = 100
+    ) -> List[Dict[str, Any]]:
+        """
+        Get ALL sales data using pagination (equivalent to original fetch_sales_data)
+        Retrieves all records without pagination limits
+        """
+        params = {"aggregation": aggregation}
+        if start_date:
+            params["start_date"] = start_date
+        if end_date:
+            params["end_date"] = end_date
+        if product_id:
+            params["product_id"] = product_id
+        
+        # Use the inherited paginated request method
+        try:
+            all_records = await self.get_paginated(
+                "sales",
+                tenant_id=tenant_id,
+                params=params,
+                page_size=page_size,
+                max_pages=max_pages,
+                timeout=2000.0
+            )
+            
+            logger.info(f"Successfully fetched {len(all_records)} total sales records via sales service", 
+                       tenant_id=tenant_id)
+            return all_records
+            
+        except Exception as e:
+            logger.error(f"Failed to fetch paginated sales data: {e}")
+            return []
+    
+    async def upload_sales_data(
+        self,
+        tenant_id: str,
+        sales_data: List[Dict[str, Any]]
+    ) -> Optional[Dict[str, Any]]:
+        """Upload sales data"""
+        data = {"sales": sales_data}
+        return await self.post("sales", data=data, tenant_id=tenant_id)
+    
+    # ================================================================
+    # PRODUCTS
+    # ================================================================
+    
+    async def get_products(self, tenant_id: str) -> Optional[List[Dict[str, Any]]]:
+        """Get all products for a tenant"""
+        result = await self.get("products", tenant_id=tenant_id)
+        return result.get("products", []) if result else None
+    
+    async def get_product(self, tenant_id: str, product_id: str) -> Optional[Dict[str, Any]]:
+        """Get a specific product"""
+        return await self.get(f"products/{product_id}", tenant_id=tenant_id)
+    
+    async def create_product(
+        self,
+        tenant_id: str,
+        name: str,
+        category: str,
+        price: float,
+        **kwargs
+    ) -> Optional[Dict[str, Any]]:
+        """Create a new product"""
+        data = {
+            "name": name,
+            "category": category,
+            "price": price,
+            **kwargs
+        }
+        return await self.post("products", data=data, tenant_id=tenant_id)
+    
+    async def update_product(
+        self,
+        tenant_id: str,
+        product_id: str,
+        **updates
+    ) -> Optional[Dict[str, Any]]:
+        """Update a product"""
+        return await self.put(f"products/{product_id}", data=updates, tenant_id=tenant_id)
+    
+    # ================================================================
+    # DATA IMPORT
+    # ================================================================
+    
+    async def import_sales_data(
+        self,
+        tenant_id: str,
+        file_content: str,
+        file_format: str,
+        filename: Optional[str] = None
+    ) -> Optional[Dict[str, Any]]:
+        """Import sales data from CSV/Excel/JSON"""
+        data = {
+            "content": file_content,
+            "format": file_format,
+            "filename": filename
+        }
+        return await self.post("import", data=data, tenant_id=tenant_id)