unrealon 1.1.5__py3-none-any.whl → 2.0.4__py3-none-any.whl

unrealon_core/models/websocket/base.py

@@ -0,0 +1,49 @@
+"""
+Base WebSocket Models.
+
+Core message types and base classes for WebSocket communication.
+Strictly typed without raw dictionaries.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from typing import Optional
+from datetime import datetime
+
+from pydantic import Field, ConfigDict, field_serializer
+from ..base import UnrealOnBaseModel, TimestampedModel, IdentifiedModel
+from ...enums.types import MessageType
+from ...utils.time import datetime_to_iso
+
+
+class WebSocketMessage(IdentifiedModel, TimestampedModel):
+    """
+    Base WebSocket message model.
+    
+    All WebSocket messages inherit from this base class for consistency.
+    Provides message ID, timestamp, and optional correlation ID.
+    
+    CRITICAL: No raw Dict[str, Any] - all data must be typed!
+    """
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        use_enum_values=True
+    )
+    
+    type: MessageType = Field(
+        description="Type of the WebSocket message"
+    )
+    
+    correlation_id: Optional[str] = Field(
+        default=None,
+        description="ID of the message this is responding to"
+    )
+    
+    @field_serializer('created_at', 'updated_at', when_used='json')
+    def serialize_datetime_fields(self, value: Optional[datetime]) -> Optional[str]:
+        """Serialize datetime fields to ISO format."""
+        return datetime_to_iso(value)
+    
+

unrealon_core/models/websocket/config.py

@@ -0,0 +1,200 @@
+"""
+Configuration WebSocket Models.
+
+Strictly typed models for configuration updates.
+No raw dictionaries - all data structures are Pydantic models.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from typing import Optional, List
+from pydantic import Field, ConfigDict
+
+from ..base import UnrealOnBaseModel
+from .base import WebSocketMessage, MessageType
+
+
+class LoggingConfiguration(UnrealOnBaseModel):
+    """Logging configuration - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    level: str = Field(
+        default="INFO",
+        pattern=r"^(DEBUG|INFO|WARNING|ERROR|CRITICAL)$",
+        description="Logging level"
+    )
+    
+    batch_size: int = Field(
+        default=10,
+        ge=1,
+        le=100,
+        description="Log batch size"
+    )
+    
+    send_interval: float = Field(
+        default=1.0,
+        ge=0.1,
+        le=60.0,
+        description="Log send interval in seconds"
+    )
+    
+    local_logging: bool = Field(
+        default=True,
+        description="Enable local logging"
+    )
+
+
+class TaskConfiguration(UnrealOnBaseModel):
+    """Task configuration - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    max_concurrent_tasks: int = Field(
+        default=1,
+        ge=1,
+        le=100,
+        description="Maximum concurrent tasks"
+    )
+    
+    default_timeout: float = Field(
+        default=300.0,
+        ge=30.0,
+        le=3600.0,
+        description="Default task timeout in seconds"
+    )
+    
+    max_retries: int = Field(
+        default=3,
+        ge=0,
+        le=10,
+        description="Maximum task retries"
+    )
+    
+    retry_delay: float = Field(
+        default=1.0,
+        ge=0.1,
+        le=60.0,
+        description="Retry delay in seconds"
+    )
+
+
+class ProxyConfiguration(UnrealOnBaseModel):
+    """Proxy configuration - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    enabled: bool = Field(
+        default=False,
+        description="Enable proxy usage"
+    )
+    
+    rotation_enabled: bool = Field(
+        default=True,
+        description="Enable proxy rotation"
+    )
+    
+    rotation_interval: int = Field(
+        default=10,
+        ge=1,
+        le=1000,
+        description="Proxy rotation interval (requests)"
+    )
+    
+    health_check_interval: float = Field(
+        default=60.0,
+        ge=10.0,
+        le=3600.0,
+        description="Proxy health check interval in seconds"
+    )
+
+
+class DriverConfiguration(UnrealOnBaseModel):
+    """Complete driver configuration - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    heartbeat_interval: int = Field(
+        default=30,
+        ge=5,
+        le=300,
+        description="Heartbeat interval in seconds"
+    )
+    
+    logging: LoggingConfiguration = Field(
+        default_factory=LoggingConfiguration,
+        description="Logging configuration"
+    )
+    
+    tasks: TaskConfiguration = Field(
+        default_factory=TaskConfiguration,
+        description="Task configuration"
+    )
+    
+    proxy: ProxyConfiguration = Field(
+        default_factory=ProxyConfiguration,
+        description="Proxy configuration"
+    )
+    
+    enabled_managers: List[str] = Field(
+        default_factory=lambda: ["logger", "http", "proxy"],
+        description="List of enabled managers"
+    )
+
+
+class ConfigurationUpdateData(UnrealOnBaseModel):
+    """Configuration update data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    configuration: DriverConfiguration = Field(
+        description="Updated configuration parameters"
+    )
+    
+    apply_immediately: bool = Field(
+        default=True,
+        description="Whether to apply configuration immediately"
+    )
+    
+    restart_required: bool = Field(
+        default=False,
+        description="Whether driver restart is required"
+    )
+    
+    config_version: Optional[str] = Field(
+        default=None,
+        description="Configuration version"
+    )
+
+
+class ConfigurationUpdateMessage(WebSocketMessage):
+    """Configuration update message (Server → Driver)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.COMMAND_CONFIG_UPDATE,
+        frozen=True
+    )
+    
+    data: ConfigurationUpdateData = Field(
+        description="Configuration update data"
+    )

