使用os.walk自动加载discord.py Cogs失败，求原因与解决方法

问题背景 #

作为Python新手开发discord.py机器人时，尝试用os.walk自动扫描加载Cogs扩展，启动后却没有任何Cogs被加载。后续将所有Cogs整理到单个文件夹后，代码能运行，但之前针对层级文件夹的处理逻辑存在明显错误，还因过度设计增加了不必要的工作量。

原代码错误分析 #

先看最初的错误代码片段：

@bot.event
async def setup_hook():
    for dirs in next(os.walk('.'))[1]:
        for i in range(len(dirs)):
            if dirs[i].startswith("."):
                for file in os.listdir(''.join(dirs[i])):
                    if file.endswith(".py"):
                        await bot.load_extension(f"{dirs[i]}.{file[:-3]}")
    print('Cogs should be done now.')

这里的核心错误有3点：

1. 循环变量理解错误：next(os.walk('.'))[1]返回的是当前目录下的子文件夹列表，循环变量dirs是单个文件夹名称（比如'src'），而非文件夹列表。
2. 字符遍历逻辑混乱：for i in range(len(dirs))是在遍历文件夹名称的每个字符（比如'src'会遍历s、r、c），完全偏离了遍历文件夹的初衷。
3. 隐藏文件夹判断失效：dirs[i].startswith(".")是判断单个字符是否以.开头，根本无法匹配隐藏文件夹（比如.hidden），导致逻辑完全失效。

层级结构Cogs的正确加载方案 #

如果需要处理src/coggroup1/type1cogshere.py这种多层级的Cogs结构，可使用以下代码实现自动扫描：

import os
import pathlib
from discord.ext import commands

@bot.event
async def setup_hook():
    # 遍历指定根目录下的所有子文件夹
    for root, _, files in os.walk("src"):
        for file in files:
            # 只处理.py结尾且非下划线开头的文件（避免加载__init__.py等辅助文件）
            if file.endswith(".py") and not file.startswith("_"):
                # 将系统路径分隔符替换为点，生成符合要求的模块路径
                module_path = pathlib.Path(root, file).stem.replace(os.sep, ".")
                await bot.load_extension(module_path)
    print("所有层级Cogs加载完成")

这段代码的逻辑：

• os.walk("src")会递归遍历src下的所有子目录，root记录当前遍历的目录路径，files是该目录下的所有文件。
• 通过pathlib工具将系统路径转换为discord.py要求的点分模块路径（比如src/coggroup1/xxx.py转为src.coggroup1.xxx）。
• 过滤掉下划线开头的文件，避免加载不必要的辅助文件。

单文件夹Cogs的简化写法 #

如果已将所有Cogs集中到cogs单个文件夹，可使用更简洁的代码：

@bot.event
async def setup_hook():
    for file in os.listdir("cogs"):
        if file.endswith(".py") and not file.startswith("_"):
            await bot.load_extension(f"cogs.{file[:-3]}")
    print("单文件夹Cogs加载完成")

这段代码仅需判断文件后缀，同时排除下划线开头的文件，避免无效加载尝试。

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