telegrinder 0.4.2__py3-none-any.whl → 0.5.0__py3-none-any.whl

telegrinder/bot/polling/polling.py

@@ -1,139 +1,260 @@
+from __future__ import annotations
+
 import asyncio
 import sys
 import typing
 from http import HTTPStatus
 
 import msgspec
-from fntypes.result import Error, Ok
+from fntypes.misc import is_ok
 
-from telegrinder.api.api import API, HTTPClient
+from telegrinder.api.api import API
 from telegrinder.api.error import APIServerError, InvalidTokenError
 from telegrinder.bot.polling.abc import ABCPolling
 from telegrinder.modules import logger
 from telegrinder.msgspec_utils import decoder
+from telegrinder.tools.aio import maybe_awaitable
 from telegrinder.types.objects import Update, UpdateType
 
+DEFAULT_OFFSET: typing.Final[int] = 0
+DEFAULT_RECONNECT_AFTER: typing.Final[float] = 5.0
+DEFAULT_MAX_RECONNECTS: typing.Final[int] = 15
+
+
+def _compute_number(
+    default: int | float,
+    input_value: int | float,
+    conditional_value: int | float,
+    /,
+) -> int | float:
+    return max(default, input_value) * (input_value <= conditional_value) + input_value * (
+        input_value >= conditional_value
+    )
+
+
+class PollingErrorHandler:
+    _handlers: dict[type[BaseException], typing.Callable[[BaseException], typing.Any]]
+
+    __slots__ = ("_polling", "_handlers")
+
+    def __init__(self, polling: Polling, /) -> None:
+        self._polling = polling
+        self._handlers = {  # type: ignore
+            InvalidTokenError: self._handle_invalid_token_error,
+            asyncio.CancelledError: self._handle_cancelled_error,
+            APIServerError: self._handle_api_server_error,
+            **{e: self._handle_connection_timeout_error for e in polling.api.http.CONNECTION_TIMEOUT_ERRORS},
+            **{e: self._handle_client_connection_error for e in polling.api.http.CLIENT_CONNECTION_ERRORS},
+        }
+
+    async def handle(self, error: BaseException) -> bool:
+        error_class = type(error)
+
+        if error_class is SystemExit:
+            self._handle_system_exit(error)  # type: ignore
+
+        if error_class not in self._handlers:
+            return False
+
+        try:
+            await maybe_awaitable(self._handlers[error_class](error))
+            return True
+        except SystemExit as sys_exit_err:
+            self._handle_system_exit(sys_exit_err)
+
+    def _handle_system_exit(self, error: SystemExit) -> typing.NoReturn:
+        logger.error(f"Forced exit from the program with code {error.code}.")
+        raise error from None
+
+    def _handle_invalid_token_error(
+        self,
+        error: InvalidTokenError,
+    ) -> typing.NoReturn:
+        logger.error(error)
+        self._polling.stop()
+        sys.exit(3)
+
+    def _handle_cancelled_error(self, _: asyncio.CancelledError) -> None:
+        logger.info("Caught cancel, stopping polling...")
+        self._polling.stop()
+
+    async def _handle_connection_timeout_error(self, _: BaseException) -> None:
+        if self._polling.reconnects_counter > self._polling.max_reconnects:
+            logger.error(
+                "Failed to reconnect to Telegram API server after {} attempts, stopping polling...",
+                self._polling.max_reconnects,
+            )
+            self._polling.stop()
+            sys.exit(6)
+
+        logger.warning(
+            "Server disconnected, waiting {} seconds to reconnect...",
+            self._polling.reconnect_after,
+        )
+        await asyncio.sleep(self._polling.reconnect_after)
+
+    async def _handle_client_connection_error(self, _: BaseException) -> None:
+        logger.error(
+            "Client connection failed, attempt to reconnect after {} seconds...",
+            self._polling.reconnect_after,
+        )
+        await asyncio.sleep(self._polling.reconnect_after)
+
+    async def _handle_api_server_error(
+        self,
+        error: APIServerError,
+    ) -> None:
+        logger.error(f"{error}, waiting {error.retry_after} seconds to the next request...")
+        await asyncio.sleep(error.retry_after)
+
+
+class Polling(ABCPolling):
+    __slots__ = (
+        "api",
+        "timeout",
+        "limit",
+        "allowed_updates",
+        "reconnect_after",
+        "max_reconnects",
+        "offset",
+        "_running",
+        "_reconnects_counter",
+        "_error_handler",
+    )
 
