mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
28 lines
681 B
Python
28 lines
681 B
Python
|
from typing import ClassVar
|
||
|
|
|
||
|
|
|
||
|
|
class MessageManager:
|
||
|
|
data: ClassVar[dict[str, list[str]]] = {}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def add(cls, uid: str, msg_id: str):
|
||
|
|
if uid not in cls.data:
|
||
|
|
cls.data[uid] = []
|
||
|
|
cls.data[uid].append(msg_id)
|
||
|
|
cls.remove_check(uid)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def check(cls, uid: str, msg_id: str) -> bool:
|
||
|
|
return msg_id in cls.data.get(uid, [])
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def remove_check(cls, uid: str):
|
||
|
|
if len(cls.data[uid]) > 200:
|
||
|
|
cls.data[uid] = cls.data[uid][100:]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get(cls, uid: str) -> list[str]:
|
||
|
|
if uid in cls.data:
|
||
|
|
return cls.data[uid]
|
||
|
|
return []
|