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

unrealon_core/models/bridge_stats.py

@@ -0,0 +1,58 @@
+"""
+Bridge Statistics Models
+
+Strictly typed models for WebSocket bridge statistics.
+Following critical requirements - no raw Dict[str, Any].
+
+Phase 2: Core Systems - Bridge Statistics
+"""
+
+from datetime import datetime
+from typing import Optional
+from pydantic import BaseModel, Field, ConfigDict
+
+from .base import UnrealOnBaseModel
+from ..utils.time import utc_now
+
+
+class BridgeStatsData(UnrealOnBaseModel):
+    """Internal bridge statistics data."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    tasks_sent: int = Field(default=0, description="Number of tasks sent")
+    tasks_failed: int = Field(default=0, description="Number of tasks failed")
+    configs_sent: int = Field(default=0, description="Number of configurations sent")
+    configs_failed: int = Field(default=0, description="Number of configurations failed")
+    messages_processed: int = Field(default=0, description="Number of messages processed")
+    start_time: str = Field(description="Start time ISO string")
+    
+    def increment_tasks_sent(self) -> None:
+        """Increment tasks sent counter."""
+        self.tasks_sent += 1
+    
+    def increment_tasks_failed(self) -> None:
+        """Increment tasks failed counter."""
+        self.tasks_failed += 1
+    
+    def increment_configs_sent(self) -> None:
+        """Increment configs sent counter."""
+        self.configs_sent += 1
+    
+    def increment_configs_failed(self) -> None:
+        """Increment configs failed counter."""
+        self.configs_failed += 1
+    
+    def increment_messages_processed(self) -> None:
+        """Increment messages processed counter."""
+        self.messages_processed += 1
+
+
+def create_initial_bridge_stats() -> BridgeStatsData:
+    """Create initial bridge statistics."""
+    return BridgeStatsData(
+        start_time=utc_now().isoformat()
+    )

unrealon_core/models/communication.py

@@ -0,0 +1,39 @@
+"""
+WebSocket Communication Models for UnrealOn RPC.
+
+CRITICAL REQUIREMENTS COMPLIANT:
+- No raw Dict[str, Any] usage
+- 100% Pydantic v2 validation  
+- Strict typing throughout
+- Files under 500 lines
+
+This file imports from strictly typed websocket models.
+All raw dictionaries have been replaced with Pydantic models.
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+# Import all strictly typed models from websocket package
+from .websocket import *
+from .websocket.logging import LogContext
+
+# Export commonly used data models for easy access
+__all__ = [
+    # Message types and base
+    'MessageType', 'WebSocketMessage',
+    
+    # Message classes
+    'DriverRegistrationMessage', 'RegistrationResponseMessage',
+    'TaskAssignmentMessage', 'TaskResultMessage', 
+    'LogEntryMessage', 'LogBatchMessage',
+    'HeartbeatMessage', 'ConfigurationUpdateMessage',
+    'ErrorMessage', 'AckMessage',
+    
+    # Data models for strict typing
+    'TaskAssignmentData', 'TaskResultData', 'TaskParameters',
+    'DriverRegistrationData', 'HeartbeatData', 'LogEntryData', 'LogContext',
+    
+    # Utilities
+    'create_error_message', 'create_ack_message',
+    
+]

unrealon_core/models/config.py

@@ -0,0 +1,47 @@
+"""
+Config models - Phase 2 update.
+
+Import from strictly typed websocket models to avoid duplication.
+Following critical requirements - no raw Dict[str, Any].
+"""
+
+# Import strictly typed models from websocket package
+from .websocket.config import (
+    DriverConfiguration,
+    LoggingConfiguration,
+    TaskConfiguration,
+    ProxyConfiguration
+)
+from .base import ConfigBaseModel
+
+# System config that doesn't exist in websocket models yet
+class SystemConfig(ConfigBaseModel):
+    """System-wide configuration."""
+    
+    redis_url: str = "redis://localhost:6379/0"
+    websocket_url: str = "ws://localhost:8000/ws"
+    max_workers: int = 10
+    debug: bool = False
+
+# Legacy compatibility - map to new typed models
+HttpConfig = TaskConfiguration  # HTTP settings are part of task config
+ProxyConfig = ProxyConfiguration
+LoggingConfig = LoggingConfiguration
+BrowserConfig = TaskConfiguration  # Browser settings are part of task config
+CacheConfig = SystemConfig  # Cache settings are system-wide
+ThreadConfig = TaskConfiguration  # Thread settings are part of task config
+
+__all__ = [
+    'DriverConfiguration',
+    'LoggingConfiguration', 
+    'TaskConfiguration',
+    'ProxyConfiguration',
+    'SystemConfig',
+    # Legacy names
+    'HttpConfig',
+    'ProxyConfig',
+    'LoggingConfig',
+    'BrowserConfig',
+    'CacheConfig',
+    'ThreadConfig'
+]

