mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-22 05:19:19 +00:00
77 lines
2.1 KiB
Python
77 lines
2.1 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
|
|
from pyrogram.types import Message
|
|
|
|
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 Client(Client): # noqa
|
|
job: Optional[AsyncIOScheduler] = None
|
|
|
|
async def listen(self, chat_id, filters=None, timeout=None) -> Optional[Message]:
|
|
return
|
|
|
|
async def ask(self, chat_id, text, filters=None, timeout=None, *args, **kwargs) -> Optional[Message]:
|
|
return
|
|
|
|
def cancel_listener(self, chat_id):
|
|
""" Cancel the conversation with the given chat_id. """
|
|
return
|
|
|
|
def cancel_all_listeners(self):
|
|
""" Cancel all conversations. """
|
|
return
|
|
|
|
def conversation(self, chat_id: Union[int, str],
|
|
once_timeout: int = 60, filters=None) -> Optional[Conversation]:
|
|
""" Initialize a conversation with the given chat_id. """
|
|
return
|
|
|
|
|
|
class Message(Message): # noqa
|
|
arguments: str
|
|
parameter: List
|
|
bot: Client
|
|
request: Optional[AsyncClient] = None
|
|
|
|
def obtain_message(self) -> Optional[str]:
|
|
""" Obtains a message from either the reply message or command arguments. """
|
|
return
|
|
|
|
def obtain_user(self) -> Optional[int]:
|
|
""" Obtains a user from either the reply message or command arguments. """
|
|
return
|
|
|
|
async def delay_delete(self, delete_seconds: int = 60) -> Optional[bool]:
|
|
return
|
|
|
|
async def safe_delete(self, revoke: bool = True) -> None:
|
|
return
|