ccproxy-api 0.1.0__py3-none-any.whl

ccproxy/config/scheduler.py

@@ -0,0 +1,89 @@
+"""Scheduler configuration settings."""
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class SchedulerSettings(BaseSettings):
+    """
+    Configuration settings for the unified scheduler system.
+
+    Controls global scheduler behavior and individual task configurations.
+    Settings can be configured via environment variables with SCHEDULER__ prefix.
+    """
+
+    # Global scheduler settings
+    enabled: bool = Field(
+        default=True,
+        description="Whether the scheduler system is enabled",
+    )
+
+    max_concurrent_tasks: int = Field(
+        default=10,
+        ge=1,
+        le=100,
+        description="Maximum number of tasks that can run concurrently",
+    )
+
+    graceful_shutdown_timeout: float = Field(
+        default=30.0,
+        ge=1.0,
+        le=300.0,
+        description="Timeout in seconds for graceful task shutdown",
+    )
+
+    # Pricing updater task settings
+    pricing_update_enabled: bool = Field(
+        default=True,
+        description="Whether pricing cache update task is enabled",
+    )
+
+    pricing_update_interval_hours: int = Field(
+        default=24,
+        ge=1,
+        le=168,  # Max 1 week
+        description="Interval in hours between pricing cache updates",
+    )
+
+    pricing_force_refresh_on_startup: bool = Field(
+        default=False,
+        description="Whether to force pricing refresh immediately on startup",
+    )
+
+    # Observability tasks (migrated from ObservabilitySettings)
+    pushgateway_enabled: bool = Field(
+        default=False,
+        description="Whether pushgateway metrics pushing task is enabled",
+    )
+
+    pushgateway_interval_seconds: float = Field(
+        default=60.0,
+        ge=1.0,
+        le=3600.0,  # Max 1 hour
+        description="Interval in seconds between pushgateway metric pushes",
+    )
+
+    pushgateway_max_backoff_seconds: float = Field(
+        default=300.0,
+        ge=1.0,
+        le=1800.0,  # Max 30 minutes
+        description="Maximum backoff delay for failed pushgateway operations",
+    )
+
+    stats_printing_enabled: bool = Field(
+        default=False,
+        description="Whether stats printing task is enabled",
+    )
+
+    stats_printing_interval_seconds: float = Field(
+        default=300.0,
+        ge=1.0,
+        le=3600.0,  # Max 1 hour
+        description="Interval in seconds between stats printing",
+    )
+
+    class Config:
+        """Pydantic configuration."""
+
+        env_prefix = "SCHEDULER__"
+        case_sensitive = False

ccproxy/config/security.py

@@ -0,0 +1,14 @@
+"""Security configuration settings."""
+
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+
+class SecuritySettings(BaseModel):
+    """Security-specific configuration settings."""
+
+    auth_token: str | None = Field(
+        default=None,
+        description="Bearer token for API authentication (optional)",
+    )

ccproxy/config/server.py

@@ -0,0 +1,81 @@
+"""Server configuration settings."""
+
+from pydantic import BaseModel, Field, field_validator
+
+
+class ServerSettings(BaseModel):
+    """Server-specific configuration settings."""
+
+    host: str = Field(
+        default="127.0.0.1",
+        description="Server host address",
+    )
+
+    port: int = Field(
+        default=8000,
+        description="Server port number",
+        ge=1,
+        le=65535,
+    )
+
+    workers: int = Field(
+        default=1,
+        description="Number of worker processes",
+        ge=1,
+        le=32,
+    )
+
+    reload: bool = Field(
+        default=False,
+        description="Enable auto-reload for development",
+    )
+
+    log_level: str = Field(
+        default="INFO",
+        description="Logging level",
+    )
+
+    log_format: str = Field(
+        default="auto",
+        description="Logging output format: 'rich' for development, 'json' for production, 'auto' for automatic selection",
+    )
+
+    log_show_path: bool = Field(
+        default=False,
+        description="Whether to show module path in logs (automatically enabled for DEBUG level)",
+    )
+
+    log_show_time: bool = Field(
+        default=True,
+        description="Whether to show timestamps in logs",
+    )
+
+    log_console_width: int | None = Field(
+        default=None,
+        description="Optional console width override for Rich output",
+    )
+
+    log_file: str | None = Field(
+        default=None,
+        description="Path to JSON log file. If specified, logs will be written to this file in JSON format",
+    )
+
+    @field_validator("log_level")
+    @classmethod
+    def validate_log_level(cls, v: str) -> str:
+        """Validate and normalize log level."""
+        upper_v = v.upper()
+        valid_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
+        if upper_v not in valid_levels:
+            raise ValueError(f"Invalid log level: {v}. Must be one of {valid_levels}")
+        return upper_v
+
+    @field_validator("log_format")
+    @classmethod
+    def validate_log_format(cls, v: str) -> str:
+        """Validate and normalize log format."""
+        lower_v = v.lower()
+        valid_formats = ["auto", "rich", "json", "plain"]
+        if lower_v not in valid_formats:
+            raise ValueError(f"Invalid log format: {v}. Must be one of {valid_formats}")
+        return lower_v