如何通过sync命令动态导入Discord Bot的Python文件？

可以通过Python的importlib模块实现动态导入，配合自定义的模块缓存字典来管理用户脚本，完全满足你的需求——不用cogs，支持通过sync命令同步新增/删除的文件。

核心思路 #

1. 维护一个全局字典存储已导入的用户脚本，键为脚本名（不带.py后缀），值为对应的模块对象，方便快速访问和清理。
2. 执行sync命令时，做三件事：
   • 扫描./files/目录下的所有有效Python文件
   • 清理字典中已被删除的文件对应的模块
   • 导入/重新导入存在的脚本文件，更新缓存字典

代码实现示例 #

基础配置与缓存定义

import os
import importlib.util
from discord.ext import commands
from discord import Intents

# 初始化Bot，按需配置Intents
intents = Intents.default()
intents.message_content = True
bot = commands.Bot(command_prefix='!', intents=intents)

# 全局缓存：存储已导入的用户脚本
user_scripts = {}
# 脚本存储目录
SCRIPT_DIR = './files/'

Sync命令实现

@bot.command(name='sync')
async def sync_user_scripts(ctx):
    # 确保脚本目录存在
    os.makedirs(SCRIPT_DIR, exist_ok=True)
    
    # 获取当前目录下所有有效Python文件（排除__init__.py）
    current_files = [f for f in os.listdir(SCRIPT_DIR) 
                     if f.endswith('.py') and f != '__init__.py']
    current_script_names = [os.path.splitext(f)[0] for f in current_files]
    
    # 清理已删除文件对应的模块
    removed_scripts = [name for name in user_scripts if name not in current_script_names]
    for script_name in removed_scripts:
        del user_scripts[script_name]
        await ctx.send(f"已移除脚本: `{script_name}`")
    
    # 导入/重新导入脚本
    for filename in current_files:
        script_name = os.path.splitext(filename)[0]
        file_path = os.path.join(SCRIPT_DIR, filename)
        
        # 创建模块规范
        spec = importlib.util.spec_from_file_location(script_name, file_path)
        if not spec or not spec.loader:
            await ctx.send(f"加载失败: `{filename}`")
            continue
        
        # 加载并执行模块
        module = importlib.util.module_from_spec(spec)
        spec.loader.exec_module(module)
        
        # 更新缓存
        user_scripts[script_name] = module
        await ctx.send(f"已同步脚本: `{script_name}`")
    
    await ctx.send("✅ 脚本同步完成！")

脚本执行命令示例

假设用户脚本中定义了main(ctx)函数用于执行逻辑：

@bot.command(name='run')
async def execute_script(ctx, script_name: str):
    if script_name not in user_scripts:
        await ctx.send(f"❌ 未找到脚本 `{script_name}`，请先执行 `!sync`")
        return
    
    script_module = user_scripts[script_name]
    try:
        # 调用脚本中的main函数（需用户脚本提前定义）
        if hasattr(script_module, 'main'):
            await script_module.main(ctx)
        else:
            # 若未定义main，执行脚本顶层代码（谨慎使用，存在安全风险）
            exec(open(os.path.join(SCRIPT_DIR, f"{script_name}.py")).read())
            await ctx.send(f"✅ 脚本 `{script_name}` 执行完成")
    except Exception as e:
        await ctx.send(f"❌ 脚本执行出错: {str(e)}")

关键注意事项 #

• 安全风险：用户上传的Python脚本可能包含恶意代码，建议给sync和run命令添加管理员权限限制，或者使用沙箱（如restrictedPython）隔离执行环境。
• 更新机制：每次sync都会重新导入脚本，修改后的文件会自动覆盖旧模块。
• 模块清理：当用户删除./files/下的文件后，下一次sync会自动从缓存字典中移除对应的模块，避免调用不存在的脚本。

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