| """Proxy management module""" |
| from typing import Optional |
| from ..core.database import Database |
| from ..core.models import ProxyConfig |
|
|
| class ProxyManager: |
| """Proxy configuration manager""" |
|
|
| def __init__(self, db: Database): |
| self.db = db |
|
|
| async def get_proxy_url(self) -> Optional[str]: |
| """Get proxy URL if enabled, otherwise return None""" |
| config = await self.db.get_proxy_config() |
| if config and config.enabled and config.proxy_url: |
| return config.proxy_url |
| return None |
|
|
| async def update_proxy_config(self, enabled: bool, proxy_url: Optional[str]): |
| """Update proxy configuration""" |
| await self.db.update_proxy_config(enabled, proxy_url) |
|
|
| async def get_proxy_config(self) -> ProxyConfig: |
| """Get proxy configuration""" |
| return await self.db.get_proxy_config() |
|
|
