mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2024-11-11 09:37:29 +08:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
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())
|