Windows下Python多线程控制器优雅捕获CTRL+C退出方案

核心思路 #

利用Python标准库的signal捕获CTRL+C触发的SIGINT信号，结合threading.Event实现无CPU占用的阻塞等待，彻底替代原有的ENTER键退出逻辑，同时避免低效的死循环或定时轮询。

具体实现方案 #

仅依赖Python内置标准库，无需额外安装第三方包，步骤如下：

1. 创建threading.Event对象作为退出标记，用于主线程的阻塞等待
2. 定义信号处理函数，收到CTRL+C时设置退出标记并启动服务器关闭流程
3. 注册SIGINT信号（Windows下CTRL+C对应此信号）的处理逻辑
4. 启动aiosmtpd控制器后，调用Event的wait()方法阻塞主线程（无CPU消耗）
5. 退出信号触发后，调用控制器的stop()方法优雅关闭服务器

完整代码示例 #

import signal
import threading
from aiosmtpd.controller import Controller
from aiosmtpd.handlers import Sink

# 全局退出事件，用于主线程阻塞等待
exit_event = threading.Event()

def handle_sigint(signum, frame):
    """处理CTRL+C信号，触发服务器退出流程"""
    print("\n收到CTRL+C信号，正在关闭SMTP服务器...")
    exit_event.set()

def main():
    # 注册SIGINT信号处理（适配Windows系统的CTRL+C触发逻辑）
    signal.signal(signal.SIGINT, handle_sigint)

    # 初始化SMTP处理器与控制器
    handler = Sink()
    controller = Controller(handler, hostname="localhost", port=8025)
    
    # 启动服务器后台线程
    controller.start()
    print("SMTP服务器已启动，监听端口8025。按CTRL+C退出。")

    # 阻塞等待退出信号，无CPU资源占用
    exit_event.wait()

    # 优雅停止服务器，关闭监听端口与后台线程
    controller.stop()
    print("SMTP服务器已成功关闭。")

if __name__ == "__main__":
    main()

方案优势 #

• 零CPU消耗：threading.Event.wait()是操作系统级别的阻塞调用，不会像while True或time.sleep()那样持续占用CPU
• 优雅退出：通过aiosmtpd控制器的stop()方法，可正确关闭监听端口、终止后台线程，避免资源泄漏
• 兼容性强：专门适配Windows系统的信号处理逻辑，确保CTRL+C能被稳定捕获
• 无额外依赖：仅使用Python标准库，无需安装任何第三方包

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