unrealon_core/models/connection_stats.py

@@ -0,0 +1,47 @@
+"""
+Connection Statistics Models
+
+Strictly typed models for WebSocket connection statistics.
+Following critical requirements - no raw Dict[str, Any].
+
+Phase 2: Core Systems - Connection Statistics
+"""
+
+from pydantic import BaseModel, Field, ConfigDict
+
+from .base import UnrealOnBaseModel
+
+
+class ConnectionStatsData(UnrealOnBaseModel):
+    """Internal connection statistics data."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    total_connections: int = Field(default=0, description="Total connections made")
+    active_connections: int = Field(default=0, description="Currently active connections")
+    total_messages: int = Field(default=0, description="Total messages sent")
+    failed_messages: int = Field(default=0, description="Failed messages")
+    
+    def increment_total_connections(self) -> None:
+        """Increment total connections counter."""
+        self.total_connections += 1
+    
+    def set_active_connections(self, count: int) -> None:
+        """Set active connections count."""
+        self.active_connections = count
+    
+    def increment_total_messages(self) -> None:
+        """Increment total messages counter."""
+        self.total_messages += 1
+    
+    def increment_failed_messages(self) -> None:
+        """Increment failed messages counter."""
+        self.failed_messages += 1
+
+
+def create_initial_connection_stats() -> ConnectionStatsData:
+    """Create initial connection statistics."""
+    return ConnectionStatsData()

unrealon_core/models/driver.py

@@ -0,0 +1,30 @@
+"""
+Driver models - Phase 2 update.
+
+Import from strictly typed websocket models to avoid duplication.
+Following critical requirements - no raw Dict[str, Any].
+"""
+
+# Import strictly typed models from websocket package
+from .websocket.driver import (
+    DriverRegistrationData,
+    DriverMetadata,
+    DriverConfiguration,
+    RegistrationResponseData
+)
+
+# Legacy compatibility
+DriverInfo = DriverRegistrationData
+DriverConfig = DriverConfiguration
+DriverCapability = DriverMetadata
+
+__all__ = [
+    'DriverRegistrationData',
+    'DriverMetadata', 
+    'DriverConfiguration',
+    'RegistrationResponseData',
+    # Legacy names
+    'DriverInfo',
+    'DriverConfig',
+    'DriverCapability'
+]

unrealon_core/models/driver_details.py

@@ -0,0 +1,98 @@
+"""
+Driver Details Models
+
+Strictly typed models for driver information and statistics.
+Following critical requirements - no raw Dict[str, Any].
+
+Phase 2: Core Systems - Driver Details
+"""
+
+from datetime import datetime
+from typing import Optional, List
+from pydantic import BaseModel, Field, ConfigDict
+
+from .base import UnrealOnBaseModel
+
+
+class DriverDetails(UnrealOnBaseModel):
+    """Complete driver details with statistics."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    # Basic driver info
+    driver_id: str = Field(description="Driver ID")
+    driver_name: Optional[str] = Field(default=None, description="Driver name")
+    driver_type: str = Field(description="Driver type")
+    capabilities: List[str] = Field(default_factory=list, description="Driver capabilities")
+    version: Optional[str] = Field(default=None, description="Driver version")
+    
+    # Connection info
+    connection_id: str = Field(description="WebSocket connection ID")
+    status: str = Field(description="Driver status")
+    connected_at: datetime = Field(description="Connection timestamp")
+    last_seen: datetime = Field(description="Last activity timestamp")
+    
+    # Task statistics
+    active_tasks: int = Field(default=0, description="Number of active tasks")
+    completed_tasks: int = Field(default=0, description="Number of completed tasks")
+    failed_tasks: int = Field(default=0, description="Number of failed tasks")
+    
+    # Performance metrics
+    uptime_seconds: float = Field(default=0.0, description="Uptime in seconds")
+    memory_usage_mb: Optional[float] = Field(default=None, description="Memory usage in MB")
+    cpu_usage_percent: Optional[float] = Field(default=None, description="CPU usage percentage")
+
+
+class DriverInfo(UnrealOnBaseModel):
+    """Basic driver information."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    driver_id: str = Field(description="Driver ID")
+    connection_id: str = Field(description="WebSocket connection ID")
+    status: str = Field(description="Driver status")
+    last_heartbeat: Optional[datetime] = Field(default=None, description="Last heartbeat timestamp")
+
+
+class DriverListResult(UnrealOnBaseModel):
+    """Result of driver list query."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    drivers: List[DriverDetails] = Field(description="List of driver details")
+    total_count: int = Field(description="Total number of drivers")
+    filtered_count: int = Field(description="Number of drivers after filtering")
+    filter_applied: Optional[str] = Field(default=None, description="Filter that was applied")
+
+
+def create_driver_details_from_session(
+    driver_info: DriverInfo,
+    session: "DriverSession"
+) -> DriverDetails:
+    """Create DriverDetails from DriverInfo and DriverSession."""
+    return DriverDetails(
+        driver_id=driver_info.driver_id,
+        driver_name=getattr(session, 'driver_name', None),
+        driver_type=session.driver_type,
+        capabilities=session.capabilities,
+        version=getattr(session, 'version', None),
+        connection_id=session.connection_id,
+        status=session.status,
+        connected_at=session.connected_at,
+        last_seen=session.last_seen,
+        active_tasks=session.active_tasks,
+        completed_tasks=getattr(session, 'completed_tasks', 0),
+        failed_tasks=getattr(session, 'failed_tasks', 0),
+        uptime_seconds=(utc_now() - session.connected_at).total_seconds(),
+        memory_usage_mb=getattr(session, 'memory_usage_mb', None),
+        cpu_usage_percent=getattr(session, 'cpu_usage_percent', None)
+    )

