mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
209f16af76
* style: remove necessary imports * style: fix F841 * style: fix F401 * style: fix F811 * style: fix E402 * style: fix E721 * style: fix E722 * style: fix E722 * style: fix F541 * style: ruff format * style: all passed * style: add ruff in deps * style: more ignores in ruff.toml * style: add pre-commit
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
from ..core import app
|
|
|
|
|
|
class ProxyManager:
|
|
"""代理管理器"""
|
|
|
|
ap: app.Application
|
|
|
|
forward_proxies: dict[str, str]
|
|
|
|
def __init__(self, ap: app.Application):
|
|
self.ap = ap
|
|
|
|
self.forward_proxies = {}
|
|
|
|
async def initialize(self):
|
|
self.forward_proxies = {
|
|
'http://': os.getenv('HTTP_PROXY') or os.getenv('http_proxy'),
|
|
'https://': os.getenv('HTTPS_PROXY') or os.getenv('https_proxy'),
|
|
}
|
|
|
|
if (
|
|
'http' in self.ap.instance_config.data['proxy']
|
|
and self.ap.instance_config.data['proxy']['http']
|
|
):
|
|
self.forward_proxies['http://'] = self.ap.instance_config.data['proxy'][
|
|
'http'
|
|
]
|
|
if (
|
|
'https' in self.ap.instance_config.data['proxy']
|
|
and self.ap.instance_config.data['proxy']['https']
|
|
):
|
|
self.forward_proxies['https://'] = self.ap.instance_config.data['proxy'][
|
|
'https'
|
|
]
|
|
|
|
# 设置到环境变量
|
|
os.environ['HTTP_PROXY'] = self.forward_proxies['http://'] or ''
|
|
os.environ['HTTPS_PROXY'] = self.forward_proxies['https://'] or ''
|
|
|
|
def get_forward_proxies(self) -> dict:
|
|
return self.forward_proxies.copy()
|