telegrinder 0.3.4__py3-none-any.whl → 0.4.0__py3-none-any.whl

telegrinder/bot/dispatch/view/box.py

@@ -1,129 +1,140 @@
-import dataclasses
-
-import typing_extensions as typing
-
-from telegrinder.bot.dispatch.view import (
-    callback_query,
-    chat_join_request,
-    chat_member,
-    inline_query,
-    message,
-    raw,
-)
-from telegrinder.bot.dispatch.view.abc import ABCEventRawView, ABCView
-from telegrinder.types.enums import UpdateType
-
-CallbackQueryView = typing.TypeVar(
-    "CallbackQueryView", bound=ABCView, default=callback_query.CallbackQueryView
-)
-ChatJoinRequestView = typing.TypeVar(
-    "ChatJoinRequestView", bound=ABCView, default=chat_join_request.ChatJoinRequestView
-)
-ChatMemberView = typing.TypeVar("ChatMemberView", bound=ABCView, default=chat_member.ChatMemberView)
-InlineQueryView = typing.TypeVar("InlineQueryView", bound=ABCView, default=inline_query.InlineQueryView)
-MessageView = typing.TypeVar("MessageView", bound=ABCView, default=message.MessageView)
-RawEventView = typing.TypeVar("RawEventView", bound=ABCEventRawView, default=raw.RawEventView)
-
-
-@dataclasses.dataclass(kw_only=True)
-class ViewBox(
-    typing.Generic[
-        CallbackQueryView,
-        ChatJoinRequestView,
-        ChatMemberView,
-        InlineQueryView,
-        MessageView,
-        RawEventView,
-    ],
-):
-    callback_query_view: dataclasses.InitVar[CallbackQueryView | None] = None
-    chat_join_request_view: dataclasses.InitVar[ChatJoinRequestView | None] = None
-    chat_member_view: dataclasses.InitVar[ChatMemberView | None] = None
-    my_chat_member_view: dataclasses.InitVar[ChatMemberView | None] = None
-    inline_query_view: dataclasses.InitVar[InlineQueryView | None] = None
-    message_view: dataclasses.InitVar[MessageView | None] = None
-    business_message_view: dataclasses.InitVar[MessageView | None] = None
-    channel_post_view: dataclasses.InitVar[MessageView | None] = None
-    edited_message_view: dataclasses.InitVar[MessageView | None] = None
-    edited_business_message_view: dataclasses.InitVar[MessageView | None] = None
-    edited_channel_post_view: dataclasses.InitVar[MessageView | None] = None
-    any_message_view: dataclasses.InitVar[MessageView | None] = None
-    chat_member_updated_view: dataclasses.InitVar[ChatMemberView | None] = None
-    raw_event_view: dataclasses.InitVar[RawEventView | None] = None
-
-    def __post_init__(
-        self,
-        callback_query_view: CallbackQueryView | None = None,
-        chat_join_request_view: ChatJoinRequestView | None = None,
-        chat_member_view: ChatMemberView | None = None,
-        my_chat_member_view: ChatMemberView | None = None,
-        inline_query_view: InlineQueryView | None = None,
-        message_view: MessageView | None = None,
-        business_message_view: MessageView | None = None,
-        channel_post_view: MessageView | None = None,
-        edited_message_view: MessageView | None = None,
-        edited_business_message_view: MessageView | None = None,
-        edited_channel_post_view: MessageView | None = None,
-        any_message_view: MessageView | None = None,
-        chat_member_updated_view: ChatMemberView | None = None,
-        raw_event_view: RawEventView | None = None,
-    ) -> None:
-        self.callback_query = typing.cast(
-            CallbackQueryView,
-            callback_query_view or callback_query.CallbackQueryView(),
-        )
-        self.chat_join_request = typing.cast(
-            ChatJoinRequestView,
-            chat_join_request_view or chat_join_request.ChatJoinRequestView(),
-        )
-        self.chat_member = typing.cast(
-            ChatMemberView,
-            chat_member_view or chat_member.ChatMemberView(update_type=UpdateType.CHAT_MEMBER),
-        )
-        self.my_chat_member = typing.cast(
-            ChatMemberView,
-            my_chat_member_view or chat_member.ChatMemberView(update_type=UpdateType.MY_CHAT_MEMBER),
-        )
-        self.inline_query = typing.cast(
-            InlineQueryView,
-            inline_query_view or inline_query.InlineQueryView(),
-        )
-        self.message = typing.cast(
-            MessageView,
-            message_view or message.MessageView(update_type=UpdateType.MESSAGE),
-        )
-        self.business_message = typing.cast(
-            MessageView,
-            business_message_view or message.MessageView(update_type=UpdateType.BUSINESS_MESSAGE),
-        )
-        self.channel_post = typing.cast(
-            MessageView,
-            channel_post_view or message.MessageView(update_type=UpdateType.CHANNEL_POST),
-        )
-        self.edited_message = typing.cast(
-            MessageView,
-            edited_message_view or message.MessageView(update_type=UpdateType.EDITED_MESSAGE),
-        )
-        self.edited_business_message = typing.cast(
-            MessageView,
-            edited_business_message_view
-            or message.MessageView(update_type=UpdateType.EDITED_BUSINESS_MESSAGE),
-        )
-        self.edited_channel_post = typing.cast(
-            MessageView,
-            edited_channel_post_view or message.MessageView(update_type=UpdateType.EDITED_CHANNEL_POST),
-        )
-        self.any_message = typing.cast(MessageView, any_message_view or message.MessageView())
-        self.chat_member_updated = typing.cast(
-            ChatMemberView,
-            chat_member_updated_view or chat_member.ChatMemberView(),
-        )
-        self.raw_event = typing.cast(RawEventView, raw_event_view or raw.RawEventView())
-
-    def get_views(self) -> dict[str, ABCView]:
-        """Get all views."""
-
-        return {name: view for name, view in self.__dict__.items() if isinstance(view, ABCView)}
-
-
-__all__ = ("ViewBox",)
+import dataclasses
+
+import typing_extensions as typing
+
+from telegrinder.bot.dispatch.view import (
+    callback_query,
+    chat_join_request,
+    chat_member,
+    inline_query,
+    message,
+    pre_checkout_query,
+    raw,
+)
+from telegrinder.bot.dispatch.view.abc import ABCEventRawView, ABCView
+from telegrinder.types.enums import UpdateType
+
+CallbackQueryView = typing.TypeVar(
+    "CallbackQueryView",
+    bound=ABCView,
+    default=callback_query.CallbackQueryView,
+)
+PreCheckoutQueryView = typing.TypeVar(
+    "PreCheckoutQueryView",
+    bound=ABCView,
+    default=pre_checkout_query.PreCheckoutQueryView,
+)
+ChatJoinRequestView = typing.TypeVar(
+    "ChatJoinRequestView",
+    bound=ABCView,
+    default=chat_join_request.ChatJoinRequestView,
+)
+ChatMemberView = typing.TypeVar("ChatMemberView", bound=ABCView, default=chat_member.ChatMemberView)
+InlineQueryView = typing.TypeVar("InlineQueryView", bound=ABCView, default=inline_query.InlineQueryView)
+MessageView = typing.TypeVar("MessageView", bound=ABCView, default=message.MessageView)
+RawEventView = typing.TypeVar("RawEventView", bound=ABCEventRawView, default=raw.RawEventView)
+
+
+@dataclasses.dataclass(kw_only=True)
+class ViewBox(
+    typing.Generic[
+        CallbackQueryView,
+        ChatJoinRequestView,
+        ChatMemberView,
+        InlineQueryView,
+        MessageView,
+        PreCheckoutQueryView,
+        RawEventView,
+    ],
+):
+    callback_query_view: dataclasses.InitVar[CallbackQueryView | None] = None
+    chat_join_request_view: dataclasses.InitVar[ChatJoinRequestView | None] = None
+    chat_member_view: dataclasses.InitVar[ChatMemberView | None] = None
+    my_chat_member_view: dataclasses.InitVar[ChatMemberView | None] = None
+    inline_query_view: dataclasses.InitVar[InlineQueryView | None] = None
+    message_view: dataclasses.InitVar[MessageView | None] = None
+    business_message_view: dataclasses.InitVar[MessageView | None] = None
+    channel_post_view: dataclasses.InitVar[MessageView | None] = None
+    pre_checkout_query_view: dataclasses.InitVar[PreCheckoutQueryView | None] = None
+    edited_message_view: dataclasses.InitVar[MessageView | None] = None
+    edited_business_message_view: dataclasses.InitVar[MessageView | None] = None
+    edited_channel_post_view: dataclasses.InitVar[MessageView | None] = None
+    any_message_view: dataclasses.InitVar[MessageView | None] = None
+    chat_member_updated_view: dataclasses.InitVar[ChatMemberView | None] = None
+    raw_event_view: dataclasses.InitVar[RawEventView | None] = None
+
+    def __post_init__(
+        self,
+        callback_query_view: CallbackQueryView | None = None,
+        chat_join_request_view: ChatJoinRequestView | None = None,
+        chat_member_view: ChatMemberView | None = None,
+        my_chat_member_view: ChatMemberView | None = None,
+        inline_query_view: InlineQueryView | None = None,
+        message_view: MessageView | None = None,
+        business_message_view: MessageView | None = None,
+        channel_post_view: MessageView | None = None,
+        edited_message_view: MessageView | None = None,
+        edited_business_message_view: MessageView | None = None,
+        edited_channel_post_view: MessageView | None = None,
+        any_message_view: MessageView | None = None,
+        chat_member_updated_view: ChatMemberView | None = None,
+        pre_checkout_query_view: PreCheckoutQueryView | None = None,
+        raw_event_view: RawEventView | None = None,
+    ) -> None:
+        self.callback_query = typing.cast(
+            CallbackQueryView,
+            callback_query_view or callback_query.CallbackQueryView(),
+        )
+        self.chat_join_request = typing.cast(
+            ChatJoinRequestView,
+            chat_join_request_view or chat_join_request.ChatJoinRequestView(),
+        )
+        self.chat_member = typing.cast(
+            ChatMemberView,
+            chat_member_view or chat_member.ChatMemberView(update_type=UpdateType.CHAT_MEMBER),
+        )
+        self.my_chat_member = typing.cast(
+            ChatMemberView,
+            my_chat_member_view or chat_member.ChatMemberView(update_type=UpdateType.MY_CHAT_MEMBER),
+        )
+        self.inline_query = typing.cast(
+            InlineQueryView,
+            inline_query_view or inline_query.InlineQueryView(),
+        )
+        self.message = typing.cast(
+            MessageView,
+            message_view or message.MessageView(update_type=UpdateType.MESSAGE),
+        )
+        self.business_message = typing.cast(
+            MessageView,
+            business_message_view or message.MessageView(update_type=UpdateType.BUSINESS_MESSAGE),
+        )
+        self.channel_post = typing.cast(
+            MessageView,
+            channel_post_view or message.MessageView(update_type=UpdateType.CHANNEL_POST),
+        )
+        self.edited_message = typing.cast(
+            MessageView,
+            edited_message_view or message.MessageView(update_type=UpdateType.EDITED_MESSAGE),
+        )
+        self.edited_business_message = typing.cast(
+            MessageView,
+            edited_business_message_view or message.MessageView(update_type=UpdateType.EDITED_BUSINESS_MESSAGE),
+        )
+        self.edited_channel_post = typing.cast(
+            MessageView,
+            edited_channel_post_view or message.MessageView(update_type=UpdateType.EDITED_CHANNEL_POST),
+        )
+        self.pre_checkout_query = typing.cast(
+            PreCheckoutQueryView,
+            pre_checkout_query_view or pre_checkout_query.PreCheckoutQueryView(),
+        )
+        self.any_message = typing.cast(MessageView, any_message_view or message.MessageView())
+        self.chat_member_updated = typing.cast(
+            ChatMemberView,
+            chat_member_updated_view or chat_member.ChatMemberView(),
+        )
+        self.raw_event = typing.cast(RawEventView, raw_event_view or raw.RawEventView())
+
+
+__all__ = ("ViewBox",)