unrealon_core/models/websocket/driver.py

@@ -0,0 +1,215 @@
+"""
+Driver WebSocket Models.
+
+Strictly typed models for driver registration and responses.
+No raw dictionaries - all data structures are Pydantic models.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from typing import List, Optional
+from pydantic import Field, ConfigDict, field_validator
+
+from ..base import UnrealOnBaseModel
+from .base import WebSocketMessage, MessageType
+from ...version import get_driver_version
+
+
+class DriverMetadata(UnrealOnBaseModel):
+    """Driver metadata with strict typing."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    region: Optional[str] = Field(
+        default=None,
+        description="Driver deployment region"
+    )
+    
+    environment: Optional[str] = Field(
+        default="production",
+        description="Environment (development, staging, production)"
+    )
+    
+    max_concurrent_tasks: Optional[int] = Field(
+        default=1,
+        ge=1,
+        le=100,
+        description="Maximum concurrent tasks"
+    )
+    
+    tags: List[str] = Field(
+        default_factory=list,
+        description="Driver tags for filtering"
+    )
+
+
+class DriverRegistrationData(UnrealOnBaseModel):
+    """Driver registration data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    driver_id: str = Field(
+        min_length=1,
+        max_length=100,
+        description="Unique driver identifier"
+    )
+    
+    driver_name: str = Field(
+        min_length=1,
+        max_length=200,
+        description="Human-readable driver name"
+    )
+    
+    driver_type: str = Field(
+        min_length=1,
+        max_length=50,
+        description="Type of driver (universal, ecommerce, etc.)"
+    )
+    
+    capabilities: List[str] = Field(
+        default_factory=list,
+        description="List of supported task types"
+    )
+    
+    version: str = Field(
+        default_factory=get_driver_version,
+        pattern=r"^\d+\.\d+\.\d+$",
+        description="Driver version (semver)"
+    )
+    
+    metadata: DriverMetadata = Field(
+        default_factory=DriverMetadata,
+        description="Driver metadata"
+    )
+    
+    @field_validator('capabilities')
+    @classmethod
+    def validate_capabilities(cls, v: List[str]) -> List[str]:
+        """Validate capabilities list."""
+        if not v:
+            raise ValueError("Driver must have at least one capability")
+        
+        # Remove duplicates while preserving order
+        seen = set()
+        unique_caps = []
+        for cap in v:
+            if cap not in seen:
+                seen.add(cap)
+                unique_caps.append(cap)
+        
+        return unique_caps
+
+
+class DriverRegistrationMessage(WebSocketMessage):
+    """Driver registration message (Driver → Server)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.DRIVER_REGISTER,
+        frozen=True
+    )
+    
+    data: DriverRegistrationData = Field(
+        description="Driver registration data"
+    )
+
+
+class DriverConfiguration(UnrealOnBaseModel):
+    """Driver configuration from server - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    heartbeat_interval: int = Field(
+        default=30,
+        ge=5,
+        le=300,
+        description="Heartbeat interval in seconds"
+    )
+    
+    log_level: str = Field(
+        default="INFO",
+        pattern=r"^(DEBUG|INFO|WARNING|ERROR|CRITICAL)$",
+        description="Logging level"
+    )
+    
+    batch_size: int = Field(
+        default=10,
+        ge=1,
+        le=100,
+        description="Log batch size"
+    )
+    
+    max_retries: int = Field(
+        default=3,
+        ge=0,
+        le=10,
+        description="Maximum task retries"
+    )
+    
+    timeout_seconds: int = Field(
+        default=300,
+        ge=30,
+        le=3600,
+        description="Default task timeout"
+    )
+
+
+class RegistrationResponseData(UnrealOnBaseModel):
+    """Registration response data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    success: bool = Field(
+        description="Whether registration was successful"
+    )
+    
+    driver_id: str = Field(
+        description="Confirmed driver ID"
+    )
+    
+    session_id: str = Field(
+        description="WebSocket session ID"
+    )
+    
+    message: str = Field(
+        description="Registration status message"
+    )
+    
+    configuration: DriverConfiguration = Field(
+        default_factory=DriverConfiguration,
+        description="Driver configuration from server"
+    )
+
+
+class RegistrationResponseMessage(WebSocketMessage):
+    """Registration response message (Server → Driver)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.DRIVER_REGISTER_RESPONSE,
+        frozen=True
+    )
+    
+    data: RegistrationResponseData = Field(
+        description="Registration response data"
+    )

unrealon_core/models/websocket/errors.py

@@ -0,0 +1,138 @@
+"""
+Error and Acknowledgment WebSocket Models.
+
+Strictly typed models for error messages and acknowledgments.
+No raw dictionaries - all data structures are Pydantic models.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from typing import Optional, List
+from pydantic import Field, ConfigDict
+
+from ..base import UnrealOnBaseModel
+from .base import WebSocketMessage, MessageType
+
+
+class ErrorDetails(UnrealOnBaseModel):
+    """Error details - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    field_errors: List[str] = Field(
+        default_factory=list,
+        description="Field validation errors"
+    )
+    
+    stack_trace: Optional[str] = Field(
+        default=None,
+        description="Stack trace (if available)"
+    )
+    
+    request_id: Optional[str] = Field(
+        default=None,
+        description="Request ID that caused the error"
+    )
+    
+    timestamp: Optional[str] = Field(
+        default=None,
+        description="Error timestamp"
+    )
+
+
+class ErrorData(UnrealOnBaseModel):
+    """Error data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    error_code: str = Field(
+        min_length=1,
+        max_length=100,
+        description="Error code identifier"
+    )
+    
+    error_message: str = Field(
+        min_length=1,
+        max_length=1000,
+        description="Human-readable error message"
+    )
+    
+    details: Optional[ErrorDetails] = Field(
+        default=None,
+        description="Additional error details"
+    )
+    
+    retry_after: Optional[int] = Field(
+        default=None,
+        ge=0,
+        le=3600,
+        description="Retry after seconds (if retryable)"
+    )
+
+
+class ErrorMessage(WebSocketMessage):
+    """Error message (Bidirectional)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.ERROR,
+        frozen=True
+    )
+    
+    data: ErrorData = Field(
+        description="Error data"
+    )
+
+
+class AckData(UnrealOnBaseModel):
+    """Acknowledgment data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    acknowledged: bool = Field(
+        default=True,
+        description="Acknowledgment status"
+    )
+    
+    message: Optional[str] = Field(
+        default=None,
+        max_length=500,
+        description="Optional acknowledgment message"
+    )
+    
+    processing_time_ms: Optional[float] = Field(
+        default=None,
+        ge=0.0,
+        description="Processing time in milliseconds"
+    )
+
+
+class AckMessage(WebSocketMessage):
+    """Acknowledgment message (Bidirectional)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.ACK,
+        frozen=True
+    )
+    
+    data: AckData = Field(
+        description="Acknowledgment data"
+    )

unrealon_core/models/websocket/heartbeat.py

@@ -0,0 +1,100 @@
+"""
+Heartbeat WebSocket Models.
+
+Strictly typed models for driver heartbeat messages.
+No raw dictionaries - all data structures are Pydantic models.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from typing import Optional
+from pydantic import Field, ConfigDict
+
+from ..base import UnrealOnBaseModel
+from .base import WebSocketMessage, MessageType
+
+
+class HeartbeatData(UnrealOnBaseModel):
+    """Heartbeat data payload - strictly typed."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    driver_status: str = Field(
+        default="ready",
+        pattern=r"^(ready|busy|idle|error|maintenance)$",
+        description="Current driver status"
+    )
+    
+    active_tasks: int = Field(
+        default=0,
+        ge=0,
+        le=1000,
+        description="Number of active tasks"
+    )
+    
+    completed_tasks: int = Field(
+        default=0,
+        ge=0,
+        description="Total completed tasks"
+    )
+    
+    failed_tasks: int = Field(
+        default=0,
+        ge=0,
+        description="Total failed tasks"
+    )
+    
+    uptime_seconds: float = Field(
+        default=0.0,
+        ge=0.0,
+        description="Driver uptime in seconds"
+    )
+    
+    memory_usage_mb: Optional[float] = Field(
+        default=None,
+        ge=0.0,
+        le=100000.0,  # 100GB max
+        description="Memory usage in MB"
+    )
+    
+    cpu_usage_percent: Optional[float] = Field(
+        default=None,
+        ge=0.0,
+        le=100.0,
+        description="CPU usage percentage"
+    )
+    
+    def get_success_rate(self) -> float:
+        """Calculate task success rate."""
+        total_tasks = self.completed_tasks + self.failed_tasks
+        if total_tasks == 0:
+            return 1.0
+        return self.completed_tasks / total_tasks
+    
+    def is_healthy(self) -> bool:
+        """Check if driver is healthy."""
+        return (
+            self.driver_status in ["ready", "busy", "idle"] and
+            self.get_success_rate() >= 0.8  # 80% success rate threshold
+        )
+
+
+class HeartbeatMessage(WebSocketMessage):
+    """Heartbeat message (Driver → Server)."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    type: MessageType = Field(
+        default=MessageType.DRIVER_HEARTBEAT,
+        frozen=True
+    )
+    
+    data: HeartbeatData = Field(
+        description="Heartbeat data"
+    )