LiteyukiBot-TriM/liteyuki/utils/data_manager.py

import os

from pydantic import Field

from liteyuki.utils.data import LiteModel, Database as DB

DATA_PATH = "data/liteyuki"

user_db = DB(os.path.join(DATA_PATH, "users.ldb"))
group_db = DB(os.path.join(DATA_PATH, "groups.ldb"))
plugin_db = DB(os.path.join(DATA_PATH, "plugins.ldb"))
common_db = DB(os.path.join(DATA_PATH, "common.ldb"))


class User(LiteModel):
    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 GroupChat(LiteModel):
    # 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")


class InstalledPlugin(LiteModel):
    module_name: str = Field(str(), alias="module_name")
    version: str = Field(str(), alias="version")


class GlobalPlugin(LiteModel):
    module_name: str = Field(str(), alias="module_name")
    enabled: bool = Field(True, alias="enabled")


def auto_migrate():
    user_db.auto_migrate(User)
    group_db.auto_migrate(GroupChat)
    plugin_db.auto_migrate(InstalledPlugin)
    common_db.auto_migrate(GlobalPlugin)