telegrinder/bot/dispatch/view/callback_query.py

@@ -1,17 +1,16 @@
-from telegrinder.bot.cute_types.callback_query import CallbackQueryCute
-from telegrinder.bot.dispatch.return_manager.callback_query import CallbackQueryReturnManager
-from telegrinder.bot.dispatch.view.base import BaseStateView
-
-
-class CallbackQueryView(BaseStateView[CallbackQueryCute]):
-    def __init__(self) -> None:
-        self.auto_rules = []
-        self.handlers = []
-        self.middlewares = []
-        self.return_manager = CallbackQueryReturnManager()
-
-    def get_state_key(self, event: CallbackQueryCute) -> int | None:
-        return event.message_id.unwrap_or_none()
-
-
-__all__ = ("CallbackQueryView",)
+from telegrinder.bot.cute_types.callback_query import CallbackQueryCute
+from telegrinder.bot.dispatch.return_manager.callback_query import CallbackQueryReturnManager
+from telegrinder.bot.dispatch.view.base import BaseStateView
+
+
+class CallbackQueryView(BaseStateView[CallbackQueryCute]):
+    def __init__(self) -> None:
+        super().__init__()
+        self.return_manager = CallbackQueryReturnManager()
+
+    @classmethod
+    def get_state_key(cls, event: CallbackQueryCute) -> int | None:
+        return event.message_id.unwrap_or_none()
+
+
+__all__ = ("CallbackQueryView",)

