mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-24 07:10:37 +00:00
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
|
from typing import Union, Optional, TYPE_CHECKING
|
||
|
|
|
||
|
|
from telegram import Chat
|
||
|
|
|
||
|
|
from gram_core.dependence.redisdb import RedisDB
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from . import PluginFuncMethods
|
||
|
|
|
||
|
|
try:
|
||
|
|
import ujson as jsonlib
|
||
|
|
except ImportError:
|
||
|
|
import json as jsonlib
|
||
|
|
|
||
|
|
|
||
|
|
class GetChat:
|
||
|
|
async def get_chat(
|
||
|
|
self: "PluginFuncMethods",
|
||
|
|
chat_id: Union[str, int],
|
||
|
|
redis_db: Optional[RedisDB] = None,
|
||
|
|
expire: int = 86400,
|
||
|
|
) -> Chat:
|
||
|
|
application = self.application
|
||
|
|
redis_db: RedisDB = redis_db or self.application.managers.dependency_map.get(RedisDB, None)
|
||
|
|
|
||
|
|
if not redis_db:
|
||
|
|
return await application.bot.get_chat(chat_id)
|
||
|
|
|
||
|
|
qname = f"bot:chat:{chat_id}"
|
||
|
|
|
||
|
|
data = await redis_db.client.get(qname)
|
||
|
|
if data:
|
||
|
|
json_data = jsonlib.loads(data)
|
||
|
|
return Chat.de_json(json_data, application.telegram.bot)
|
||
|
|
|
||
|
|
chat_info = await application.telegram.bot.get_chat(chat_id)
|
||
|
|
await redis_db.client.set(qname, chat_info.to_json(), ex=expire)
|
||
|
|
return chat_info
|