unrealon_core/models/logging.py

@@ -0,0 +1,28 @@
+"""
+Logging models - Phase 2 update.
+
+Import from strictly typed websocket models to avoid duplication.
+Following critical requirements - no raw Dict[str, Any].
+"""
+
+# Import strictly typed models from websocket package
+from .websocket.logging import (
+    LogEntryData,
+    LogBatchData,
+    LogContext
+)
+
+# Legacy compatibility
+LogEntry = LogEntryData
+LogQuery = LogContext
+LogMetrics = LogBatchData
+
+__all__ = [
+    'LogEntryData',
+    'LogBatchData',
+    'LogContext',
+    # Legacy names
+    'LogEntry',
+    'LogQuery',
+    'LogMetrics'
+]

unrealon_core/models/task.py

@@ -0,0 +1,21 @@
+"""
+Task models - Phase 2 update.
+
+Import from strictly typed websocket models to avoid duplication.
+Following critical requirements - no raw Dict[str, Any].
+"""
+
+# Import strictly typed models from websocket package
+from .websocket.tasks import (
+    TaskAssignmentData,
+    TaskResultData,
+    TaskParameters,
+    TaskMetadata
+)
+
+__all__ = [
+    'TaskAssignmentData',
+    'TaskResultData',
+    'TaskParameters',
+    'TaskMetadata',
+]

unrealon_core/models/typed_responses.py

