mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
136 lines
3.6 KiB
Python
136 lines
3.6 KiB
Python
from asyncio import Lock
|
|
from collections.abc import AsyncGenerator
|
|
from contextlib import asynccontextmanager
|
|
from datetime import datetime, timezone
|
|
from enum import Enum, auto
|
|
from typing import TYPE_CHECKING, Literal, TypeVar, overload
|
|
|
|
from nonebot.exception import FinishedException
|
|
from nonebot.log import logger
|
|
from nonebot_plugin_orm import AsyncSession, get_session
|
|
from nonebot_plugin_user import User
|
|
from sqlalchemy import select
|
|
|
|
from ..utils.typing import AllCommandType, BaseCommandType, GameType, TETRIOCommandType
|
|
from .models import Bind, TriggerHistoricalData
|
|
|
|
UTC = timezone.utc
|
|
|
|
if TYPE_CHECKING:
|
|
from ..games.tetrio.api.models import TETRIOHistoricalData
|
|
from ..games.top.api.models import TOPHistoricalData
|
|
from ..games.tos.api.models import TOSHistoricalData
|
|
|
|
|
|
class BindStatus(Enum):
|
|
SUCCESS = auto()
|
|
UPDATE = auto()
|
|
|
|
|
|
async def query_bind_info(
|
|
session: AsyncSession,
|
|
user: User,
|
|
game_platform: GameType,
|
|
) -> Bind | None:
|
|
return (
|
|
await session.scalars(select(Bind).where(Bind.user_id == user.id).where(Bind.game_platform == game_platform))
|
|
).one_or_none()
|
|
|
|
|
|
async def create_or_update_bind(
|
|
session: AsyncSession,
|
|
user: User,
|
|
game_platform: GameType,
|
|
game_account: str,
|
|
) -> BindStatus:
|
|
bind = await query_bind_info(
|
|
session=session,
|
|
user=user,
|
|
game_platform=game_platform,
|
|
)
|
|
if bind is None:
|
|
bind = Bind(
|
|
user_id=user.id,
|
|
game_platform=game_platform,
|
|
game_account=game_account,
|
|
)
|
|
session.add(bind)
|
|
status = BindStatus.SUCCESS
|
|
else:
|
|
bind.game_account = game_account
|
|
status = BindStatus.UPDATE
|
|
await session.commit()
|
|
return status
|
|
|
|
|
|
T = TypeVar('T', 'TETRIOHistoricalData', 'TOPHistoricalData', 'TOSHistoricalData')
|
|
|
|
lock = Lock()
|
|
|
|
|
|
async def anti_duplicate_add(model: T) -> None:
|
|
async with lock, get_session() as session:
|
|
result = (
|
|
await session.scalars(
|
|
select(cls := model.__class__)
|
|
.where(cls.update_time == model.update_time)
|
|
.where(cls.user_unique_identifier == model.user_unique_identifier)
|
|
.where(cls.api_type == model.api_type)
|
|
)
|
|
).all()
|
|
if result:
|
|
for i in result:
|
|
if i.data == model.data:
|
|
logger.debug('Anti duplicate successfully')
|
|
return
|
|
session.add(model)
|
|
await session.commit()
|
|
|
|
|
|
@asynccontextmanager
|
|
@overload
|
|
async def trigger(
|
|
session_persist_id: int,
|
|
game_platform: Literal['IO'],
|
|
command_type: TETRIOCommandType,
|
|
command_args: list[str],
|
|
) -> AsyncGenerator:
|
|
yield
|
|
|
|
|
|
@asynccontextmanager
|
|
@overload
|
|
async def trigger(
|
|
session_persist_id: int,
|
|
game_platform: GameType,
|
|
command_type: BaseCommandType,
|
|
command_args: list[str],
|
|
) -> AsyncGenerator:
|
|
yield
|
|
|
|
|
|
@asynccontextmanager
|
|
async def trigger(
|
|
session_persist_id: int,
|
|
game_platform: GameType,
|
|
command_type: AllCommandType,
|
|
command_args: list[str],
|
|
) -> AsyncGenerator:
|
|
trigger_time = datetime.now(UTC)
|
|
try:
|
|
yield
|
|
except FinishedException:
|
|
async with get_session() as session:
|
|
session.add(
|
|
TriggerHistoricalData(
|
|
trigger_time=trigger_time,
|
|
session_persist_id=session_persist_id,
|
|
game_platform=game_platform,
|
|
command_type=command_type,
|
|
command_args=command_args,
|
|
finish_time=datetime.now(UTC),
|
|
)
|
|
)
|
|
await session.commit()
|
|
raise
|