-class Polling(ABCPolling, typing.Generic[HTTPClient]):
     def __init__(
         self,
-        api: API[HTTPClient],
+        api: API,
         *,
-        offset: int = 0,
-        reconnection_timeout: float = 5.0,
-        max_reconnetions: int = 15,
-        include_updates: set[str | UpdateType] | None = None,
-        exclude_updates: set[str | UpdateType] | None = None,
+        timeout: int | None = None,
+        limit: int | None = None,
+        offset: int = DEFAULT_OFFSET,
+        reconnect_after: float = DEFAULT_RECONNECT_AFTER,
+        max_reconnects: int = DEFAULT_MAX_RECONNECTS,
+        include_updates: set[UpdateType] | None = None,
+        exclude_updates: set[UpdateType] | None = None,
     ) -> None:
         self.api = api
+        self.timeout = timeout
+        self.limit = limit
         self.allowed_updates = self.get_allowed_updates(
             include_updates=include_updates,
             exclude_updates=exclude_updates,
         )
-        self.reconnection_timeout = 5.0 if reconnection_timeout < 0 else reconnection_timeout
-        self.max_reconnetions = 15 if max_reconnetions < 0 else max_reconnetions
-        self.offset = offset
-        self._stop = True
+        self.reconnect_after = _compute_number(DEFAULT_RECONNECT_AFTER, reconnect_after, 0.0)
+        self.max_reconnects = _compute_number(DEFAULT_MAX_RECONNECTS, max_reconnects, 0)
+        self.offset = max(DEFAULT_OFFSET, offset)
+        self._running = False
+        self._reconnects_counter = 0
+        self._error_handler = PollingErrorHandler(self)
 
     def __repr__(self) -> str:
         return (
-            "<{}: with api={!r}, stopped={}, offset={}, allowed_updates={!r}, "
-            "max_reconnetions={}, reconnection_timeout={}>"
+            "<{}: api={!r}, running={}, offset={}, timeout={}, limit={}, "
+            "allowed_updates={!r}, max_reconnects={}, reconnect_after={}>"
         ).format(
-            self.__class__.__name__,
+            type(self).__name__,
             self.api,
-            self._stop,
+            self._running,
             self.offset,
+            self.timeout,
+            self.limit,
             self.allowed_updates,
-            self.max_reconnetions,
-            self.reconnection_timeout,
+            self.max_reconnects,
+            self.reconnect_after,
         )
 
     @staticmethod
     def get_allowed_updates(
         *,
-        include_updates: set[str | UpdateType] | None = None,
-        exclude_updates: set[str | UpdateType] | None = None,
-    ) -> list[str]:
-        allowed_updates: list[str] = list(x.value for x in UpdateType)
-        if not include_updates and not exclude_updates:
-            return allowed_updates
+        include_updates: set[UpdateType] | None = None,
+        exclude_updates: set[UpdateType] | None = None,
+    ) -> list[UpdateType]:
+        allowed_updates = list(UpdateType)
 
         if include_updates and exclude_updates:
-            allowed_updates = [x for x in allowed_updates if x in include_updates and x not in exclude_updates]
-        elif exclude_updates:
-            allowed_updates = [x for x in allowed_updates if x not in exclude_updates]
-        elif include_updates:
-            allowed_updates = [x for x in allowed_updates if x in include_updates]
+            return [x for x in allowed_updates if x in include_updates and x not in exclude_updates]
+
+        if exclude_updates:
+            return [x for x in allowed_updates if x not in exclude_updates]
+
+        if include_updates:
+            return [x for x in allowed_updates if x in include_updates]
 
-        return [x.value if isinstance(x, UpdateType) else x for x in allowed_updates]
+        return allowed_updates
+
+    @property
+    def reconnects_counter(self) -> int:
+        return self._reconnects_counter
+
+    @property
+    def running(self) -> bool:
+        return self._running
+
+    def _reset_reconnects_counter(self) -> None:
+        self._reconnects_counter = 0
 
     async def get_updates(self) -> msgspec.Raw:
