telegrinder 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

telegrinder/bot/rules/abc.py

@@ -1,238 +1,213 @@
-import inspect
-from abc import ABC, abstractmethod
-from functools import cached_property
-
-import typing_extensions as typing
-
-from telegrinder.bot.cute_types import MessageCute, UpdateCute
-from telegrinder.bot.dispatch.context import Context
-from telegrinder.bot.dispatch.process import check_rule
-from telegrinder.bot.rules.adapter import ABCAdapter, RawUpdateAdapter
-from telegrinder.bot.rules.adapter.node import Event
-from telegrinder.node.base import Node, get_nodes, is_node
-from telegrinder.tools.i18n.abc import ABCTranslator
-from telegrinder.tools.magic import (
-    cache_translation,
-    get_annotations,
-    get_cached_translation,
-    get_default_args,
-)
-from telegrinder.types.objects import Update as UpdateObject
-
-if typing.TYPE_CHECKING:
-    from telegrinder.node.composer import NodeCollection
-
-AdaptTo = typing.TypeVar("AdaptTo", default=typing.Any, contravariant=True)
-
-CheckResult: typing.TypeAlias = bool | typing.Coroutine[typing.Any, typing.Any, bool]
-Message: typing.TypeAlias = MessageCute
-Update: typing.TypeAlias = UpdateCute
-
-
-def with_caching_translations(func: typing.Callable[..., typing.Any]):
-    """Should be used as decorator for .translate method. Caches rule translations."""
-
-    async def wrapper(self: "ABCRule", translator: ABCTranslator):
-        if translation := get_cached_translation(self, translator.locale):
-            return translation
-        translation = await func(self, translator)
-        cache_translation(self, translator.locale, translation)
-        return translation
-
-    return wrapper
-
-
-class ABCRule(ABC, typing.Generic[AdaptTo]):
-    adapter: ABCAdapter[UpdateObject, AdaptTo]
-    requires: list["ABCRule"] = []
-
-    if typing.TYPE_CHECKING:
-
-        @typing.overload
-        def check(self) -> CheckResult: ...
-
-        @typing.overload
-        def check(self, event: AdaptTo, /) -> CheckResult: ...
-
-        @typing.overload
-        def check(self, event: AdaptTo, ctx: Context, /) -> CheckResult: ...
-
-        @typing.overload
-        def check(
-            self,
-            event: AdaptTo,
-            ctx: Context,
-            /,
-            *args: typing.Any,
-            **kwargs: typing.Any,
-        ) -> CheckResult: ...
-
-        @typing.overload
-        def check(self, event: AdaptTo, /, *args: typing.Any, **kwargs: typing.Any) -> CheckResult: ...
-
-        @typing.overload
-        def check(self, ctx: Context, /, *args: typing.Any, **kwargs: typing.Any) -> CheckResult: ...
-
-        @abstractmethod
-        def check(self, *args: typing.Any, **kwargs: typing.Any) -> CheckResult:
-            pass
-    else:
-        adapter = RawUpdateAdapter()
-
-        @abstractmethod
-        def check(self, *args, **kwargs):
-            pass
-
-    def __init_subclass__(cls, requires: list["ABCRule"] | None = None) -> None:
-        """Merges requirements from inherited classes and rule-specific requirements."""
-
-        requirements = []
-        for base in inspect.getmro(cls):
-            if issubclass(base, ABCRule) and base != cls:
-                requirements.extend(base.requires or ())  # type: ignore
-
-        requirements.extend(requires or ())
-        cls.requires = list(dict.fromkeys(requirements))
-
-    def __and__(self, other: "ABCRule") -> "AndRule":
-        """And Rule.
-
-        ```python
-        rule = HasText() & HasCaption()
-        rule #> AndRule(HasText(), HasCaption()) -> True if all rules in an AndRule are True, otherwise False.
-        ```
-        """
-
-        return AndRule(self, other)
-
-    def __or__(self, other: "ABCRule") -> "OrRule":
-        """Or Rule.
-
-        ```python
-        rule = HasText() | HasCaption()
-        rule #> OrRule(HasText(), HasCaption()) -> True if any rule in an OrRule are True, otherwise False.
-        ```
-        """
-
-        return OrRule(self, other)
-
-    def __invert__(self) -> "NotRule":
-        """Not Rule.
-
-        ```python
-        rule = ~HasText()
-        rule # NotRule(HasText()) -> True if rule returned False, otherwise False.
-        ```
-        """
-
-        return NotRule(self)
-
-    def __repr__(self) -> str:
-        return "<{}: adapter={!r}>".format(
-            self.__class__.__name__,
-            self.adapter,
-        )
-
-    @cached_property
-    def required_nodes(self) -> dict[str, type[Node]]:
-        return get_nodes(self.check)
-
-    def as_optional(self) -> "ABCRule":
-        return self | Always()
-
-    def should_fail(self) -> "ABCRule":
-        return self & Never()
-
-    async def bounding_check(
-        self,
-        adapted_value: AdaptTo,
-        ctx: Context,
-        node_col: "NodeCollection | None" = None,
-    ) -> bool:
-        bound_check_rule = self.check
-        kw = {}
-        node_col_values = node_col.values if node_col is not None else {}
-        temp_ctx = get_default_args(bound_check_rule) | ctx
-
-        for i, (k, v) in enumerate(get_annotations(bound_check_rule).items()):
-            if (isinstance(adapted_value, Event) and i == 0) or (  # First arg is Event
-                isinstance(v, type) and isinstance(adapted_value, v)
-            ):
-                kw[k] = adapted_value if not isinstance(adapted_value, Event) else adapted_value.obj
-            elif is_node(v):
-                assert k in node_col_values, "Node is undefined, error while bounding."
-                kw[k] = node_col_values[k]
-            elif k in temp_ctx:
-                kw[k] = temp_ctx[k]
-            elif v is Context:
-                kw[k] = ctx
-            else:
-                raise LookupError(
-                    f"Cannot bound {k!r} of type {v!r} to '{self.__class__.__qualname__}.check()', "
-                    "because it cannot be resolved."
-                )
-
-        result = bound_check_rule(**kw)  # type: ignore
-        if inspect.isawaitable(result):
-            result = await result
-        return result
-
-    async def translate(self, translator: ABCTranslator) -> typing.Self:
-        return self
-
-
-class AndRule(ABCRule):
-    def __init__(self, *rules: ABCRule) -> None:
-        self.rules = rules
-
-    async def check(self, event: Update, ctx: Context) -> bool:
-        ctx_copy = ctx.copy()
-        for rule in self.rules:
-            if not await check_rule(event.ctx_api, rule, event, ctx_copy):
-                return False
-        ctx |= ctx_copy
-        return True
-
-
-class OrRule(ABCRule):
-    def __init__(self, *rules: ABCRule) -> None:
-        self.rules = rules
-
-    async def check(self, event: Update, ctx: Context) -> bool:
-        for rule in self.rules:
-            ctx_copy = ctx.copy()
-            if await check_rule(event.ctx_api, rule, event, ctx_copy):
-                ctx |= ctx_copy
-                return True
-        return False
-
-
-class NotRule(ABCRule):
-    def __init__(self, rule: ABCRule) -> None:
-        self.rule = rule
-
-    async def check(self, event: Update, ctx: Context) -> bool:
-        ctx_copy = ctx.copy()
-        return not await check_rule(event.ctx_api, self.rule, event, ctx_copy)
-
-
-class Never(ABCRule):
-    async def check(self) -> typing.Literal[False]:
-        return False
-
-
-class Always(ABCRule):
-    async def check(self) -> typing.Literal[True]:
-        return True
-
-
-__all__ = (
-    "ABCRule",
-    "Always",
-    "AndRule",
-    "Never",
-    "NotRule",
-    "OrRule",
-    "CheckResult",
-    "with_caching_translations",
-)
+import inspect
+from abc import ABC, abstractmethod
+from functools import cached_property
+
+import typing_extensions as typing
+
+from telegrinder.bot.cute_types import MessageCute, UpdateCute
+from telegrinder.bot.dispatch.context import Context
+from telegrinder.bot.dispatch.process import check_rule
+from telegrinder.bot.rules.adapter import ABCAdapter, RawUpdateAdapter
+from telegrinder.bot.rules.adapter.node import Event
+from telegrinder.node.base import Node, get_nodes, is_node
+from telegrinder.tools.i18n.abc import ABCTranslator
+from telegrinder.tools.magic import (
+    cache_translation,
+    get_annotations,
+    get_cached_translation,
+    get_default_args,
+)
+from telegrinder.types.objects import Update as UpdateObject
+
+if typing.TYPE_CHECKING:
+    from telegrinder.node.composer import NodeCollection
+
+AdaptTo = typing.TypeVar("AdaptTo", default=typing.Any, contravariant=True)
+
+CheckResult: typing.TypeAlias = bool | typing.Awaitable[bool]
+Message: typing.TypeAlias = MessageCute
+Update: typing.TypeAlias = UpdateCute
+
+
+def with_caching_translations(func: typing.Callable[..., typing.Any]):
+    """Should be used as decorator for .translate method. Caches rule translations."""
+
+    async def wrapper(self: "ABCRule", translator: ABCTranslator):
+        if translation := get_cached_translation(self, translator.locale):
+            return translation
+        translation = await func(self, translator)
+        cache_translation(self, translator.locale, translation)
+        return translation
+
+    return wrapper
+
+
+class ABCRule(ABC, typing.Generic[AdaptTo]):
+    adapter: ABCAdapter[UpdateObject, AdaptTo]
+    requires: list["ABCRule"] = []
+
+    if typing.TYPE_CHECKING:
+
+        @abstractmethod
+        def check(self, *args: typing.Any, **kwargs: typing.Any) -> CheckResult:
+            pass
+    else:
+        adapter = RawUpdateAdapter()
+
+        @abstractmethod
+        def check(self, *args, **kwargs):
+            pass
+
+    def __init_subclass__(cls, requires: list["ABCRule"] | None = None) -> None:
+        """Merges requirements from inherited classes and rule-specific requirements."""
+
+        requirements = []
+        for base in inspect.getmro(cls):
+            if issubclass(base, ABCRule) and base != cls:
+                requirements.extend(base.requires or ())  # type: ignore
+
+        requirements.extend(requires or ())
+        cls.requires = list(dict.fromkeys(requirements))
+
+    def __and__(self, other: "ABCRule") -> "AndRule":
+        """And Rule.
+
+        ```python
+        rule = HasText() & HasCaption()
+        rule #> AndRule(HasText(), HasCaption()) -> True if all rules in an AndRule are True, otherwise False.
+        ```
+        """
+
+        return AndRule(self, other)
+
+    def __or__(self, other: "ABCRule") -> "OrRule":
+        """Or Rule.
+
+        ```python
+        rule = HasText() | HasCaption()
+        rule #> OrRule(HasText(), HasCaption()) -> True if any rule in an OrRule are True, otherwise False.
+        ```
+        """
+
+        return OrRule(self, other)
+
+    def __invert__(self) -> "NotRule":
+        """Not Rule.
+
+        ```python
+        rule = ~HasText()
+        rule # NotRule(HasText()) -> True if rule returned False, otherwise False.
+        ```
+        """
+
+        return NotRule(self)
+
+    def __repr__(self) -> str:
+        return "<{}: adapter={!r}>".format(
+            self.__class__.__name__,
+            self.adapter,
+        )
+
+    @cached_property
+    def required_nodes(self) -> dict[str, type[Node]]:
+        return get_nodes(self.check)
+
+    def as_optional(self) -> "ABCRule":
+        return self | Always()
+
+    def should_fail(self) -> "ABCRule":
+        return self & Never()
+
+    async def bounding_check(
+        self,
+        adapted_value: AdaptTo,
+        ctx: Context,
+        node_col: "NodeCollection | None" = None,
+    ) -> bool:
+        bound_check_rule = self.check
+        kw = {}
+        node_col_values = node_col.values if node_col is not None else {}
+        temp_ctx = get_default_args(bound_check_rule) | ctx
+
+        for i, (k, v) in enumerate(get_annotations(bound_check_rule).items()):
+            if (isinstance(adapted_value, Event) and i == 0) or (  # First arg is Event
+                isinstance(v, type) and isinstance(adapted_value, v)
+            ):
+                kw[k] = adapted_value if not isinstance(adapted_value, Event) else adapted_value.obj
+            elif is_node(v):
+                assert k in node_col_values, "Node is undefined, error while bounding."
+                kw[k] = node_col_values[k]
+            elif k in temp_ctx:
+                kw[k] = temp_ctx[k]
+            elif v is Context:
+                kw[k] = ctx
+            else:
+                raise LookupError(
+                    f"Cannot bound {k!r} of type {v!r} to '{self.__class__.__qualname__}.check()', "
+                    "because it cannot be resolved."
+                )
+
+        result = bound_check_rule(**kw)  # type: ignore
+        if inspect.isawaitable(result):
+            result = await result
+        return result
+
+    async def translate(self, translator: ABCTranslator) -> typing.Self:
+        return self
+
+
+class AndRule(ABCRule):
+    def __init__(self, *rules: ABCRule) -> None:
+        self.rules = rules
+
+    async def check(self, event: Update, ctx: Context) -> bool:
+        ctx_copy = ctx.copy()
+        for rule in self.rules:
+            if not await check_rule(event.ctx_api, rule, event, ctx_copy):
+                return False
+        ctx |= ctx_copy
+        return True
+
+
+class OrRule(ABCRule):
+    def __init__(self, *rules: ABCRule) -> None:
+        self.rules = rules
+
+    async def check(self, event: Update, ctx: Context) -> bool:
+        for rule in self.rules:
+            ctx_copy = ctx.copy()
+            if await check_rule(event.ctx_api, rule, event, ctx_copy):
+                ctx |= ctx_copy
+                return True
+        return False
+
+
+class NotRule(ABCRule):
+    def __init__(self, rule: ABCRule) -> None:
+        self.rule = rule
+
+    async def check(self, event: Update, ctx: Context) -> bool:
+        ctx_copy = ctx.copy()
+        return not await check_rule(event.ctx_api, self.rule, event, ctx_copy)
+
+
+class Never(ABCRule):
+    async def check(self) -> typing.Literal[False]:
+        return False
+
+
+class Always(ABCRule):
+    async def check(self) -> typing.Literal[True]:
+        return True
+
+
+__all__ = (
+    "ABCRule",
+    "Always",
+    "AndRule",
+    "CheckResult",
+    "Never",
+    "NotRule",
+    "OrRule",
+    "with_caching_translations",
+)