telegrinder/bot/dispatch/view/chat_join_request.py

@@ -1,16 +1,11 @@
-from telegrinder.bot.cute_types.chat_join_request import ChatJoinRequestCute
-from telegrinder.bot.dispatch.view.base import BaseStateView
-
-
-class ChatJoinRequestView(BaseStateView[ChatJoinRequestCute]):
-    def __init__(self) -> None:
-        self.auto_rules = []
-        self.handlers = []
-        self.middlewares = []
-        self.return_manager = None
-
-    def get_state_key(self, event: ChatJoinRequestCute) -> int | None:
-        return event.chat_id
-
-
-__all__ = ("ChatJoinRequestView",)
+from telegrinder.bot.cute_types.chat_join_request import ChatJoinRequestCute
+from telegrinder.bot.dispatch.view.base import BaseStateView
+
+
+class ChatJoinRequestView(BaseStateView[ChatJoinRequestCute]):
+    @classmethod
+    def get_state_key(cls, event: ChatJoinRequestCute) -> int | None:
+        return event.chat_id
+
+
+__all__ = ("ChatJoinRequestView",)

telegrinder/bot/dispatch/view/chat_member.py

@@ -1,39 +1,37 @@
-import typing
-
-from telegrinder.bot.cute_types.chat_member_updated import ChatMemberUpdatedCute
-from telegrinder.bot.dispatch.view.base import BaseStateView
-from telegrinder.types.enums import UpdateType
-from telegrinder.types.objects import Update
-
-ChatMemberUpdateType: typing.TypeAlias = typing.Literal[
-    UpdateType.CHAT_MEMBER,
-    UpdateType.MY_CHAT_MEMBER,
-]
-
-
-class ChatMemberView(BaseStateView[ChatMemberUpdatedCute]):
-    def __init__(self, *, update_type: ChatMemberUpdateType | None = None) -> None:
-        self.auto_rules = []
-        self.handlers = []
-        self.middlewares = []
-        self.return_manager = None
-        self.update_type = update_type
-
-    def __repr__(self) -> str:
-        return "<{}: {!r}>".format(
-            self.__class__.__name__,
-            "chat_member_updated" if self.update_type is None else self.update_type.value,
-        )
-
-    def get_state_key(self, event: ChatMemberUpdatedCute) -> int | None:
-        return event.chat_id
-
-    async def check(self, event: Update) -> bool:
-        return not (
-            self.update_type is not None
-            and self.update_type != event.update_type
-            or not await super().check(event)
-        )
-
-
-__all__ = ("ChatMemberView",)
+import typing
+
+from telegrinder.bot.cute_types.chat_member_updated import ChatMemberUpdatedCute
+from telegrinder.bot.dispatch.view.base import BaseStateView
+from telegrinder.types.enums import UpdateType
+from telegrinder.types.objects import Update
+
+ChatMemberUpdateType: typing.TypeAlias = typing.Literal[
+    UpdateType.CHAT_MEMBER,
+    UpdateType.MY_CHAT_MEMBER,
+]
+
+
+class ChatMemberView(BaseStateView[ChatMemberUpdatedCute]):
+    def __init__(self, *, update_type: ChatMemberUpdateType | None = None) -> None:
+        super().__init__()
+        self.update_type = update_type
+
+    def __repr__(self) -> str:
+        return "<{}: {!r}>".format(
+            self.__class__.__name__,
+            "chat_member_updated" if self.update_type is None else self.update_type.value,
+        )
+
+    @classmethod
+    def get_state_key(cls, event: ChatMemberUpdatedCute) -> int | None:
+        return event.chat_id
+
+    async def check(self, event: Update) -> bool:
+        return not (
+            self.update_type is not None
+            and self.update_type != event.update_type
+            or not await super().check(event)
+        )
+
+
+__all__ = ("ChatMemberView",)

