mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
21 lines
757 B
Python
21 lines
757 B
Python
from nonebot_plugin_alconna.uniseg import UniMessage
|
|
from nonebot_plugin_orm import async_scoped_session
|
|
from nonebot_plugin_user import User # type: ignore[import-untyped]
|
|
from sqlalchemy import select
|
|
|
|
from . import alc
|
|
from .models import TETRIOUserConfig
|
|
from .typing import Template
|
|
|
|
|
|
@alc.assign('TETRIO.config')
|
|
async def _(user: User, session: async_scoped_session, template: Template):
|
|
config = (await session.scalars(select(TETRIOUserConfig).where(TETRIOUserConfig.id == user.id))).one_or_none()
|
|
if config is None:
|
|
config = TETRIOUserConfig(id=user.id, query_template=template)
|
|
session.add(config)
|
|
else:
|
|
config.query_template = template
|
|
await session.commit()
|
|
await UniMessage('配置成功').finish()
|