mirror of
https://github.com/UrloMythus/UnHided.git
synced 2026-04-09 02:40:47 +00:00
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from typing import Dict, Optional, Union
|
|
|
|
import httpx
|
|
from pydantic import BaseModel, Field
|
|
from pydantic_settings import BaseSettings
|
|
|
|
|
|
class RouteConfig(BaseModel):
|
|
"""Configuration for a specific route"""
|
|
|
|
proxy: bool = True
|
|
proxy_url: Optional[str] = None
|
|
verify_ssl: bool = True
|
|
|
|
|
|
class TransportConfig(BaseSettings):
|
|
"""Main proxy configuration"""
|
|
|
|
proxy_url: Optional[str] = Field(
|
|
None, description="Primary proxy URL. Example: socks5://user:pass@proxy:1080 or http://proxy:8080"
|
|
)
|
|
all_proxy: bool = Field(False, description="Enable proxy for all routes by default")
|
|
transport_routes: Dict[str, RouteConfig] = Field(
|
|
default_factory=dict, description="Pattern-based route configuration"
|
|
)
|
|
|
|
def get_mounts(
|
|
self, async_http: bool = True
|
|
) -> Dict[str, Optional[Union[httpx.HTTPTransport, httpx.AsyncHTTPTransport]]]:
|
|
"""
|
|
Get a dictionary of httpx mount points to transport instances.
|
|
"""
|
|
mounts = {}
|
|
transport_cls = httpx.AsyncHTTPTransport if async_http else httpx.HTTPTransport
|
|
|
|
# Configure specific routes
|
|
for pattern, route in self.transport_routes.items():
|
|
mounts[pattern] = transport_cls(
|
|
verify=route.verify_ssl, proxy=route.proxy_url or self.proxy_url if route.proxy else None
|
|
)
|
|
|
|
# Set default proxy for all routes if enabled
|
|
if self.all_proxy:
|
|
mounts["all://"] = transport_cls(proxy=self.proxy_url)
|
|
|
|
return mounts
|
|
|
|
class Config:
|
|
env_file = ".env"
|
|
extra = "ignore"
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
api_password: str | None = None # The password for protecting the API endpoints.
|
|
log_level: str = "INFO" # The logging level to use.
|
|
transport_config: TransportConfig = Field(default_factory=TransportConfig) # Configuration for httpx transport.
|
|
enable_streaming_progress: bool = False # Whether to enable streaming progress tracking.
|
|
|
|
user_agent: str = (
|
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3" # The user agent to use for HTTP requests.
|
|
)
|
|
|
|
class Config:
|
|
env_file = ".env"
|
|
extra = "ignore"
|
|
|
|
|
|
settings = Settings()
|