unrealon 1.0.9__py3-none-any.whl → 1.1.0__py3-none-any.whl

unrealon_bridge/models/parser.py

@@ -0,0 +1,63 @@
+"""
+Parser-related models.
+
+Contains models for parser registration, information, and statistics.
+"""
+
+from typing import List, Optional, Dict, Literal
+from datetime import datetime
+from pydantic import Field
+from typing_extensions import Annotated
+
+from .base import BaseParserModel
+
+
+class ParserInfo(BaseParserModel):
+    """Parser registration and status information."""
+
+    parser_id: Annotated[str, Field(min_length=1, description="Unique parser identifier")]
+    parser_type: Annotated[str, Field(min_length=1, description="Type of parser (encar, autotrader, etc.)")]
+    version: Annotated[str, Field(min_length=1, description="Parser version")]
+    capabilities: List[str] = Field(default_factory=list, description="List of parser capabilities")
+    status: Literal["active", "inactive", "error"] = "active"
+    metadata: Dict[str, str] = Field(default_factory=dict, description="Additional parser metadata")
+    registered_at: datetime = Field(default_factory=datetime.now)
+    last_heartbeat: datetime = Field(default_factory=datetime.now)
+
+
+class ParserStats(BaseParserModel):
+    """Parser performance and usage statistics."""
+
+    parser_id: Annotated[str, Field(min_length=1, description="Parser ID")]
+    total_commands: Annotated[int, Field(ge=0)] = 0
+    successful_commands: Annotated[int, Field(ge=0)] = 0
+    failed_commands: Annotated[int, Field(ge=0)] = 0
+    average_response_time: Annotated[float, Field(ge=0)] = 0.0
+    uptime_seconds: Annotated[int, Field(ge=0)] = 0
+    memory_usage_mb: Annotated[float, Field(ge=0)] = 0.0
+    cpu_usage_percent: Annotated[float, Field(ge=0, le=100)] = 0.0
+    last_updated: datetime = Field(default_factory=datetime.now)
+
+
+class ParserHealth(BaseParserModel):
+    """Parser health check information."""
+
+    parser_id: Annotated[str, Field(min_length=1, description="Parser ID")]
+    status: Literal["healthy", "degraded", "unhealthy"] = "healthy"
+    last_check: datetime = Field(default_factory=datetime.now)
+    response_time: Annotated[float, Field(ge=0)] = 0.0
+    memory_usage: Annotated[float, Field(ge=0)] = 0.0
+    cpu_usage: Annotated[float, Field(ge=0, le=100)] = 0.0
+    active_connections: Annotated[int, Field(ge=0)] = 0
+    queue_size: Annotated[int, Field(ge=0)] = 0
+    errors: List[str] = Field(default_factory=list)
+
+
+class ParserSystemStats(BaseParserModel):
+    """System-wide parser statistics."""
+
+    total_parsers: Annotated[int, Field(ge=0)] = 0
+    active_sessions: Annotated[int, Field(ge=0)] = 0
+    total_commands: Annotated[int, Field(ge=0)] = 0
+    allocated_proxies: Annotated[int, Field(ge=0)] = 0
+    parser_types: Dict[str, int] = Field(default_factory=dict, description="Count by parser type")

unrealon_bridge/models/proxy.py

@@ -0,0 +1,41 @@
+"""
+Proxy-related models.
+
+Contains models for proxy allocation, management, and health checking.
+"""
+
+from typing import Optional, Literal
+from datetime import datetime
+from pydantic import Field
+from typing_extensions import Annotated
+
+from .base import BaseParserModel
+
+
+class ProxyRequest(BaseParserModel):
+    """Request for proxy allocation."""
+
+    parser_id: Annotated[str, Field(min_length=1, description="Parser ID requesting proxy")]
+    proxy_type: Literal["http", "https", "socks4", "socks5"] = "http"
+    location: Optional[Annotated[str, Field(min_length=1, description="Preferred proxy location")]] = None
+    session_duration: Annotated[int, Field(gt=0, le=86400, description="Session duration in seconds")] = 3600
+    bandwidth_limit: Optional[Annotated[int, Field(gt=0, description="Bandwidth limit in MB/s")]] = None
+
+
+class ProxyInfo(BaseParserModel):
+    """Proxy allocation information."""
+
+    proxy_id: Annotated[str, Field(min_length=1, description="Unique proxy identifier")]
+    parser_id: Annotated[str, Field(min_length=1, description="Parser ID using this proxy")]
+    host: Annotated[str, Field(min_length=1, description="Proxy host")]
+    port: Annotated[int, Field(gt=0, le=65535, description="Proxy port")]
+    proxy_type: Literal["http", "https", "socks4", "socks5"] = "http"
+    username: Optional[str] = None
+    password: Optional[str] = None
+    location: Optional[str] = None
+    status: Literal["active", "inactive", "error"] = "active"
+    allocated_at: datetime = Field(default_factory=datetime.now)
+    expires_at: Optional[datetime] = None
+    last_used: Optional[datetime] = None
+    usage_count: Annotated[int, Field(ge=0)] = 0
+    bandwidth_used: Annotated[float, Field(ge=0)] = 0.0

unrealon_bridge/models/requests.py

