add windows client
This commit is contained in:
parent
43f61be8b8
commit
976eb1baf1
|
|
@ -0,0 +1,26 @@
|
|||
@echo off
|
||||
setlocal
|
||||
|
||||
echo === Activating virtual environment ===
|
||||
call venv310\Scripts\activate.bat
|
||||
if errorlevel 1 (
|
||||
echo [ERROR] Failed to activate virtual environment!
|
||||
pause
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
echo === Installing PyInstaller (if not present) ===
|
||||
pip show pyinstaller >nul 2>&1
|
||||
if errorlevel 1 (
|
||||
echo Installing PyInstaller...
|
||||
pip install pyinstaller
|
||||
) else (
|
||||
echo PyInstaller already installed.
|
||||
)
|
||||
|
||||
echo === Building win_udp_listener.exe ===
|
||||
pyinstaller --noconsole --onefile win_udp_listener.py
|
||||
|
||||
echo.
|
||||
echo === Build complete! Check the dist\ directory. ===
|
||||
pause
|
||||
|
|
@ -0,0 +1,66 @@
|
|||
import socket
|
||||
import json
|
||||
import threading
|
||||
import webbrowser
|
||||
import tkinter as tk
|
||||
from tkinter import messagebox
|
||||
|
||||
UDP_PORT = 9876
|
||||
popup_shown = False # 确保只弹一次
|
||||
|
||||
def open_browser(url):
|
||||
try:
|
||||
webbrowser.open(url)
|
||||
except Exception as e:
|
||||
print(f"Failed to open browser: {e}")
|
||||
|
||||
def listen_udp():
|
||||
global popup_shown
|
||||
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
sock.bind(("", UDP_PORT))
|
||||
|
||||
print(f"Listening for UDP broadcasts on port {UDP_PORT}...")
|
||||
|
||||
while True:
|
||||
try:
|
||||
data, addr = sock.recvfrom(1024)
|
||||
print(f"Received raw data from {addr}: {data}")
|
||||
msg = json.loads(data.decode("utf-8"))
|
||||
|
||||
if msg.get("type") == "ai_server_announce":
|
||||
ip = msg.get("ip")
|
||||
port = msg.get("port")
|
||||
name = msg.get("name", "Unknown")
|
||||
|
||||
url = f"http://{ip}:80"
|
||||
print(f"Discovered {name} at {url}")
|
||||
|
||||
if not popup_shown:
|
||||
popup_shown = True
|
||||
|
||||
def show_popup():
|
||||
if messagebox.askyesno("发现AI服务器", f"名称: {name}\n地址: {url}\n\n是否打开浏览器访问?"):
|
||||
open_browser(url)
|
||||
|
||||
root.after(0, show_popup)
|
||||
|
||||
except json.JSONDecodeError:
|
||||
print("Failed to decode JSON")
|
||||
except Exception as e:
|
||||
print(f"Error while receiving UDP: {e}")
|
||||
|
||||
# 初始化 Tkinter 主窗口(显示)
|
||||
root = tk.Tk()
|
||||
root.title("AI 服务器发现工具")
|
||||
root.geometry("400x200")
|
||||
label = tk.Label(root, text="正在监听 AI 服务器广播...\n端口: 9876", font=("Arial", 12))
|
||||
label.pack(pady=60)
|
||||
|
||||
# 启动监听线程
|
||||
listener_thread = threading.Thread(target=listen_udp, daemon=True)
|
||||
listener_thread.start()
|
||||
|
||||
# 启动 UI 主循环
|
||||
root.mainloop()
|
||||
Loading…
Reference in New Issue