binalyze-air-sdk 1.0.2__py3-none-any.whl → 1.0.3__py3-none-any.whl

binalyze_air/commands/baseline.py

@@ -1,396 +1,32 @@
-"""
-Baseline-related commands for the Binalyze AIR SDK.
-"""
-
-from typing import Dict, Any, Union
-
-from ..base import Command
-from ..models.baseline import (
-    Baseline, BaselineProfile, BaselineComparison, BaselineSchedule, BaselineChange,
-    CreateBaselineRequest, UpdateBaselineRequest, CreateBaselineProfileRequest, 
-    CompareBaselineRequest
-)
-from ..http_client import HTTPClient
-
-
-class CreateBaselineCommand(Command[Baseline]):
-    """Command to create a new baseline."""
-    
-    def __init__(self, http_client: HTTPClient, request: CreateBaselineRequest):
-        self.http_client = http_client
-        self.request = request
-    
-    def execute(self) -> Baseline:
-        """Execute the command to create a baseline."""
-        data = self.request.model_dump(exclude_none=True)
-        
-        response = self.http_client.post("baselines", json_data=data)
-        
-        entity_data = response.get("result", {})
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "name": entity_data.get("name"),
-            "description": entity_data.get("description"),
-            "type": entity_data.get("type"),
-            "status": entity_data.get("status", "creating"),
-            "endpoint_id": entity_data.get("endpointId"),
-            "endpoint_name": entity_data.get("endpointName"),
-            "organization_id": entity_data.get("organizationId", 0),
-            "created_at": entity_data.get("createdAt"),
-            "updated_at": entity_data.get("updatedAt"),
-            "created_by": entity_data.get("createdBy"),
-            "item_count": entity_data.get("itemCount", 0),
-            "file_count": entity_data.get("fileCount", 0),
-            "registry_count": entity_data.get("registryCount", 0),
-            "service_count": entity_data.get("serviceCount", 0),
-            "process_count": entity_data.get("processCount", 0),
-            "network_connection_count": entity_data.get("networkConnectionCount", 0),
-            "tags": entity_data.get("tags", []),
-            "profile_id": entity_data.get("profileId"),
-            "profile_name": entity_data.get("profileName"),
-            "last_comparison": entity_data.get("lastComparison"),
-            "comparison_count": entity_data.get("comparisonCount", 0),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return Baseline(**mapped_data)
-
-
-class UpdateBaselineCommand(Command[Baseline]):
-    """Command to update an existing baseline."""
-    
-    def __init__(self, http_client: HTTPClient, baseline_id: str, request: UpdateBaselineRequest):
-        self.http_client = http_client
-        self.baseline_id = baseline_id
-        self.request = request
-    
-    def execute(self) -> Baseline:
-        """Execute the command to update a baseline."""
-        data = self.request.model_dump(exclude_none=True)
-        
-        response = self.http_client.put(f"baselines/{self.baseline_id}", json_data=data)
-        
-        entity_data = response.get("result", {})
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "name": entity_data.get("name"),
-            "description": entity_data.get("description"),
-            "type": entity_data.get("type"),
-            "status": entity_data.get("status", "creating"),
-            "endpoint_id": entity_data.get("endpointId"),
-            "endpoint_name": entity_data.get("endpointName"),
-            "organization_id": entity_data.get("organizationId", 0),
-            "created_at": entity_data.get("createdAt"),
-            "updated_at": entity_data.get("updatedAt"),
-            "created_by": entity_data.get("createdBy"),
-            "item_count": entity_data.get("itemCount", 0),
-            "file_count": entity_data.get("fileCount", 0),
-            "registry_count": entity_data.get("registryCount", 0),
-            "service_count": entity_data.get("serviceCount", 0),
-            "process_count": entity_data.get("processCount", 0),
-            "network_connection_count": entity_data.get("networkConnectionCount", 0),
-            "tags": entity_data.get("tags", []),
-            "profile_id": entity_data.get("profileId"),
-            "profile_name": entity_data.get("profileName"),
-            "last_comparison": entity_data.get("lastComparison"),
-            "comparison_count": entity_data.get("comparisonCount", 0),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return Baseline(**mapped_data)
-
-
-class DeleteBaselineCommand(Command[Dict[str, Any]]):
-    """Command to delete a baseline."""
-    
-    def __init__(self, http_client: HTTPClient, baseline_id: str):
-        self.http_client = http_client
-        self.baseline_id = baseline_id
-    
-    def execute(self) -> Dict[str, Any]:
-        """Execute the command to delete a baseline."""
-        response = self.http_client.delete(f"baselines/{self.baseline_id}")
-        return response
-
-
-class CompareBaselineCommand(Command[BaselineComparison]):
-    """Command to run baseline comparison."""
-    
-    def __init__(self, http_client: HTTPClient, request: Union[CompareBaselineRequest, Dict[str, Any]]):
-        self.http_client = http_client
-        self.request = request
-    
-    def execute(self) -> BaselineComparison:
-        """Execute the command to run baseline comparison."""
-        # Handle both dict and model objects
-        if isinstance(self.request, dict):
-            data = self.request
-        else:
-            data = self.request.model_dump(exclude_none=True)
-        
-        response = self.http_client.post("baselines/compare", json_data=data)
-        
-        entity_data = response.get("result", {})
-        
-        # Parse changes if present
-        changes = []
-        for change_data in entity_data.get("changes", []):
-            change_mapped = {
-                "id": change_data.get("_id"),
-                "comparison_id": change_data.get("comparisonId"),
-                "change_type": change_data.get("changeType"),
-                "item_type": change_data.get("itemType"),
-                "path": change_data.get("path"),
-                "old_value": change_data.get("oldValue"),
-                "new_value": change_data.get("newValue"),
-                "severity": change_data.get("severity", "medium"),
-                "category": change_data.get("category"),
-                "description": change_data.get("description"),
-                "detected_at": change_data.get("detectedAt"),
-                "risk_score": change_data.get("riskScore", 0.0),
-            }
-            
-            # Remove None values
-            change_mapped = {k: v for k, v in change_mapped.items() if v is not None}
-            changes.append(BaselineChange(**change_mapped))
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "baseline_id": entity_data.get("baselineId"),
-            "baseline_name": entity_data.get("baselineName"),
-            "endpoint_id": entity_data.get("endpointId"),
-            "endpoint_name": entity_data.get("endpointName"),
-            "status": entity_data.get("status", "pending"),
-            "started_at": entity_data.get("startedAt"),
-            "completed_at": entity_data.get("completedAt"),
-            "duration": entity_data.get("duration"),
-            "total_changes": entity_data.get("totalChanges", 0),
-            "added_items": entity_data.get("addedItems", 0),
-            "removed_items": entity_data.get("removedItems", 0),
-            "modified_items": entity_data.get("modifiedItems", 0),
-            "moved_items": entity_data.get("movedItems", 0),
-            "high_risk_changes": entity_data.get("highRiskChanges", 0),
-            "medium_risk_changes": entity_data.get("mediumRiskChanges", 0),
-            "low_risk_changes": entity_data.get("lowRiskChanges", 0),
-            "changes": changes,
-            "organization_id": entity_data.get("organizationId", 0),
-            "triggered_by": entity_data.get("triggeredBy"),
-            "error_message": entity_data.get("errorMessage"),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return BaselineComparison(**mapped_data)
-
-
-class CreateBaselineProfileCommand(Command[BaselineProfile]):
-    """Command to create a new baseline profile."""
-    
-    def __init__(self, http_client: HTTPClient, request: CreateBaselineProfileRequest):
-        self.http_client = http_client
-        self.request = request
-    
-    def execute(self) -> BaselineProfile:
-        """Execute the command to create a baseline profile."""
-        data = self.request.model_dump(exclude_none=True)
-        
-        response = self.http_client.post("baselines/profiles", json_data=data)
-        
-        entity_data = response.get("result", {})
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "name": entity_data.get("name"),
-            "description": entity_data.get("description"),
-            "settings": entity_data.get("settings", {}),
-            "categories": entity_data.get("categories", []),
-            "exclusions": entity_data.get("exclusions", []),
-            "organization_id": entity_data.get("organizationId", 0),
-            "created_at": entity_data.get("createdAt"),
-            "updated_at": entity_data.get("updatedAt"),
-            "created_by": entity_data.get("createdBy"),
-            "is_default": entity_data.get("isDefault", False),
-            "baseline_count": entity_data.get("baselineCount", 0),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return BaselineProfile(**mapped_data)
-
-
-class UpdateBaselineProfileCommand(Command[BaselineProfile]):
-    """Command to update an existing baseline profile."""
-    
-    def __init__(self, http_client: HTTPClient, profile_id: str, request: CreateBaselineProfileRequest):
-        self.http_client = http_client
-        self.profile_id = profile_id
-        self.request = request
-    
-    def execute(self) -> BaselineProfile:
-        """Execute the command to update a baseline profile."""
-        data = self.request.model_dump(exclude_none=True)
-        
-        response = self.http_client.put(f"baselines/profiles/{self.profile_id}", json_data=data)
-        
-        entity_data = response.get("result", {})
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "name": entity_data.get("name"),
-            "description": entity_data.get("description"),
-            "settings": entity_data.get("settings", {}),
-            "categories": entity_data.get("categories", []),
-            "exclusions": entity_data.get("exclusions", []),
-            "organization_id": entity_data.get("organizationId", 0),
-            "created_at": entity_data.get("createdAt"),
-            "updated_at": entity_data.get("updatedAt"),
-            "created_by": entity_data.get("createdBy"),
-            "is_default": entity_data.get("isDefault", False),
-            "baseline_count": entity_data.get("baselineCount", 0),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return BaselineProfile(**mapped_data)
-
-
-class DeleteBaselineProfileCommand(Command[Dict[str, Any]]):
-    """Command to delete a baseline profile."""
-    
-    def __init__(self, http_client: HTTPClient, profile_id: str):
-        self.http_client = http_client
-        self.profile_id = profile_id
-    
-    def execute(self) -> Dict[str, Any]:
-        """Execute the command to delete a baseline profile."""
-        response = self.http_client.delete(f"baselines/profiles/{self.profile_id}")
-        return response
-
-
-class CreateBaselineScheduleCommand(Command[BaselineSchedule]):
-    """Command to create a baseline schedule."""
-    
-    def __init__(self, http_client: HTTPClient, baseline_id: str, schedule_data: Dict[str, Any]):
-        self.http_client = http_client
-        self.baseline_id = baseline_id
-        self.schedule_data = schedule_data
-    
-    def execute(self) -> BaselineSchedule:
-        """Execute the command to create a baseline schedule."""
-        response = self.http_client.post("baselines/schedules", json_data=self.schedule_data)
-        
-        entity_data = response.get("result", {})
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "baseline_id": entity_data.get("baselineId"),
-            "baseline_name": entity_data.get("baselineName"),
-            "schedule_type": entity_data.get("scheduleType"),
-            "cron_expression": entity_data.get("cronExpression"),
-            "interval_minutes": entity_data.get("intervalMinutes"),
-            "enabled": entity_data.get("enabled", True),
-            "next_run": entity_data.get("nextRun"),
-            "last_run": entity_data.get("lastRun"),
-            "run_count": entity_data.get("runCount", 0),
-            "organization_id": entity_data.get("organizationId", 0),
-            "created_at": entity_data.get("createdAt"),
-            "updated_at": entity_data.get("updatedAt"),
-            "created_by": entity_data.get("createdBy"),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return BaselineSchedule(**mapped_data)
-
-
-class DeleteBaselineScheduleCommand(Command[Dict[str, Any]]):
-    """Command to delete a baseline schedule."""
-    
-    def __init__(self, http_client: HTTPClient, schedule_id: str):
-        self.http_client = http_client
-        self.schedule_id = schedule_id
-    
-    def execute(self) -> Dict[str, Any]:
-        """Execute the command to delete a baseline schedule."""
-        response = self.http_client.delete(f"baselines/schedules/{self.schedule_id}")
-        return response
-
-
-class RefreshBaselineCommand(Command[Baseline]):
-    """Command to refresh/rebuild a baseline."""
-    
-    def __init__(self, http_client: HTTPClient, baseline_id: str):
-        self.http_client = http_client
-        self.baseline_id = baseline_id
-    
-    def execute(self) -> Baseline:
-        """Execute the command to refresh a baseline."""
-        response = self.http_client.post(f"baselines/{self.baseline_id}/refresh")
-        
-        entity_data = response.get("result", {})
-        
-        mapped_data = {
-            "id": entity_data.get("_id"),
-            "name": entity_data.get("name"),
-            "description": entity_data.get("description"),
-            "type": entity_data.get("type"),
-            "status": entity_data.get("status", "creating"),
-            "endpoint_id": entity_data.get("endpointId"),
-            "endpoint_name": entity_data.get("endpointName"),
-            "organization_id": entity_data.get("organizationId", 0),
-            "created_at": entity_data.get("createdAt"),
-            "updated_at": entity_data.get("updatedAt"),
-            "created_by": entity_data.get("createdBy"),
-            "item_count": entity_data.get("itemCount", 0),
-            "file_count": entity_data.get("fileCount", 0),
-            "registry_count": entity_data.get("registryCount", 0),
-            "service_count": entity_data.get("serviceCount", 0),
-            "process_count": entity_data.get("processCount", 0),
-            "network_connection_count": entity_data.get("networkConnectionCount", 0),
-            "tags": entity_data.get("tags", []),
-            "profile_id": entity_data.get("profileId"),
-            "profile_name": entity_data.get("profileName"),
-            "last_comparison": entity_data.get("lastComparison"),
-            "comparison_count": entity_data.get("comparisonCount", 0),
-        }
-        
-        # Remove None values
-        mapped_data = {k: v for k, v in mapped_data.items() if v is not None}
-        
-        return Baseline(**mapped_data)
-
-
-class AcquireBaselineByFilterCommand(Command[Dict[str, Any]]):
-    """Command to acquire baselines by asset filter criteria."""
-    
-    def __init__(self, http_client: HTTPClient, payload: Dict[str, Any]):
-        self.http_client = http_client
-        self.payload = payload
-    
-    def execute(self) -> Dict[str, Any]:
-        """Execute the command to acquire baselines by filter."""
-        response = self.http_client.post("baseline/acquire", json_data=self.payload)
-        return response
-
-
-class CompareBaselineByEndpointCommand(Command[Dict[str, Any]]):
-    """Command to compare baseline acquisition tasks by endpoint ID."""
-    
-    def __init__(self, http_client: HTTPClient, payload: Dict[str, Any]):
-        self.http_client = http_client
-        self.payload = payload
-    
-    def execute(self) -> Dict[str, Any]:
-        """Execute the command to compare baselines by endpoint."""
-        response = self.http_client.post("baseline/compare", json_data=self.payload)
-        return response 
+"""
+Baseline-related commands for the Binalyze AIR SDK.
+"""
+
+from typing import Dict, Any
+
+from ..base import Command
+from ..http_client import HTTPClient
+
+
+class AcquireBaselineByFilterCommand(Command[Dict[str, Any]]):
+    """Command to acquire baselines by asset filter criteria."""
+    
+    def __init__(self, http_client: HTTPClient, payload: Dict[str, Any]):
+        self.http_client = http_client
+        self.payload = payload
+    
+    def execute(self) -> Dict[str, Any]:
+        """Execute the command to acquire baselines by filter."""
+        return self.http_client.post("baseline/acquire", json_data=self.payload)
+
+
+class CompareBaselineByEndpointCommand(Command[Dict[str, Any]]):
+    """Command to compare baseline acquisition tasks by endpoint ID."""
+    
+    def __init__(self, http_client: HTTPClient, payload: Dict[str, Any]):
+        self.http_client = http_client
+        self.payload = payload
+    
+    def execute(self) -> Dict[str, Any]:
+        """Execute the command to compare baseline tasks."""
+        return self.http_client.post("baseline/compare", json_data=self.payload)