update

main.py

@@ -1,23 +1,7 @@
-# -*- coding: utf-8 -*-
-from urllib.parse import quote
-import subprocess
+from nicegui import ui
 import httpx
-import tkinter as tk
 
-
-nodes = [
-    ["192.168.31.201", ['58001', '58002', '58003', '58004', '58005', '58006', '58007', '58008', '58009',
-                        '58010'], ],
-    ["192.168.31.201", ['32001', '32002', '32003', '32004', '32005', '32006', '32007', '32008', '32009',
-                        '32010', '32011', '32012']],
-    ["127.0.0.1", ['17888']]
-]
-
-selected_nodes = nodes[0]
-output_messages = []  # 用于存储输出信息
-
-
-def patch_config(url_and_port):
+async def check_now_node(url_and_port):
     url = f"http://{url_and_port}"
     key = "/api/configs"
     full_url = url + key
@@ -25,115 +9,71 @@ def patch_config(url_and_port):
     data = {"mode": "Global"}
     headers = {"Content-Type": "application/json"}
 
-    try:
-        response = httpx.patch(full_url, json=data, headers=headers)
-        state = response.status_code
-        if state == 204:
-            pass
-            # print(f"{url_and_port}: 切换全局代理 OK")
-        else:
-            raise Exception(f"请求失败: {response.status_code}")
-    except httpx.HTTPError as exc:
-        print(f"请求失败: {exc}")
-        exit(1)
-
-
-def check_proxy(proxy_url, choose_proxy):
-    encode_proxy_name = quote(choose_proxy, safe="")
-    command = [
-        "curl",
-        "-X", "GET",
-        f"{proxy_url}/api/proxies/{encode_proxy_name}/delay?timeout=5000&url=http:%2F%2Fwww.gstatic.com%2Fgenerate_204"
-    ]
-
-    try:
-        result = subprocess.run(command, capture_output=True, text=True, check=True)
-        # print("Output:", result.stdout)
-        if 'Timeout' in result.stdout:
-            return "Timeout"
-        res = eval(result.stdout).get("meanDelay")
-        return f"meanDelay: {res}"
-    except subprocess.CalledProcessError as e:
-        print("Error:", e.stderr)
-        return str(e)
-
-
-def check_now_proxy(url_and_port):
-    url = f"http://{url_and_port}/api/proxies"
-    print(url)
-    headers = {
-        "Accept": "application/json, text/plain, */*",
-        "Accept-Encoding": "gzip, deflate, br, zstd",
-        "Accept-Language": "zh-CN,zh;q=0.8",
-        "Connection": "keep-alive",
-        "Host": url_and_port,
-        "Referer": f"http://{url_and_port}/",
-        "Sec-CH-UA": '"Chromium";v="134", "Not:A-Brand";v="24", "Brave";v="134"',
-        "Sec-CH-UA-Mobile": "?0",
-        "Sec-CH-UA-Platform": '"macOS"',
-        "Sec-Fetch-Dest": "empty",
-        "Sec-Fetch-Mode": "cors",
-        "Sec-Fetch-Site": "same-origin",
-        "Sec-GPC": "1",
-        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/134.0.0.0 Safari/537.36"
-    }
-
-    try:
-        response = httpx.get(url, headers=headers)
-        json_data = response.json()
-        if not json_data or response.status_code != 200:
-            print("JSON data is empty or request failed")
-            return
-
-        proxies = json_data.get("proxies")
-        proxy_global = proxies.get("GLOBAL")
-        now_proxy = proxy_global.get("now")
-
-        return now_proxy
-
-    except httpx.RequestError as e:
-        print(f"Request failed: {e}")
-        return False
-
-
-def run():
-    ip = selected_nodes[0]
-    for port in selected_nodes[1]:
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.patch(full_url, json=data, headers=headers)
+            if response.status_code != 204:
+                raise Exception(f"请求失败: {response.status_code}")
+        except httpx.HTTPError as exc:
+            print(f"请求失败: {exc}")
+            return None
+
+        url_proxies = f"http://{url_and_port}/api/proxies"
+        headers_proxies = {
+            # 这里可以复用之前的headers
+            "Accept": "application/json, text/plain, */*",
+            "Host": url_and_port,
+            # 其他可按需设置
+        }
+
+        try:
+            response = await client.get(url_proxies, headers=headers_proxies)
+            json_data = response.json()
+            if not json_data or response.status_code != 200:
+                print("JSON data is empty or request failed")
+                return None
+
+            proxies = json_data.get("proxies")
+            proxy_global = proxies.get("GLOBAL")
+            now_proxy = proxy_global.get("now")
+
+            return now_proxy
+        except httpx.RequestError as e:
+            print(f"Request failed: {e}")
+            return None
+
+async def handle_check_now_node():
+    ip = '192.168.31.201'
+    port_list = ['58001', '58002', '58003', '58004', '58005', '58006', '58007', '58008', '58009', '58010']
+    results = []
+
+    for port in port_list:
         url_and_port = f"{ip}:{port}"
+        now_proxy = await check_now_node(url_and_port)
+        results.append(f"{url_and_port}: \t{now_proxy}")
 
-        # 切换全局代理
-        patch_config(url_and_port)
-
-        # 获取当前代理节点
-        now_proxy = check_now_proxy(url_and_port)
-
-        # 检测当前代理节点的延迟
-        check_result = check_proxy(url_and_port, now_proxy)
-
-        message = f"{url_and_port} --- {now_proxy} --- {check_result}\n{'*' * 88}\n\n"
-        print(message)
-        output_messages.append(message)  # 将输出信息添加到列表中
+    msg = "\n".join(results)
+    print(msg)
 
+    dialog = ui.dialog()
+    with dialog:
+        ui.label(msg).classes("whitespace-pre-wrap text-left")
 
-def display_output():
-    run()  # 执行主逻辑
-    output_text.delete(1.0, tk.END)  # 清空文本框
-    for message in output_messages:  # 将所有输出信息显示到文本框中
-        output_text.insert(tk.END, message)
+    dialog.props("max-width=700px")
+    dialog.open()
 
+def handle_switch_node():
+    print("Switch IP button clicked")
 
-# 创建主窗口
-root = tk.Tk()
-root.title("Proxy Checker")
-root.geometry("800x900")
+def main():
+    with ui.row().classes("mx-auto"):
+        ui.button("Check Now Node", on_click=handle_check_now_node).classes("mt-1")
+        ui.button("Switch Node", on_click=handle_switch_node).classes("mt-1")
 
-# 创建按钮
-check_button = tk.Button(root, text="Check Node", command=display_output)
-check_button.pack(pady=10)
+        ui.label("Filter Node").classes("text-lg font-bold text-blue-500 mt-4")
+        ui.input(value="JP").classes("w-96")
 
-# 创建文本框（不带滚动条）
-output_text = tk.Text(root, width=100, height=60)
-output_text.pack(pady=10)
+    ui.run(port=48345)
 
-# 启动主循环
-root.mainloop()
+if __name__ in {"__main__", "__mp_main__"}:
+    main()