3 | MuiceBot
4 |Muice-Chatbot 的 NoneBot2 实现
5 |
3 | Muice-Chatbot 的 NoneBot2 实现
5 |
8 | 
9 | 
10 | 
11 | 
12 | 
15 | 
16 | 
17 | 
142 | 
143 |
144 | Star History：
145 |
146 | [](https://star-history.com/#Moemu/MuiceBot&Date)
--------------------------------------------------------------------------------
/bot.py:
--------------------------------------------------------------------------------
1 | import importlib
2 | import sys
3 |
4 | import nonebot
5 | from nonebot.drivers import Driver
6 |
7 | PLUGINS_CONFIG_PATH = "./configs/plugins.yml"
8 |
9 |
10 | def load_specified_adapter(driver: Driver, adapter: str):
11 | """
12 | 加载指定的 Nonebot 适配器
13 | """
14 | try:
15 | module = importlib.import_module(adapter)
16 | adapter = module.Adapter
17 | driver.register_adapter(adapter) # type:ignore
18 | except ImportError:
19 | print(f"\33[35m{adapter}不存在，请检查拼写错误或是否已安装该适配器？")
20 | sys.exit(1)
21 |
22 |
23 | nonebot.init()
24 |
25 | driver = nonebot.get_driver()
26 |
27 | enable_adapters: list[str] = driver.config.model_dump().get("enable_adapters", [])
28 |
29 | for adapter in enable_adapters:
30 | load_specified_adapter(driver, adapter)
31 |
32 | nonebot.load_plugin("muicebot")
33 |
34 | nonebot.run()
35 |
--------------------------------------------------------------------------------
/migrations.py:
--------------------------------------------------------------------------------
1 | import asyncio
2 | import json
3 | import os
4 | import sys
5 | from dataclasses import dataclass, field
6 | from datetime import datetime
7 | from typing import List
8 |
9 | import aiosqlite
10 |
11 |
12 | @dataclass
13 | class Message:
14 | id: int | None = None
15 | """每条消息的唯一ID"""
16 | time: str = datetime.strftime(datetime.now(), "%Y.%m.%d %H:%M:%S")
17 | """
18 | 字符串形式的时间数据：%Y.%m.%d %H:%M:%S
19 | 若要获取格式化的 datetime 对象，请使用 format_time
20 | """
21 | userid: str = ""
22 | """Nonebot 的用户id"""
23 | message: str = ""
24 | """消息主体"""
25 | respond: str = ""
26 | """模型回复（不包含思维过程）"""
27 | history: int = 1
28 | """消息是否可用于对话历史中，以整数形式映射布尔值"""
29 | images: List[str] = field(default_factory=list)
30 | """多模态中使用的图像，默认为空列表"""
31 |
32 | def __post_init__(self):
33 | if isinstance(self.images, str):
34 | self.images = json.loads(self.images)
35 | elif self.images is None:
36 | self.images = []
37 |
38 |
39 | class Database:
40 | def __init__(self, db_path: str) -> None:
41 | self.DB_PATH = db_path
42 |
43 | def __connect(self) -> aiosqlite.Connection:
44 | return aiosqlite.connect(self.DB_PATH)
45 |
46 | async def __execute(self, query: str, params=(), fetchone=False, fetchall=False) -> list | None:
47 | """
48 | 异步执行SQL查询，支持可选参数。
49 |
50 | :param query: 要执行的SQL查询语句
51 | :param params: 传递给查询的参数
52 | :param fetchone: 是否获取单个结果
53 | :param fetchall: 是否获取所有结果
54 | """
55 | async with self.__connect() as conn:
56 | cursor = await conn.cursor()
57 | await cursor.execute(query, params)
58 | if fetchone:
59 | return await cursor.fetchone() # type: ignore
60 | if fetchall:
61 | return await cursor.fetchall() # type: ignore
62 | await conn.commit()
63 |
64 | return None
65 |
66 | async def __create_database(self) -> None:
67 | await self.__execute(
68 | """CREATE TABLE MSG(
69 | ID INTEGER PRIMARY KEY AUTOINCREMENT,
70 | TIME TEXT NOT NULL,
71 | USERID TEXT NOT NULL,
72 | MESSAGE TEXT NOT NULL,
73 | RESPOND TEXT NOT NULL,
74 | HISTORY INTEGER NOT NULL DEFAULT (1),
75 | IMAGES TEXT NOT NULL DEFAULT "[]");"""
76 | )
77 |
78 | async def add_item(self, message: Message):
79 | """
80 | 将消息保存到数据库
81 | """
82 | params = (message.time, message.userid, message.message, message.respond, json.dumps(message.images))
83 | query = """INSERT INTO MSG (TIME, USERID, MESSAGE, RESPOND, IMAGES)
84 | VALUES (?, ?, ?, ?, ?)"""
85 | await self.__execute(query, params)
86 |
87 |
88 | async def migrate_old_data_to_new_db(old_data_dir: str, new_database_path: str):
89 | db = Database(new_database_path)
90 |
91 | current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
92 |
93 | # 遍历 memory 目录下的 JSON 文件
94 | for filename in os.listdir(old_data_dir):
95 | if not filename.endswith(".json"):
96 | continue
97 |
98 | file_path = os.path.join(old_data_dir, filename)
99 | user_id = filename.replace(".json", "")
100 |
101 | with open(file_path, "r", encoding="utf-8") as file:
102 | for line in file:
103 | try:
104 | data = json.loads(line.strip())
105 | prompt = data.get("prompt", "").strip()
106 | completion = data.get("completion", "").strip()
107 |
108 | # 跳过空数据
109 | if not prompt or not completion:
110 | continue
111 |
112 | message = Message(time=current_time, userid=user_id, message=prompt, respond=completion, images=[])
113 |
114 | await db.add_item(message)
115 |
116 | except json.JSONDecodeError:
117 | print(f"⚠️ JSON 解析失败: {file_path}")
118 |
119 | print("✅ 迁移完成！")
120 |
121 |
122 | if __name__ == "__main__":
123 |
124 | if len(sys.argv) < 3:
125 | print("❌ 使用方式: python migrations.py