mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2024-11-25 00:25:04 +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)
|