sticker-captcha-bot/sticker/single_utils.py
2022-07-02 19:44:57 +08:00

42 lines
1.1 KiB
Python

import contextlib
from os import sep, remove, mkdir
from os.path import exists
from typing import Optional
from pyrogram import Client
from pyrogram.types import Message
from pyromod.utils.errors import TimeoutConversationError, ListenerCanceled
from sqlitedict import SqliteDict
# init folders
if not exists("data"):
mkdir("data")
sqlite = SqliteDict(f"data{sep}data.sqlite", autocommit=True)
def safe_remove(name: str) -> None:
with contextlib.suppress(FileNotFoundError):
remove(name)
class Client(Client): # noqa
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
class Message(Message): # noqa
async def delay_delete(self, delete_seconds: int = 60) -> Optional[bool]:
return
async def safe_delete(self, revoke: bool = True) -> None:
return