zydi-lab/app/docs/cores/db.md

cores/db.py

文件说明

该文件包含数据库相关的工具和配置，主要处理MongoDB和Redis的连接和操作。

主要组件

Redis数据库映射

REDIS_DB_MAPPING: 详细的Redis数据库映射配置，包含：

• 实验状态 (db: 199)
• 原始数据 (db: 200)
• 实验分析 (db: 201)
• 项目分析 (db: 202)
• 文献分析 (db: 203)
• 文献汇总 (db: 204)
• 项目备忘录 (db: 205)
• 实验备忘录 (db: 206)
• 聊天历史 (db: 207)
• 问答任务 (db: 208)

工具类

• PyObjectId: 用于Pydantic模型中处理MongoDB ObjectId的自定义类
• Database: 数据库连接管理类

数据库操作函数

• get_database(): 获取MongoDB数据库实例
• connect_to_mongo(): 建立MongoDB连接
• close_mongo_connection(): 关闭MongoDB连接
• get_redis(): 获取Redis连接
• get_redis_key(): 获取Redis键名和数据库编号
• get_redis_db_number(): 获取Redis数据库编号