mirror of
https://github.com/Xtao-Labs/sticker-captcha-bot.git
synced 2024-11-25 09:37:34 +00:00
29 lines
785 B
Python
29 lines
785 B
Python
|
import contextlib
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
from cashews import cache
|
||
|
|
|
||
|
|
from sticker import bot
|
||
|
|
|
||
|
|
|
||
|
|
class ServiceMessage:
|
||
|
|
@staticmethod
|
||
|
|
async def set_cache(uid: int, cid: int, mid: int):
|
||
|
|
old = await ServiceMessage.get_cache(uid, cid)
|
||
|
|
old.append(mid)
|
||
|
|
await cache.set(f"service_message:{uid}:{cid}", old, expire=600)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def get_cache(uid: int, cid: int) -> List[int]:
|
||
|
|
data = await cache.get(f"service_message:{uid}:{cid}")
|
||
|
|
if data:
|
||
|
|
return data
|
||
|
|
return []
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def try_delete(uid: int, cid: int):
|
||
|
|
mid = await ServiceMessage.get_cache(uid, cid)
|
||
|
|
if mid:
|
||
|
|
with contextlib.suppress(Exception):
|
||
|
|
await bot.delete_messages(cid, list(mid))
|