telegrinder/bot/dispatch/view/inline_query.py

@@ -1,17 +1,16 @@
-from telegrinder.bot.cute_types.inline_query import InlineQueryCute
-from telegrinder.bot.dispatch.return_manager import InlineQueryReturnManager
-from telegrinder.bot.dispatch.view.base import BaseStateView
-
-
-class InlineQueryView(BaseStateView[InlineQueryCute]):
-    def __init__(self) -> None:
-        self.auto_rules = []
-        self.handlers = []
-        self.middlewares = []
-        self.return_manager = InlineQueryReturnManager()
-
-    def get_state_key(self, event: InlineQueryCute) -> int | None:
-        return event.from_user.id
-
-
-__all__ = ("InlineQueryView",)
+from telegrinder.bot.cute_types.inline_query import InlineQueryCute
+from telegrinder.bot.dispatch.return_manager import InlineQueryReturnManager
+from telegrinder.bot.dispatch.view.base import BaseStateView
+
+
+class InlineQueryView(BaseStateView[InlineQueryCute]):
+    def __init__(self) -> None:
+        super().__init__()
+        self.return_manager = InlineQueryReturnManager()
+
+    @classmethod
+    def get_state_key(cls, event: InlineQueryCute) -> int | None:
+        return event.from_user.id
+
+
+__all__ = ("InlineQueryView",)