-        raw_updates = await self.api.request_raw(
-            method="getUpdates",
-            data=dict(
-                offset=self.offset,
-                allowed_updates=self.allowed_updates,
-            ),
-        )
+        try:
+            raw_updates = await self.api.request_raw(
+                method="getUpdates",
+                data=dict(
+                    offset=self.offset,
+                    limit=self.limit,
+                    timeout=self.timeout,
+                    allowed_updates=self.allowed_updates,
+                ),
+                timeout=self.timeout or 0 + self.api.http.timeout,
+            )
+        except TimeoutError:
+            return msgspec.Raw(b"")
 
-        match raw_updates:
-            case Ok(value):
-                return value
-            case Error(err):
-                if err.code in (HTTPStatus.UNAUTHORIZED, HTTPStatus.NOT_FOUND):
-                    raise InvalidTokenError("Token seems to be invalid")
-                if err.code in (HTTPStatus.BAD_GATEWAY, HTTPStatus.GATEWAY_TIMEOUT):
-                    raise APIServerError("Unavilability of the API Telegram server")
-                raise err from None
+        if is_ok(raw_updates):
+            return raw_updates.value
+
+        match (error := raw_updates.error).status_code:
+            case HTTPStatus.TOO_MANY_REQUESTS:
+                error = APIServerError(
+                    message="Too many requests to get updates",
+                    retry_after=error.retry_after.unwrap_or(int(self.reconnect_after)),
+                )
+            case HTTPStatus.UNAUTHORIZED | HTTPStatus.NOT_FOUND as status:
+                error = InvalidTokenError(
+                    "Token seems to be invalid"
+                    if status == HTTPStatus.NOT_FOUND
+                    else "Invalid token (unauthorized)",
+                )
+            case HTTPStatus.BAD_GATEWAY | HTTPStatus.GATEWAY_TIMEOUT as status:
+                error = APIServerError(
+                    message="Telegram API server responded a {}".format(status.name.replace("_", " ")),
+                    retry_after=int(self.reconnect_after),
+                )
+
+        raise error from None
 
     async def listen(self) -> typing.AsyncGenerator[list[Update], None]:
         logger.debug("Listening polling")
-        reconn_counter = 0
-        self._stop = False
+        self._running = True
 
-        with decoder(list[Update]) as dec:  # For improve performance
-            while not self._stop:
+        with decoder(list[Update]) as dec:
+            while self._running:
                 try:
-                    updates = await self.get_updates()
-                    reconn_counter = 0
-                    updates_list = dec.decode(updates)
-                    if updates_list:
-                        yield updates_list
-                        self.offset = updates_list[-1].update_id + 1
-                except InvalidTokenError as e:
-                    logger.error(e)
-                    self.stop()
-                    sys.exit(3)
-                except APIServerError as e:
-                    logger.error(f"{e}, waiting {self.reconnection_timeout} seconds to the next request...")
-                    await asyncio.sleep(self.reconnection_timeout)
-                except asyncio.CancelledError:
-                    logger.info("Caught cancel, polling stopping...")
-                    self.stop()
-                except self.api.http.CONNECTION_TIMEOUT_ERRORS:
-                    if reconn_counter > self.max_reconnetions:
-                        logger.error(
-                            "Failed to reconnect to the server after {} attempts, polling stopping.",
-                            self.max_reconnetions,
-                        )
-                        self.stop()
-                        sys.exit(6)
-                    else:
-                        logger.warning(
-                            "Server disconnected, waiting {} seconds to reconnect...",
-                            self.reconnection_timeout,
-                        )
-                        reconn_counter += 1
-                        await asyncio.sleep(self.reconnection_timeout)
-                except self.api.http.CLIENT_CONNECTION_ERRORS:
-                    logger.error("Client connection failed, attempted to reconnect...")
-                    await asyncio.sleep(self.reconnection_timeout)
-                except BaseException as e:
-                    logger.exception("Traceback message below:")
+                    if (raw := await self.get_updates()) and (updates := dec.decode(raw)):
+                        yield updates
+                        self.offset = updates[-1].update_id + 1
+
+                    if self._reconnects_counter != 0:
+                        self._reset_reconnects_counter()
+                except BaseException as error:
+                    if not await self._error_handler.handle(error):
+                        logger.exception("Traceback message below:")
+
+                    if isinstance(error, self.api.http.CONNECTION_TIMEOUT_ERRORS):
+                        self._reconnects_counter += 1
 
     def stop(self) -> None:
