mirror of
https://github.com/Viren070/mediaflow-proxy.git
synced 2025-12-01 23:22:12 +01:00
mhdzumair
035ae1a4c7
Changed the proxy configuration system to a more flexible transport configuration using HTTPX routing. Introduced new settings for global proxy enablement and robust pattern-based proxy and SSL configurations. Updated README.md with new examples and detailed explanations of the new configurations.
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
from typing import Dict, Optional
|
|
|
|
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[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 # The password for accessing the API endpoints.
|
|
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()
|