telegrinder/bot/rules/adapter/__init__.py

@@ -1,14 +1,14 @@
-from telegrinder.bot.rules.adapter.abc import ABCAdapter, Event
-from telegrinder.bot.rules.adapter.errors import AdapterError
-from telegrinder.bot.rules.adapter.event import EventAdapter
-from telegrinder.bot.rules.adapter.node import NodeAdapter
-from telegrinder.bot.rules.adapter.raw_update import RawUpdateAdapter
-
-__all__ = (
+from telegrinder.bot.rules.adapter.abc import ABCAdapter, Event
+from telegrinder.bot.rules.adapter.errors import AdapterError
+from telegrinder.bot.rules.adapter.event import EventAdapter
+from telegrinder.bot.rules.adapter.node import NodeAdapter
+from telegrinder.bot.rules.adapter.raw_update import RawUpdateAdapter
+
+__all__ = (
     "ABCAdapter",
     "AdapterError",
     "Event",
     "EventAdapter",
     "NodeAdapter",
-    "RawUpdateAdapter",
-)
+    "RawUpdateAdapter",
+)

telegrinder/bot/rules/adapter/abc.py

@@ -1,29 +1,29 @@
-import abc
-import dataclasses
-import typing
-
-from fntypes.result import Result
-
-from telegrinder.api import API
-from telegrinder.bot.dispatch.context import Context
-from telegrinder.bot.rules.adapter.errors import AdapterError
-from telegrinder.model import Model
-
-From = typing.TypeVar("From", bound=Model)
-To = typing.TypeVar("To")
-
-
-class ABCAdapter(abc.ABC, typing.Generic[From, To]):
-    ADAPTED_VALUE_KEY: str | None = None
-
-    @abc.abstractmethod
-    async def adapt(self, api: API, update: From, context: Context) -> Result[To, AdapterError]:
-        pass
-
-
-@dataclasses.dataclass(slots=True)
-class Event(typing.Generic[To]):
-    obj: To
-
-
-__all__ = ("ABCAdapter", "Event")
+import abc
+import dataclasses
+import typing
+
+from fntypes.result import Result
+
+from telegrinder.api.api import API
+from telegrinder.bot.dispatch.context import Context
+from telegrinder.bot.rules.adapter.errors import AdapterError
+from telegrinder.model import Model
+
+From = typing.TypeVar("From", bound=Model)
+To = typing.TypeVar("To")
+
+
+class ABCAdapter(abc.ABC, typing.Generic[From, To]):
+    ADAPTED_VALUE_KEY: str | None = None
+
+    @abc.abstractmethod
+    async def adapt(self, api: API, update: From, context: Context) -> Result[To, AdapterError]:
+        pass
+
+
+@dataclasses.dataclass(slots=True)
+class Event(typing.Generic[To]):
+    obj: To
+
+
+__all__ = ("ABCAdapter", "Event")

telegrinder/bot/rules/adapter/errors.py

@@ -1,5 +1,5 @@
-class AdapterError(RuntimeError):
-    pass
-
-
-__all__ = ("AdapterError",)
+class AdapterError(RuntimeError):
+    pass
+
+
+__all__ = ("AdapterError",)