点击Pystray菜单退出含Asyncio+FastAPI的Tkinter Python程序失败

完整可行代码实现 #

import threading
import tkinter as tk
from tkinter import ttk
import pystray
from PIL import Image, ImageDraw
import uvicorn
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
import sys
import signal

# FastAPI 实例配置
app = FastAPI()
app.mount("/static", StaticFiles(directory="static"), name="static")
server = None  # 保存Uvicorn服务器实例，用于后续关闭

def start_fastapi():
    """启动FastAPI服务器的线程函数"""
    global server
    config = uvicorn.Config(app, host="0.0.0.0", port=8000)
    server = uvicorn.Server(config)
    server.run()

class UserManagerGUI:
    def __init__(self, root):
        self.root = root
        self.root.title("用户管理")
        self.root.withdraw()  # 初始隐藏主窗口
        
        # 构建示例GUI
        self.frame = ttk.Frame(root, padding="10")
        self.frame.pack(fill=tk.BOTH, expand=True)
        ttk.Label(self.frame, text="用户管理后台").pack(pady=5)
        ttk.Button(self.frame, text="显示管理窗口", command=self.show_window).pack(pady=5)
        
        # 启动系统托盘
        self.setup_tray()

    def show_window(self):
        """显示主窗口"""
        self.root.deiconify()
    
    def setup_tray(self):
        """创建系统托盘图标及菜单"""
        # 生成简易托盘图标（可替换为自定义图片）
        image = Image.new('RGB', (64, 64), (77, 148, 255))
        draw = ImageDraw.Draw(image)
        draw.text((10, 10), "UM", fill=(255, 255, 255))
        
        # 托盘菜单
        menu = (
            pystray.MenuItem("显示窗口", self.show_window),
            pystray.MenuItem("退出", self.handle_exit)
        )
        
        self.tray_icon = pystray.Icon("user_manager", image, "用户管理", menu)
        # 托盘运行单独开守护线程，避免阻塞Tk主循环
        threading.Thread(target=self.tray_icon.run, daemon=True).start()
    
    def handle_exit(self, icon, item):
        """统一处理退出逻辑"""
        # 1. 停止托盘图标
        icon.stop()
        
        # 2. 优雅关闭FastAPI服务器
        if server:
            server.should_exit = True
        
        # 3. 终止Tk事件循环并销毁窗口
        self.root.quit()
        self.root.destroy()
        
        # 4. 强制退出进程，确保无残留
        sys.exit(0)

def main():
    # 处理系统终止信号（Ctrl+C/系统kill）
    def signal_handler(signum, frame):
        gui.handle_exit(gui.tray_icon, None)
    
    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    
    # 初始化Tk主窗口
    root = tk.Tk()
    global gui
    gui = UserManagerGUI(root)
    
    # 启动FastAPI守护线程
    fastapi_thread = threading.Thread(target=start_fastapi, daemon=True)
    fastapi_thread.start()
    
    # 启动Tk主循环
    root.mainloop()
    
    # 等待FastAPI线程收尾（可选，守护线程随主进程退出）
    fastapi_thread.join()

if __name__ == "__main__":
    main()

关键解决要点 #

1. FastAPI优雅关闭

   • 不能直接用uvicorn.run(app)启动，必须通过uvicorn.Server实例启动并保存引用，利用server.should_exit = True触发优雅关闭，避免强制终止导致的资源泄漏。

2. Tkinter与Pystray事件循环协调

   • Pystray的run()方法会阻塞线程，必须放在守护线程中运行，否则会卡死Tk的mainloop()。
   • 退出时先停止托盘事件循环，再终止Tk主循环并销毁窗口，顺序不能颠倒。

3. 多线程与进程退出保障

   • FastAPI线程设置为daemon=True，确保主进程退出时该线程自动终止。
   • 最后调用sys.exit(0)强制结束进程，避免因后台线程未完全停止导致的进程残留。

4. 系统信号处理

   • 监听SIGINT和SIGTERM信号，统一调用退出逻辑，确保手动终止进程时也能正常清理资源。

内容的提问来源于stack exchange，提问作者edwin_uestc