@@ -0,0 +1,95 @@
+"""
+RPC request models.
+
+Strict typing for all RPC requests - NO Dict[str, Any] allowed!
+"""
+
+from typing import Optional, Dict, List
+from pydantic import Field
+
+from .base import BaseRPCRequest
+
+
+class ParserRegisterRequest(BaseRPCRequest):
+    """Request for parser registration."""
+
+    parser_id: str = Field(description="Unique parser identifier")
+    parser_type: str = Field(description="Type of parser")
+    version: str = Field(description="Parser version")
+    capabilities: List[str] = Field(default_factory=list, description="Parser capabilities")
+    metadata: Optional[Dict[str, str]] = Field(default=None, description="Additional metadata")
+
+
+class ParserStatusRequest(BaseRPCRequest):
+    """Request for parser status."""
+
+    parser_id: str = Field(description="Parser ID to get status for")
+
+
+class ParserListRequest(BaseRPCRequest):
+    """Request for parser list."""
+
+    parser_type: Optional[str] = Field(default=None, description="Filter by parser type")
+
+
+class ParserHealthRequest(BaseRPCRequest):
+    """Request for parser health check."""
+
+    parser_id: str = Field(description="Parser ID to check health for")
+
+
+class SessionStartRequest(BaseRPCRequest):
+    """Request for session start."""
+
+    parser_id: str = Field(description="Parser ID")
+    session_type: str = Field(description="Type of session")
+    metadata: Optional[Dict[str, str]] = Field(default=None, description="Session metadata")
+
+
+class SessionEndRequest(BaseRPCRequest):
+    """Request for session end."""
+
+    session_id: str = Field(description="Session ID to end")
+
+
+class CommandExecuteRequest(BaseRPCRequest):
+    """Request for command execution."""
+
+    command_type: str = Field(description="Type of command")
+    parser_id: str = Field(description="Parser ID")
+    parameters: Dict[str, str] = Field(default_factory=dict, description="Command parameters")
+    timeout: int = Field(default=300, description="Command timeout in seconds")
+
+
+class CommandCreateRequest(BaseRPCRequest):
+    """Request for command creation."""
+
+    command_type: str = Field(description="Type of command")
+    parser_id: str = Field(description="Parser ID")
+    parameters: Dict[str, str] = Field(default_factory=dict, description="Command parameters")
+
+
+class CommandStatusRequest(BaseRPCRequest):
+    """Request for command status."""
+
+    command_id: str = Field(description="Command ID to get status for")
+
+
+class ProxyAllocateRequest(BaseRPCRequest):
+    """Request for proxy allocation."""
+
+    parser_id: str = Field(description="Parser ID requesting proxy")
+    proxy_type: str = Field(default="http", description="Type of proxy needed")
+    location: Optional[str] = Field(default=None, description="Preferred proxy location")
+
+
+class ProxyReleaseRequest(BaseRPCRequest):
+    """Request for proxy release."""
+
+    proxy_id: str = Field(description="Proxy ID to release")
+
+
+class ProxyCheckRequest(BaseRPCRequest):
+    """Request for proxy health check."""
+
+    proxy_id: str = Field(description="Proxy ID to check")

unrealon_bridge/models/responses.py

@@ -0,0 +1,88 @@
+"""
+RPC response models.
+
+Strict typing for all RPC responses - NO raw dicts allowed!
+"""
+
+from typing import Optional, List
+from pydantic import Field
+
+from .base import BaseRPCResponse
+from .parser import ParserInfo, ParserHealth
+from .session import ParserSession
+from .command import ParserCommand, CommandResult
+from .proxy import ProxyInfo
+
+
+class ParserRegisterResponse(BaseRPCResponse):
+    """Response for parser registration."""
+
+    parser_id: Optional[str] = Field(default=None, description="Registered parser ID")
+
+
+class ParserStatusResponse(BaseRPCResponse):
+    """Response for parser status request."""
+
+    parser: Optional[ParserInfo] = Field(default=None, description="Parser information")
+
+
+class ParserListResponse(BaseRPCResponse):
+    """Response for parser list request."""
+
+    parsers: List[ParserInfo] = Field(default_factory=list, description="List of parsers")
+    total: int = Field(default=0, description="Total number of parsers")
+
+
+class ParserHealthResponse(BaseRPCResponse):
+    """Response for parser health check."""
+
+    health: Optional[ParserHealth] = Field(default=None, description="Parser health information")
+
+
+class SessionStartResponse(BaseRPCResponse):
+    """Response for session start."""
+
+    session: Optional[ParserSession] = Field(default=None, description="Started session information")
+
+
+class SessionEndResponse(BaseRPCResponse):
+    """Response for session end."""
+
+    session_id: Optional[str] = Field(default=None, description="Ended session ID")
+
+
+class CommandExecuteResponse(BaseRPCResponse):
+    """Response for command execution."""
+
+    result: Optional[CommandResult] = Field(default=None, description="Command execution result")
+
+
+class CommandCreateResponse(BaseRPCResponse):
+    """Response for command creation."""
+
+    command: Optional[ParserCommand] = Field(default=None, description="Created command")
+
+
+class CommandStatusResponse(BaseRPCResponse):
+    """Response for command status request."""
+
+    command: Optional[ParserCommand] = Field(default=None, description="Command information")
+
+
+class ProxyAllocateResponse(BaseRPCResponse):
+    """Response for proxy allocation."""
+
+    proxy: Optional[ProxyInfo] = Field(default=None, description="Allocated proxy information")
+
+
+class ProxyReleaseResponse(BaseRPCResponse):
+    """Response for proxy release."""
+
+    proxy_id: Optional[str] = Field(default=None, description="Released proxy ID")
+
+
+class ProxyCheckResponse(BaseRPCResponse):
+    """Response for proxy health check."""
+
+    proxy_id: Optional[str] = Field(default=None, description="Checked proxy ID")
+    status: Optional[str] = Field(default=None, description="Proxy status")