@@ -0,0 +1,210 @@
+"""
+Typed Response Models
+
+Strictly typed models to replace Dict[str, Any] usage.
+Following critical requirements - no raw Dict[str, Any].
+
+Phase 2: Core Systems - Typed Responses
+"""
+
+from datetime import datetime
+from typing import Optional, List
+from pydantic import BaseModel, Field, ConfigDict
+
+from .base import UnrealOnBaseModel
+
+
+class HealthCheckDetails(UnrealOnBaseModel):
+    """Health check details model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    manager_status: str = Field(description="Manager status")
+    success_rate: float = Field(description="Success rate percentage")
+    total_operations: int = Field(description="Total operations count")
+    uptime_seconds: float = Field(description="Uptime in seconds")
+
+
+class HealthCheckResult(UnrealOnBaseModel):
+    """Health check result model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    status: str = Field(description="Health status")
+    message: str = Field(description="Health message")
+    details: HealthCheckDetails = Field(description="Health details")
+
+
+class StatusInfo(UnrealOnBaseModel):
+    """Status information model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    name: str = Field(description="Component name")
+    status: str = Field(description="Component status")
+    enabled: bool = Field(description="Whether component is enabled")
+    uptime_seconds: Optional[float] = Field(default=None, description="Uptime in seconds")
+    initialization_time: Optional[str] = Field(default=None, description="Initialization time ISO string")
+    shutdown_time: Optional[str] = Field(default=None, description="Shutdown time ISO string")
+    stats: "StatsModel" = Field(description="Component statistics")
+    config: "ConfigModel" = Field(description="Component configuration")
+
+
+class StatsModel(UnrealOnBaseModel):
+    """Statistics model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    operations_total: int = Field(default=0, description="Total operations")
+    operations_successful: int = Field(default=0, description="Successful operations")
+    operations_failed: int = Field(default=0, description="Failed operations")
+    last_operation_time: Optional[str] = Field(default=None, description="Last operation time ISO string")
+    average_operation_time: float = Field(default=0.0, description="Average operation time")
+
+
+class ConfigModel(UnrealOnBaseModel):
+    """Configuration model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    enabled: bool = Field(description="Whether enabled")
+    timeout_seconds: float = Field(description="Timeout in seconds")
+    retry_count: int = Field(description="Retry count")
+    log_level: str = Field(description="Log level")
+
+
+class SessionInfo(UnrealOnBaseModel):
+    """HTTP session information model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    cookies_count: int = Field(description="Number of cookies")
+    has_proxy_manager: bool = Field(description="Whether proxy manager is available")
+    use_proxy_manager: bool = Field(description="Whether using proxy manager")
+    max_connections: int = Field(description="Maximum connections")
+    default_timeout: float = Field(description="Default timeout")
+
+
+class ProxyStats(UnrealOnBaseModel):
+    """Proxy statistics model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    total_proxies: int = Field(description="Total proxies")
+    healthy_proxies: int = Field(description="Healthy proxies")
+    banned_proxies: int = Field(description="Banned proxies")
+    average_success_rate: float = Field(description="Average success rate")
+    rotation_strategy: str = Field(description="Rotation strategy")
+    current_index: int = Field(description="Current rotation index")
+    has_rpc_client: bool = Field(description="Whether RPC client is available")
+
+
+class ThreadStats(UnrealOnBaseModel):
+    """Thread statistics model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    thread_pools: "ThreadPoolsInfo" = Field(description="Thread pools information")
+    concurrency: "ConcurrencyInfo" = Field(description="Concurrency information")
+    active_task_names: List[str] = Field(description="Active task names")
+
+
+class ThreadPoolsInfo(UnrealOnBaseModel):
+    """Thread pools information model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    io_bound: "ThreadPoolInfo" = Field(description="I/O bound thread pool")
+    cpu_bound: "ThreadPoolInfo" = Field(description="CPU bound thread pool")
+    general: "ThreadPoolInfo" = Field(description="General thread pool")
+
+
+class ThreadPoolInfo(UnrealOnBaseModel):
+    """Thread pool information model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    max_workers: int = Field(description="Maximum workers")
+    available: bool = Field(description="Whether available")
+
+
+class ConcurrencyInfo(UnrealOnBaseModel):
+    """Concurrency information model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    max_concurrent_tasks: int = Field(description="Maximum concurrent tasks")
+    active_tasks: int = Field(description="Active tasks")
+    semaphore_available: int = Field(description="Available semaphore permits")
+
+
+class CacheStats(UnrealOnBaseModel):
+    """Cache statistics model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    entries: int = Field(description="Number of entries")
+    max_entries: int = Field(description="Maximum entries")
+    memory_usage_mb: float = Field(description="Memory usage in MB")
+    max_memory_mb: int = Field(description="Maximum memory in MB")
+    hits: int = Field(description="Cache hits")
+    misses: int = Field(description="Cache misses")
+    hit_rate_percent: float = Field(description="Hit rate percentage")
+    evictions: int = Field(description="Number of evictions")
+    eviction_strategy: str = Field(description="Eviction strategy")
+
+
+class BatchStats(UnrealOnBaseModel):
+    """Batch statistics model."""
+    
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="forbid"
+    )
+    
+    current_batch_size: int = Field(description="Current batch size")
+    max_batch_size: int = Field(description="Maximum batch size")
+    batch_timeout: float = Field(description="Batch timeout")
+    has_rpc_client: bool = Field(description="Whether RPC client is available")
+
+
+# Update forward references
+StatusInfo.model_rebuild()
+ThreadStats.model_rebuild()
+ThreadPoolsInfo.model_rebuild()