telegrinder/bot/dispatch/view/message.py

@@ -1,44 +1,43 @@
-import typing
-
-from telegrinder.bot.cute_types.message import MessageCute
-from telegrinder.bot.dispatch.return_manager.message import MessageReturnManager
-from telegrinder.bot.dispatch.view.base import BaseStateView
-from telegrinder.types.enums import UpdateType
-from telegrinder.types.objects import Update
-
-MessageUpdateType: typing.TypeAlias = typing.Literal[
-    UpdateType.MESSAGE,
-    UpdateType.BUSINESS_MESSAGE,
-    UpdateType.CHANNEL_POST,
-    UpdateType.EDITED_BUSINESS_MESSAGE,
-    UpdateType.EDITED_CHANNEL_POST,
-    UpdateType.EDITED_MESSAGE,
-]
-
-
-class MessageView(BaseStateView[MessageCute]):
-    def __init__(self, *, update_type: MessageUpdateType | None = None) -> None:
-        self.auto_rules = []
-        self.handlers = []
-        self.update_type = update_type
-        self.middlewares = []
-        self.return_manager = MessageReturnManager()
-
-    def __repr__(self) -> str:
-        return "<{}: {!r}>".format(
-            self.__class__.__name__,
-            "any message update" if self.update_type is None else self.update_type.value,
-        )
-
-    def get_state_key(self, event: MessageCute) -> int | None:
-        return event.chat_id
-
-    async def check(self, event: Update) -> bool:
-        return not (
-            self.update_type is not None
-            and self.update_type != event.update_type
-            or not await super().check(event)
-        )
-
-
-__all__ = ("MessageView",)
+import typing
+
+from telegrinder.bot.cute_types.message import MessageCute
+from telegrinder.bot.dispatch.return_manager.message import MessageReturnManager
+from telegrinder.bot.dispatch.view.base import BaseStateView
+from telegrinder.types.enums import UpdateType
+from telegrinder.types.objects import Update
+
+MessageUpdateType: typing.TypeAlias = typing.Literal[
+    UpdateType.MESSAGE,
+    UpdateType.BUSINESS_MESSAGE,
+    UpdateType.CHANNEL_POST,
+    UpdateType.EDITED_BUSINESS_MESSAGE,
+    UpdateType.EDITED_CHANNEL_POST,
+    UpdateType.EDITED_MESSAGE,
+]
+
+
+class MessageView(BaseStateView[MessageCute]):
+    def __init__(self, *, update_type: MessageUpdateType | None = None) -> None:
+        super().__init__()
+        self.update_type = update_type
+        self.return_manager = MessageReturnManager()
+
+    def __repr__(self) -> str:
+        return "<{}: {!r}>".format(
+            self.__class__.__name__,
+            "any message update" if self.update_type is None else self.update_type.value,
+        )
+
+    @classmethod
+    def get_state_key(cls, event: MessageCute) -> int | None:
+        return event.chat_id
+
+    async def check(self, event: Update) -> bool:
+        return not (
+            self.update_type is not None
+            and self.update_type != event.update_type
+            or not await super().check(event)
+        )
+
+
+__all__ = ("MessageView",)

telegrinder/bot/dispatch/view/pre_checkout_query.py

@@ -0,0 +1,16 @@
+from telegrinder.bot.cute_types.pre_checkout_query import PreCheckoutQueryCute
+from telegrinder.bot.dispatch.return_manager.pre_checkout_query import PreCheckoutQueryManager
+from telegrinder.bot.dispatch.view.base import BaseStateView
+
+
+class PreCheckoutQueryView(BaseStateView[PreCheckoutQueryCute]):
+    def __init__(self) -> None:
+        super().__init__()
+        self.return_manager = PreCheckoutQueryManager()
+
+    @classmethod
+    def get_state_key(cls, event: PreCheckoutQueryCute) -> int | None:
+        return event.from_user.id
+
+
+__all__ = ("PreCheckoutQueryView",)