sub / app.py
sfun's picture
Update app.py
8e6dd44 verified
raw
history blame
3.17 kB
import re
import yaml
import requests
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import urlparse, parse_qs
def extract_and_transform_proxies(input_text):
proxies_match = re.search(r'proxies:\s*\n((?:{.*\n)*)', input_text)
if not proxies_match:
return "未找到proxies部分"
proxies_text = proxies_match.group(1)
try:
# 尝试解析YAML
proxies_list = yaml.safe_load(proxies_text)
# 检查proxies_list是否为None或不是列表
if proxies_list is None or not isinstance(proxies_list, list):
# 如果不是有效的YAML列表,尝试使用正则表达式解析
proxies_list = re.findall(r'{(.*?)}', proxies_text, re.DOTALL)
proxies_list = [dict(item.split(': ') for item in proxy.split(', ')) for proxy in proxies_list]
except yaml.YAMLError as e:
return f"YAML解析错误: {str(e)}"
if not proxies_list:
return "未找到有效的代理配置"
transformed_proxies = []
for proxy in proxies_list:
if isinstance(proxy, str):
# 如果proxy是字符串,说明使用了正则表达式解析,需要进一步处理
proxy = dict(item.split(': ') for item in proxy.split(', '))
if proxy.get('type') == 'ss':
name = proxy.get('name', '')
server = proxy.get('server', '')
port = proxy.get('port', '')
cipher = proxy.get('cipher', '')
password = proxy.get('password', '')
udp = 'true' if proxy.get('udp') == 'true' else 'false'
transformed = f"{name} = ss, {server}, {port}, encrypt-method={cipher}, password={password}, udp-relay={udp}"
transformed_proxies.append(transformed)
return "\n".join(transformed_proxies) if transformed_proxies else "未找到有效的SS代理配置"
class RequestHandler(BaseHTTPRequestHandler):
def do_GET(self):
parsed_path = urlparse(self.path)
query_params = parse_qs(parsed_path.query)
if 'url' in query_params:
url = query_params['url'][0]
try:
response = requests.get(url)
response.raise_for_status() # 这将抛出HTTPError,如果响应状态码不是200
input_text = response.text
result = extract_and_transform_proxies(input_text)
self.send_response(200)
self.send_header('Content-type', 'text/plain; charset=utf-8')
self.end_headers()
self.wfile.write(result.encode('utf-8'))
except requests.RequestException as e:
self.send_error(500, f"Error fetching data: {str(e)}")
else:
self.send_error(400, "Missing 'url' parameter")
def log_message(self, format, *args):
# 重写此方法以禁用默认的服务器日志
pass
def run_server(port=8080):
server_address = ('', port)
httpd = HTTPServer(server_address, RequestHandler)
print(f"Server running on port {port}")
httpd.serve_forever()
if __name__ == "__main__":
run_server()