Python多数据源Discord词典机器人查询功能实现求助

核心思路 #

将所有数据源的get_response函数统一收集，查询时依次调用各函数，直到获取到有效结果或遍历完所有数据源。

具体实现步骤 #

1. 批量导入数据源模块 #

在bot.py中可以选择显式导入或动态导入两种方式加载数据源：

# 显式导入（适合数据源数量固定的场景）
from pool0_help import get_response as get_help_response
from pool2_B import get_response as get_b_response
# 按命名规则继续导入其他数据源...

# 动态导入（适合数据源数量不固定，自动匹配命名规则加载）
import os
import importlib

response_functions = []
# 遍历当前目录下以pool开头的py文件
for filename in os.listdir('.'):
    if filename.startswith('pool') and filename.endswith('.py') and filename != '__init__.py':
        module_name = filename[:-3]  # 移除.py后缀
        module = importlib.import_module(module_name)
        if hasattr(module, 'get_response'):
            response_functions.append(module.get_response)

2. 编写统一查询函数 #

封装遍历逻辑，依次调用各数据源的查询函数：

def query_all_sources(query):
    for get_response in response_functions:
        result = get_response(query)
        if result:  # 找到有效结果则立即返回
            return result
    return "未找到匹配的词汇或帮助信息"  # 所有数据源无匹配时的默认回复

3. 修改消息处理逻辑 #

将原来调用单个get_response的代码替换为新的统一查询函数：

# 假设原消息处理命令如下
@bot.command()
async def lookup(ctx, *, query):
    # 替换单数据源调用为多数据源查询
    response = query_all_sources(query)
    await ctx.send(response)

优化建议 #

• 优先级调整：若某些数据源（如帮助类）需要优先查询，可调整response_functions的列表顺序
• 异步优化：如果get_response是耗时操作，建议改为异步函数，避免阻塞机器人运行
• 缓存机制：对高频查询结果做缓存，减少重复遍历数据源的开销

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