mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-16 14:21:11 +00:00
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
import contextlib
|
|
from os import sep, remove, mkdir
|
|
from os.path import exists
|
|
from typing import List, Optional, Union
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from httpx import AsyncClient
|
|
|
|
from pyrogram import Client as OldClient
|
|
from pyrogram.types import Chat as OldChat, Message as OldMessage, Dialog
|
|
|
|
from pyromod.utils.conversation import Conversation
|
|
from pyromod.utils.errors import AlreadyInConversationError, TimeoutConversationError, ListenerCanceled
|
|
|
|
from sqlitedict import SqliteDict
|
|
|
|
# init folders
|
|
if not exists("data"):
|
|
mkdir("data")
|
|
sqlite = SqliteDict(f"data{sep}data.sqlite", autocommit=True)
|
|
|
|
|
|
def get_sudo_list():
|
|
return sqlite.get("sudo_list", [])
|
|
|
|
|
|
def _status_sudo():
|
|
return sqlite.get("sudo_enable", False)
|
|
|
|
|
|
def safe_remove(name: str) -> None:
|
|
with contextlib.suppress(FileNotFoundError):
|
|
remove(name)
|
|
|
|
|
|
class Message(OldMessage):
|
|
arguments: str
|
|
parameter: List
|
|
forum_topic: Optional[bool] = None
|
|
chat: "Chat"
|
|
|
|
def obtain_message(self) -> Optional[str]:
|
|
""" Obtains a message from either the reply message or command arguments. """
|
|
|
|
def obtain_user(self) -> Optional[int]:
|
|
""" Obtains a user from either the reply message or command arguments. """
|
|
|
|
async def delay_delete(self, delete_seconds: int = 60) -> Optional[bool]:
|
|
""" Deletes the message after a specified amount of seconds. """
|
|
|
|
async def safe_delete(self, revoke: bool = True) -> None:
|
|
""" Safely deletes the message. """
|
|
|
|
|
|
class Client(OldClient):
|
|
job: Optional[AsyncIOScheduler] = None
|
|
|
|
async def listen(self, chat_id, filters=None, timeout=None) -> Optional[Message]:
|
|
""" Listen for a message in a conversation. """
|
|
|
|
async def ask(self, chat_id, text, filters=None, timeout=None, *args, **kwargs) -> Optional[Message]:
|
|
""" Ask a message in a conversation. """
|
|
|
|
def cancel_listener(self, chat_id):
|
|
""" Cancel the conversation with the given chat_id. """
|
|
|
|
def cancel_all_listeners(self):
|
|
""" Cancel all conversations. """
|
|
|
|
def conversation(self, chat_id: Union[int, str],
|
|
once_timeout: int = 60, filters=None) -> Optional[Conversation]:
|
|
""" Initialize a conversation with the given chat_id. """
|
|
|
|
async def get_dialogs_list(self) -> List[Dialog]:
|
|
""" Get a list of all dialogs. """
|
|
|
|
|
|
class Chat(OldChat):
|
|
is_forum: Optional[bool] = None
|
|
|
|
async def listen(self, chat_id, filters=None, timeout=None) -> Optional[Message]:
|
|
""" Listen for a message in a conversation. """
|
|
|
|
async def ask(self, chat_id, text, filters=None, timeout=None, *args, **kwargs) -> Optional[Message]:
|
|
""" Ask a message in a conversation. """
|
|
|
|
def cancel_listener(self, chat_id):
|
|
""" Cancel the conversation with the given chat_id. """
|