VSCode中按红色停止按钮终止Python程序时如何触发指定函数

首先明确前提：VSCode集成终端的红色停止按钮没有固定可捕获的退出码，它的默认逻辑是先向运行进程发送SIGINT信号（和手动按Ctrl+C触发的信号完全一致），如果进程在3~5秒内未自行退出，会直接发送SIGKILL强制终止进程，被强杀时进程没有任何执行后续代码的机会，因此不要依赖退出码判断触发保存。

方案1：使用discord.py原生钩子（最推荐） #

discord.py的Bot/Client实例自带on_close协程事件，在进程收到终止信号、Bot正常断开网关连接时都会自动触发，直接把保存逻辑写在这个事件里即可，适配绝大多数场景：

import discord
from discord.ext import commands
import json

# 初始化Bot
intents = discord.Intents.default()
bot = commands.Bot(command_prefix="!", intents=intents)

# 这里写你的所有命令、业务逻辑
# 假设你全局维护了一个需要持久化的字典对象
global_user_data = {}

@bot.event
async def on_close():
    # 退出保存逻辑
    print("触发退出流程，正在保存数据...")
    with open("user_data_backup.json", "w", encoding="utf-8") as f:
        json.dump(global_user_data, f, ensure_ascii=False, indent=2)
    print("数据保存完成")

bot.run("你的Bot令牌")

注意：如果你的保存逻辑是阻塞型的耗时操作，建议用await asyncio.to_thread(你的保存函数)包装执行，避免卡住事件循环导致进程超时被强杀。

方案2：注册系统信号兜底（适配异常场景） #

如果遇到事件循环阻塞、on_close未正常触发的情况，可以手动注册系统信号处理函数兜底，覆盖SIGINT（VSCode停止按钮/Ctrl+C）、SIGTERM（系统常规终止信号）两种场景：

import asyncio
import signal
import discord
from discord.ext import commands
import json

intents = discord.Intents.default()
bot = commands.Bot(command_prefix="!", intents=intents)
global_user_data = {}

async def do_save():
    # 加标记避免重复执行保存
    if getattr(bot, "exit_saved", False):
        return
    print("收到终止信号，正在保存数据...")
    with open("user_data_backup.json", "w", encoding="utf-8") as f:
        json.dump(global_user_data, f, ensure_ascii=False, indent=2)
    bot.exit_saved = True
    await bot.close()

def _signal_handler(sig, frame):
    # 信号处理函数内不能直接执行协程，需要把任务投递到运行中的事件循环
    loop = asyncio.get_running_loop()
    loop.create_task(do_save())

# 注册信号
signal.signal(signal.SIGINT, _signal_handler)
signal.signal(signal.SIGTERM, _signal_handler)

# 原有业务逻辑、命令写在这里

@bot.event
async def on_close():
    if not getattr(bot, "exit_saved", False):
        await do_save()

bot.run("你的Bot令牌")

注意事项 #

• 保存逻辑尽量精简，耗时不要超过2秒，否则VSCode超时后会直接强杀进程，保存逻辑不会执行完。如果数据量很大，建议平时做增量落盘，退出时只把内存里的少量缓存刷到磁盘即可。
• 不要尝试捕获SystemExit异常或者判断进程退出码实现保存，强杀场景下这类逻辑完全不会生效。
• 如果修改过VSCode终端配置，可搜索terminal.integrated.confirmOnExit相关配置确认停止按钮行为，默认配置下上述方案完全可以正常触发。

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