如何为持续运行的discord.py机器人添加启动加载进度条

核心思路 #

无需重构现有代码，利用discord.py的setup_hook()异步钩子（v2.0+版本支持），在机器人登录Discord前执行初始化任务，同时通过单独线程实时输出进度条，完全不阻塞机器人的异步事件循环，既满足进度展示需求，又保证程序持续运行。

实现步骤 #

1. 编写轻量进度条工具 #

不用额外安装第三方库，自己实现一个终端友好的文本进度条，支持实时更新：

import sys

def update_progress(progress, total, bar_length=40):
    percent = float(progress) / total
    hashes = '#' * int(round(percent * bar_length))
    spaces = ' ' * (bar_length - len(hashes))
    sys.stdout.write(f'\r启动进度: [{hashes}{spaces}] {int(percent*100)}%')
    sys.stdout.flush()
    if progress == total:
        sys.stdout.write('\n启动完成！机器人开始运行\n')

2. 改造机器人代码，整合进度条 #

将现有启动阶段的初始化逻辑拆分为独立任务，配合setup_hook()执行，同时启动进度条线程：

import discord
from discord.ext import commands
import threading
import time
import asyncio

# 进度控制变量，根据实际初始化步骤调整总数
current_step = 0
total_steps = 3

bot = commands.Bot(command_prefix='!', intents=discord.Intents.all())

def progress_thread():
    global current_step
    while current_step < total_steps:
        update_progress(current_step, total_steps)
        time.sleep(0.1)
    update_progress(total_steps, total_steps)

@bot.event
async def on_ready():
    print(f'已登录为 {bot.user}')

# 以下为示例初始化任务，替换成你自己的启动逻辑
async def load_cogs():
    global current_step
    # 加载你的cogs文件
    await bot.load_extension('cogs.admin')
    await bot.load_extension('cogs.fun')
    current_step += 1

async def init_database():
    global current_step
    # 数据库连接、表初始化等操作
    await asyncio.sleep(1)  # 模拟耗时操作
    current_step += 1

async def load_config():
    global current_step
    # 读取配置文件、初始化全局参数
    await asyncio.sleep(0.5)  # 模拟耗时操作
    current_step += 1

@bot.setup_hook
async def setup():
    # 启动守护线程运行进度条，机器人退出时自动终止
    threading.Thread(target=progress_thread, daemon=True).start()
    # 按顺序执行所有初始化任务
    await load_cogs()
    await init_database()
    await load_config()

# 启动机器人
bot.run('你的机器人令牌')

3. 适配现有代码的注意点 #

• 把你原本在启动阶段的同步代码，用asyncio.to_thread()包装成异步任务，避免阻塞事件循环，比如：await asyncio.to_thread(你的同步初始化函数)
• 根据实际初始化步骤的数量，调整total_steps的数值
• 如果使用discord.py 2.0以下版本，可以把初始化逻辑放在bot.run()之前执行，同时启动进度条线程，注意确保初始化完成后再调用bot.run()

内容的提问来源于stack exchange，提问作者Dr. Glitchgo