-        self._stop = True
+        self._running = False
+        self._reset_reconnects_counter()
 
 
 __all__ = ("Polling",)

telegrinder/bot/rules/__init__.py

@@ -1,4 +1,4 @@
-from telegrinder.bot.rules.abc import ABCRule, AndRule, NotRule, OrRule
+from telegrinder.bot.rules.abc import ABCRule, AndRule, NotRule, OrRule, check_rule
 from telegrinder.bot.rules.callback_data import (
     CallbackDataEq,
     CallbackDataJsonEq,
@@ -6,11 +6,9 @@ from telegrinder.bot.rules.callback_data import (
     CallbackDataMap,
     CallbackDataMarkup,
     CallbackQueryDataRule,
-    CallbackQueryRule,
     HasData,
 )
 from telegrinder.bot.rules.chat_join import (
-    ChatJoinRequestRule,
     HasInviteLink,
     InviteLinkByCreator,
     InviteLinkName,
@@ -19,12 +17,10 @@ from telegrinder.bot.rules.command import Argument, Command
 from telegrinder.bot.rules.enum_text import EnumTextRule
 from telegrinder.bot.rules.func import FuncRule
 from telegrinder.bot.rules.fuzzy import FuzzyText
-from telegrinder.bot.rules.id import IdRule
 from telegrinder.bot.rules.inline import (
     HasLocation,
     InlineQueryChatType,
     InlineQueryMarkup,
-    InlineQueryRule,
     InlineQueryText,
 )
 from telegrinder.bot.rules.integer import IntegerInRange, IsInteger
@@ -49,7 +45,6 @@ from telegrinder.bot.rules.is_from import (
 from telegrinder.bot.rules.logic import If
 from telegrinder.bot.rules.markup import Markup
 from telegrinder.bot.rules.mention import HasMention
-from telegrinder.bot.rules.message import MessageRule
 from telegrinder.bot.rules.message_entities import HasEntities, MessageEntities
 from telegrinder.bot.rules.node import NodeRule
 from telegrinder.bot.rules.payload import (
@@ -59,10 +54,7 @@ from telegrinder.bot.rules.payload import (
     PayloadModelRule,
     PayloadRule,
 )
-from telegrinder.bot.rules.payment_invoice import (
-    PaymentInvoiceCurrency,
-    PaymentInvoiceRule,
-)
+from telegrinder.bot.rules.payment_invoice import PaymentInvoiceCurrency
 from telegrinder.bot.rules.regex import Regex
 from telegrinder.bot.rules.rule_enum import RuleEnum
 from telegrinder.bot.rules.start import StartCommand
@@ -80,8 +72,6 @@ __all__ = (
     "CallbackDataMap",
     "CallbackDataMarkup",
     "CallbackQueryDataRule",
-    "CallbackQueryRule",
-    "ChatJoinRequestRule",
     "Command",
     "EnumTextRule",
     "FuncRule",
@@ -93,11 +83,9 @@ __all__ = (
     "HasLocation",
     "HasMention",
     "HasText",
-    "IdRule",
     "If",
     "InlineQueryChatType",
     "InlineQueryMarkup",
-    "InlineQueryRule",
     "InlineQueryText",
     "IntegerInRange",
     "InviteLinkByCreator",
@@ -122,7 +110,6 @@ __all__ = (
     "IsUserId",
     "Markup",
     "MessageEntities",
-    "MessageRule",
     "NodeRule",
     "NotRule",
     "OrRule",
@@ -132,11 +119,11 @@ __all__ = (
     "PayloadModelRule",
     "PayloadRule",
     "PaymentInvoiceCurrency",
-    "PaymentInvoiceRule",
     "Regex",
     "RuleEnum",
     "StartCommand",
     "State",
     "StateMeta",
     "Text",
+    "check_rule",
 )