Imporve enterprise

services/distribution/app/repositories/delivery_route_repository.py

@@ -231,4 +231,82 @@ class DeliveryRouteRepository:
         await self.db_session.commit()
 
         deleted_count = result.rowcount
-        return deleted_count
+        return deleted_count
+
+    async def update_route_vrp_metrics(self, route_id: str, vrp_metrics: Dict[str, Any]) -> Optional[Dict[str, Any]]:
+        """
+        Update VRP optimization metrics for a route
+        """
+        stmt = select(DeliveryRoute).where(DeliveryRoute.id == route_id)
+        result = await self.db_session.execute(stmt)
+        route = result.scalar_one_or_none()
+
+        if not route:
+            return None
+
+        # Update VRP metrics fields
+        route.vrp_optimization_savings = vrp_metrics.get('vrp_optimization_savings')
+        route.vrp_algorithm_version = vrp_metrics.get('vrp_algorithm_version')
+        route.vrp_optimization_timestamp = vrp_metrics.get('vrp_optimization_timestamp')
+        route.vrp_constraints_satisfied = vrp_metrics.get('vrp_constraints_satisfied')
+        route.vrp_objective_value = vrp_metrics.get('vrp_objective_value')
+
+        await self.db_session.commit()
+        await self.db_session.refresh(route)
+
+        return {
+            'id': str(route.id),
+            'vrp_optimization_savings': route.vrp_optimization_savings,
+            'vrp_algorithm_version': route.vrp_algorithm_version,
+            'vrp_optimization_timestamp': route.vrp_optimization_timestamp,
+            'vrp_constraints_satisfied': route.vrp_constraints_satisfied,
+            'vrp_objective_value': route.vrp_objective_value
+        }
+
+    async def get_routes_by_tenant(self, tenant_id: str, limit: int = None, offset: int = None, order_by: str = None) -> List[Dict[str, Any]]:
+        """
+        Get all routes for a specific tenant with pagination and ordering
+        """
+        stmt = select(DeliveryRoute).where(DeliveryRoute.tenant_id == tenant_id)
+        
+        # Apply ordering if specified
+        if order_by:
+            if 'vrp_optimization_timestamp' in order_by:
+                if 'DESC' in order_by:
+                    stmt = stmt.order_by(DeliveryRoute.vrp_optimization_timestamp.desc())
+                else:
+                    stmt = stmt.order_by(DeliveryRoute.vrp_optimization_timestamp.asc())
+            elif 'route_date' in order_by:
+                if 'DESC' in order_by:
+                    stmt = stmt.order_by(DeliveryRoute.route_date.desc())
+                else:
+                    stmt = stmt.order_by(DeliveryRoute.route_date.asc())
+        
+        # Apply pagination if specified
+        if limit is not None:
+            stmt = stmt.limit(limit)
+        if offset is not None:
+            stmt = stmt.offset(offset)
+        
+        result = await self.db_session.execute(stmt)
+        routes = result.scalars().all()
+        
+        return [{
+            'id': str(route.id),
+            'tenant_id': str(route.tenant_id),
+            'route_number': route.route_number,
+            'route_date': route.route_date,
+            'vehicle_id': route.vehicle_id,
+            'driver_id': route.driver_id,
+            'total_distance_km': route.total_distance_km,
+            'estimated_duration_minutes': route.estimated_duration_minutes,
+            'route_sequence': route.route_sequence,
+            'status': route.status.value if hasattr(route.status, 'value') else route.status,
+            'created_at': route.created_at,
+            'updated_at': route.updated_at,
+            'vrp_optimization_savings': route.vrp_optimization_savings,
+            'vrp_algorithm_version': route.vrp_algorithm_version,
+            'vrp_optimization_timestamp': route.vrp_optimization_timestamp,
+            'vrp_constraints_satisfied': route.vrp_constraints_satisfied,
+            'vrp_objective_value': route.vrp_objective_value
+        } for route in routes]