mirror of https://github.com/coddrago/Heroku
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
import logging
|
|
from typing import Optional, Union
|
|
|
|
from aiogram.types import Message as AiogramMessage
|
|
|
|
from .types import InlineUnit
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class BotPM(InlineUnit):
|
|
def set_fsm_state(self, user: Union[str, int], state: Union[str, bool]) -> bool:
|
|
if not isinstance(user, (str, int)):
|
|
logger.error("Invalid type for `user` in `set_fsm_state`")
|
|
return False
|
|
|
|
if not isinstance(state, (str, bool)):
|
|
logger.error("Invalid type for `state` in `set_fsm_state`")
|
|
return False
|
|
|
|
if state:
|
|
self.fsm[str(user)] = state
|
|
elif str(user) in self.fsm:
|
|
del self.fsm[str(user)]
|
|
|
|
return True
|
|
|
|
ss = set_fsm_state
|
|
|
|
def get_fsm_state(self, user: Union[str, int]) -> Union[bool, str]:
|
|
if not isinstance(user, (str, int)):
|
|
logger.error("Invalid type for `user` in `get_fsm_state`")
|
|
return False
|
|
|
|
return self.fsm.get(str(user), False)
|
|
|
|
gs = get_fsm_state
|
|
|
|
async def _bot_message_answer( # skipcq: PYL-E0213
|
|
mod,
|
|
text: str = None,
|
|
message: AiogramMessage = None,
|
|
parse_mode: Optional[str] = "HTML",
|
|
disable_web_page_preview: Optional[bool] = True,
|
|
**kwargs,
|
|
) -> bool:
|
|
try:
|
|
await mod.bot.send_message(
|
|
message.chat.id,
|
|
text,
|
|
parse_mode=parse_mode,
|
|
disable_web_page_preview=disable_web_page_preview,
|
|
**kwargs,
|
|
)
|
|
except Exception:
|
|
return False
|
|
|
|
return True
|