forked from LiteyukiStudio/LiteyukiBot
97 lines
2.6 KiB
Python
97 lines
2.6 KiB
Python
import os
|
||
|
||
from pydantic import Field
|
||
|
||
from .data import Database, LiteModel, Database
|
||
|
||
DATA_PATH = "data/liteyuki"
|
||
|
||
user_db: Database = Database(os.path.join(DATA_PATH, "users.ldb"))
|
||
group_db: Database = Database(os.path.join(DATA_PATH, "groups.ldb"))
|
||
plugin_db: Database = Database(os.path.join(DATA_PATH, "plugins.ldb"))
|
||
common_db: Database = Database(os.path.join(DATA_PATH, "common.ldb"))
|
||
|
||
# 内存数据库，临时用于存储数据
|
||
memory_database = {
|
||
|
||
}
|
||
|
||
|
||
class User(LiteModel):
|
||
TABLE_NAME: str = "user"
|
||
user_id: str = Field(str(), alias="user_id")
|
||
username: str = Field(str(), alias="username")
|
||
profile: dict[str, str] = Field(dict(), alias="profile")
|
||
enabled_plugins: list[str] = Field(list(), alias="enabled_plugins")
|
||
disabled_plugins: list[str] = Field(list(), alias="disabled_plugins")
|
||
|
||
|
||
class Group(LiteModel):
|
||
TABLE_NAME: str = "group_chat"
|
||
# Group是一个关键字，所以这里用GroupChat
|
||
group_id: str = Field(str(), alias="group_id")
|
||
group_name: str = Field(str(), alias="group_name")
|
||
enabled_plugins: list[str] = Field([], alias="enabled_plugins")
|
||
disabled_plugins: list[str] = Field([], alias="disabled_plugins")
|
||
enable: bool = Field(True, alias="enable") # 群聊全局机器人是否启用
|
||
config: dict = Field({}, alias="config")
|
||
|
||
|
||
class InstalledPlugin(LiteModel):
|
||
TABLE_NAME: str = "installed_plugin"
|
||
module_name: str = Field(str(), alias="module_name")
|
||
version: str = Field(str(), alias="version")
|
||
|
||
|
||
class GlobalPlugin(LiteModel):
|
||
TABLE_NAME: str = "global_plugin"
|
||
liteyuki: bool = Field(True, alias="liteyuki") # 是否为LiteYuki插件
|
||
module_name: str = Field(str(), alias="module_name")
|
||
enabled: bool = Field(True, alias="enabled")
|
||
|
||
|
||
class StoredConfig(LiteModel):
|
||
TABLE_NAME: str = "stored_config"
|
||
config: dict = {}
|
||
|
||
|
||
class TempConfig(LiteModel):
|
||
"""储存临时键值对的表"""
|
||
TABLE_NAME: str = "temp_data"
|
||
data: dict = {}
|
||
|
||
|
||
def auto_migrate():
|
||
user_db.auto_migrate(User())
|
||
group_db.auto_migrate(Group())
|
||
plugin_db.auto_migrate(InstalledPlugin(), GlobalPlugin())
|
||
common_db.auto_migrate(GlobalPlugin(), StoredConfig(), TempConfig())
|
||
|
||
|
||
def set_memory_data(key: str, value) -> None:
|
||
"""
|
||
设置内存数据库的数据，类似于redis
|
||
Args:
|
||
key:
|
||
value:
|
||
|
||
Returns:
|
||
|
||
"""
|
||
return memory_database.update({
|
||
key: value
|
||
})
|
||
|
||
|
||
def get_memory_data(key: str, default=None) -> any:
|
||
"""
|
||
获取内存数据库的数据,类似于redis
|
||
Args:
|
||
key:
|
||
default:
|
||
|
||
Returns:
|
||
|
||
"""
|
||
return memory_database.get(key, default)
|