mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
|
from typing import Union, Tuple, TYPE_CHECKING, List, Any
|
||
|
|
|
||
|
|
from telegram import ChatMember
|
||
|
|
|
||
|
|
try:
|
||
|
|
import ujson as jsonlib
|
||
|
|
except ImportError:
|
||
|
|
import json as jsonlib
|
||
|
|
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from redis import Redis
|
||
|
|
|
||
|
|
from telegram.ext import ContextTypes
|
||
|
|
|
||
|
|
|
||
|
|
class ChatAdministrators:
|
||
|
|
QNAME = "plugin:group_captcha:chat_administrators"
|
||
|
|
TTL = 1 * 60 * 60
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def get_chat_administrators(
|
||
|
|
cache: "Redis",
|
||
|
|
context: "ContextTypes.DEFAULT_TYPE",
|
||
|
|
chat_id: Union[str, int],
|
||
|
|
) -> Union[Tuple[ChatMember, ...], Any]:
|
||
|
|
qname = f"{ChatAdministrators.QNAME}:{chat_id}"
|
||
|
|
result: "List[bytes]" = await cache.lrange(qname, 0, -1)
|
||
|
|
if len(result) > 0:
|
||
|
|
return ChatMember.de_list([jsonlib.loads(str(_data, encoding="utf-8")) for _data in result], context.bot)
|
||
|
|
chat_administrators = await context.bot.get_chat_administrators(chat_id)
|
||
|
|
async with cache.pipeline(transaction=True) as pipe:
|
||
|
|
for chat_administrator in chat_administrators:
|
||
|
|
await pipe.lpush(qname, chat_administrator.to_json())
|
||
|
|
await pipe.expire(qname, ChatAdministrators.TTL)
|
||
|
|
await pipe.execute()
|
||
|
|
return chat_administrators
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def is_admin(chat_administrators: Tuple[ChatMember], user_id: int) -> bool:
|
||
|
|
return any(admin.user.id == user_id for admin in chat_administrators)
|