unrealon_core/models/websocket/__init__.py

@@ -0,0 +1,91 @@
+"""
+WebSocket Models Package.
+
+Strictly typed WebSocket communication models following critical requirements:
+- No raw Dict[str, Any] usage
+- 100% Pydantic v2 validation
+- Files under 500 lines each
+- Complete type safety
+
+Phase 2: Core Systems - WebSocket Bridge
+"""
+
+from .base import MessageType, WebSocketMessage
+from .driver import (
+    DriverRegistrationData, DriverRegistrationMessage,
+    RegistrationResponseData, RegistrationResponseMessage
+)
+from .tasks import (
+    TaskAssignmentData, TaskAssignmentMessage,
+    TaskResultData, TaskResultMessage
+)
+from .logging import (
+    LogEntryData, LogEntryMessage,
+    LogBatchData, LogBatchMessage
+)
+from .heartbeat import HeartbeatData, HeartbeatMessage
+from .config import ConfigurationUpdateData, ConfigurationUpdateMessage
+from .errors import ErrorData, ErrorMessage, AckData, AckMessage
+from .proxy import (
+    # Core models
+    ProxyType, ProxyHealthStatus, ProxyCredentials, ProxyInfo, ProxyAssignment,
+    # Data payloads
+    ProxyRequestData, ProxyResponseData, ProxyHealthReportData,
+    ProxyRotationRequestData, ProxyReleaseData,
+    # WebSocket messages
+    ProxyRequestMessage, ProxyResponseMessage, ProxyHealthReportMessage,
+    ProxyRotationRequestMessage, ProxyReleaseMessage
+)
+
+from .utils import create_error_message, create_ack_message
+
+__all__ = [
+    # Base
+    'MessageType',
+    'WebSocketMessage',
+    
+    # Driver messages
+    'DriverRegistrationData',
+    'DriverRegistrationMessage', 
+    'RegistrationResponseData',
+    'RegistrationResponseMessage',
+    
+    # Task messages
+    'TaskAssignmentData',
+    'TaskAssignmentMessage',
+    'TaskResultData', 
+    'TaskResultMessage',
+    
+    # Logging messages
+    'LogEntryData',
+    'LogEntryMessage',
+    'LogBatchData',
+    'LogBatchMessage',
+    
+    # Heartbeat messages
+    'HeartbeatData',
+    'HeartbeatMessage',
+    
+    # Configuration messages
+    'ConfigurationUpdateData',
+    'ConfigurationUpdateMessage',
+    
+    # Error messages
+    'ErrorData',
+    'ErrorMessage',
+    'AckData',
+    'AckMessage',
+    
+    # Proxy core models
+    'ProxyType', 'ProxyHealthStatus', 'ProxyCredentials', 'ProxyInfo', 'ProxyAssignment',
+    # Proxy data payloads
+    'ProxyRequestData', 'ProxyResponseData', 'ProxyHealthReportData',
+    'ProxyRotationRequestData', 'ProxyReleaseData',
+    # Proxy messages
+    'ProxyRequestMessage', 'ProxyResponseMessage', 'ProxyHealthReportMessage',
+    'ProxyRotationRequestMessage', 'ProxyReleaseMessage',
+    
+    # Utilities
+    'create_error_message',
+    'create_ack_message'
+]