diff --git a/maxapi/types/message.py b/maxapi/types/message.py index 9091e7d..90fd5b9 100644 --- a/maxapi/types/message.py +++ b/maxapi/types/message.py @@ -68,8 +68,8 @@ class Recipient(BaseModel): chat_type (ChatType): Тип получателя (диалог или чат). """ - user_id: Optional[int] = None - chat_id: Optional[int] = None + user_id: int + chat_id: int chat_type: ChatType @@ -195,7 +195,9 @@ class Message(BaseModel): Any: Результат выполнения метода send_message бота. """ - assert self.bot is not None + if self.bot is None: + raise RuntimeError('Bot не инициализирован') + return await self.bot.send_message( chat_id=self.recipient.chat_id, user_id=self.recipient.user_id, @@ -227,7 +229,9 @@ class Message(BaseModel): Any: Результат выполнения метода send_message бота. """ - assert self.bot is not None + if self.bot is None: + raise RuntimeError('Bot не инициализирован') + return await self.bot.send_message( chat_id=self.recipient.chat_id, user_id=self.recipient.user_id, @@ -264,7 +268,9 @@ class Message(BaseModel): Any: Результат выполнения метода send_message бота. """ - assert self.bot is not None + if self.bot is None: + raise RuntimeError('Bot не инициализирован') + return await self.bot.send_message( chat_id=chat_id, user_id=user_id, @@ -300,7 +306,9 @@ class Message(BaseModel): Any: Результат выполнения метода edit_message бота. """ - assert self.bot is not None + if self.bot is None: + raise RuntimeError('Bot не инициализирован') + return await self.bot.edit_message( message_id=self.body.mid, text=text, @@ -335,7 +343,9 @@ class Message(BaseModel): Any: Результат выполнения метода pin_message бота. """ - assert self.bot is not None + if self.bot is None: + raise RuntimeError('Bot не инициализирован') + return await self.bot.pin_message( chat_id=self.recipient.chat_id, message_id=self.body.mid, diff --git a/maxapi/types/updates/bot_started.py b/maxapi/types/updates/bot_started.py index fde7b37..c1b5c67 100644 --- a/maxapi/types/updates/bot_started.py +++ b/maxapi/types/updates/bot_started.py @@ -1,4 +1,4 @@ -from typing import TYPE_CHECKING, Any, Optional +from typing import TYPE_CHECKING, Optional from .update import Update @@ -14,13 +14,13 @@ class BotStarted(Update): Обновление, сигнализирующее о первом старте бота. Attributes: - chat_id (Optional[int]): Идентификатор чата. + chat_id (int): Идентификатор чата. user (User): Пользователь (бот). user_locale (Optional[str]): Локаль пользователя. payload (Optional[str]): Дополнительные данные. """ - chat_id: Optional[int] = None + chat_id: int user: User user_locale: Optional[str] = None payload: Optional[str] = None diff --git a/maxapi/types/updates/chat_title_changed.py b/maxapi/types/updates/chat_title_changed.py index 341c256..021194e 100644 --- a/maxapi/types/updates/chat_title_changed.py +++ b/maxapi/types/updates/chat_title_changed.py @@ -1,4 +1,4 @@ -from typing import TYPE_CHECKING, Any, Optional +from typing import TYPE_CHECKING, Optional from .update import Update @@ -16,12 +16,12 @@ class ChatTitleChanged(Update): Attributes: chat_id (Optional[int]): Идентификатор чата. user (User): Пользователь, совершивший изменение. - title (Optional[str]): Новое название чата. + title (str): Новое название чата. """ - chat_id: Optional[int] = None + chat_id: int user: User - title: Optional[str] = None + title: str if TYPE_CHECKING: bot: Optional[Bot] diff --git a/maxapi/types/updates/message_removed.py b/maxapi/types/updates/message_removed.py index bc7e4ab..5c1ae09 100644 --- a/maxapi/types/updates/message_removed.py +++ b/maxapi/types/updates/message_removed.py @@ -1,5 +1,3 @@ -from typing import Optional - from .update import Update @@ -9,14 +7,14 @@ class MessageRemoved(Update): Класс для обработки события удаления сообщения в чате. Attributes: - message_id (Optional[str]): Идентификатор удаленного сообщения. Может быть None. - chat_id (Optional[int]): Идентификатор чата. Может быть None. - user_id (Optional[int]): Идентификатор пользователя. Может быть None. + message_id (str): Идентификатор удаленного сообщения. Может быть None. + chat_id (int): Идентификатор чата. Может быть None. + user_id (int): Идентификатор пользователя. Может быть None. """ - message_id: Optional[str] = None - chat_id: Optional[int] = None - user_id: Optional[int] = None + message_id: str + chat_id: int + user_id: int def get_ids(self):