maxapi-python 2.4.0__py3-none-any.whl → 2.5.0__py3-none-any.whl

pymax/filters/base.py

@@ -0,0 +1,218 @@
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+from inspect import isawaitable
+from typing import TYPE_CHECKING, Any, Generic, TypeVar
+
+if TYPE_CHECKING:
+    from collections.abc import Awaitable
+
+    from pymax.base import BaseClient
+
+
+MISSING = object()
+
+EventT = TypeVar("EventT")
+ClientT = TypeVar("ClientT", bound="BaseClient")
+
+
+class Filter(ABC, Generic[EventT, ClientT]):
+    """Базовый класс фильтра событий.
+
+    Реализуйте ``__call__(event, client)`` с результатом ``bool`` или
+    ``Awaitable[bool]`` и передайте экземпляр в декоратор роутера. Фильтры
+    объединяются через ``&``, ``|`` и ``~``; составной фильтр при ручном вызове
+    нужно ожидать через ``await``.
+    """
+
+    @abstractmethod
+    def __call__(
+        self,
+        event: EventT,
+        client: ClientT,
+        /,
+    ) -> bool | Awaitable[bool]: ...
+    def __and__(self, other: Filter[EventT, ClientT]) -> AndFilter:
+        return AndFilter(self, other)
+
+    def __or__(self, value: Filter[EventT, ClientT]) -> OrFilter:
+        return OrFilter(self, value)
+
+    def __invert__(self) -> NotFilter:
+        return NotFilter(self)
+
+    async def resolve_filter_result(
+        self,
+        result: bool | Awaitable[bool],
+    ) -> bool:
+        if isawaitable(result):
+            return await result
+
+        return result
+
+
+class PathResolver:
+    def __init__(self, path: list[str]) -> None:
+        self.path = path
+
+    def resolve_path(self, event: Any) -> Any | None:
+        value = event
+
+        for attr in self.path:
+            value = getattr(value, attr, MISSING)
+
+            if value is MISSING:
+                return MISSING
+
+        return value
+
+
+class IsNoneFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+    ) -> None:
+        super().__init__(path)
+        self.expected_type = expected_type
+        self.path = path
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if value is MISSING:
+            return False
+
+        return value is None
+
+
+class IsEmptyFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+    ) -> None:
+        super().__init__(path)
+        self.expected_type = expected_type
+        self.path = path
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        return value == []
+
+
+class FieldRef(Filter[EventT, ClientT], PathResolver):
+    def __init__(self, path: list[str], expected_type: type[EventT]) -> None:
+        super().__init__(path)
+        self.expected_type: type[EventT] = expected_type
+        self.path = path
+
+    def __eq__(self, value: object) -> EqFilter:  # pyright: ignore[reportIncompatibleMethodOverride] # ty: ignore[invalid-method-override]
+        return EqFilter(self.path, self.expected_type, value)
+
+    def __ne__(self, value: object) -> NeFilter:  # pyright: ignore[reportIncompatibleMethodOverride] # ty: ignore[invalid-method-override]
+        return NeFilter(self.path, self.expected_type, value)
+
+    def is_none(self) -> IsNoneFilter:
+        return IsNoneFilter(self.path, expected_type=self.expected_type)
+
+    def is_empty(self) -> IsEmptyFilter:
+        return IsEmptyFilter(self.path, expected_type=self.expected_type)
+
+    def __call__(self, event: EventT, _: ClientT, /) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if value is MISSING:
+            return False
+
+        return bool(value)
+
+
+class BinaryFilter(Filter[EventT, ClientT]):
+    def __init__(self, left: Filter[EventT, ClientT], right: Filter[EventT, ClientT]) -> None:
+        self.left = left
+        self.right = right
+
+
+class NotFilter(Filter[EventT, ClientT]):
+    def __init__(self, filter_instance: Filter[EventT, ClientT]) -> None:
+        self.filter = filter_instance
+
+    async def __call__(self, event: EventT, client: ClientT) -> bool:
+        return not await self.resolve_filter_result(self.filter(event, client))
+
+
+class NeFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        expected: object,
+    ) -> None:
+        self.path: list[str] = path
+        self.expected_type = expected_type
+        self.expected: object = expected
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if value is MISSING:
+            return False
+
+        return value != self.expected
+
+
+class AndFilter(BinaryFilter[EventT, ClientT]):
+    async def __call__(self, event: EventT, client: ClientT) -> bool:
+        left = await self.resolve_filter_result(self.left(event, client))
+
+        if not left:
+            return False
+
+        return await self.resolve_filter_result(self.right(event, client))
+
+
+class OrFilter(BinaryFilter[EventT, ClientT]):
+    async def __call__(self, event: EventT, client: ClientT) -> bool:
+        left = await self.resolve_filter_result(self.left(event, client))
+
+        if left:
+            return True
+
+        return await self.resolve_filter_result(self.right(event, client))
+
+
+class EqFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        expected: object,
+    ) -> None:
+        self.path: list[str] = path
+        self.expected_type = expected_type
+        self.expected: object = expected
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if value is MISSING:
+            return False
+
+        return value == self.expected

pymax/filters/containers/__init__.py

@@ -0,0 +1,2 @@
+from .chat import ChatContainer
+from .message import MessageContainer

pymax/filters/containers/chat.py

@@ -0,0 +1,240 @@
+from __future__ import annotations
+
+import builtins
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from pymax.filters.base import ClientT, Filter, PathResolver
+from pymax.filters.refs import FieldRef, NumberFieldRef, TextFieldRef
+
+from .message import MessageContainer
+
+if TYPE_CHECKING:
+    pass
+
+EventT = TypeVar("EventT")
+
+
+class ChatContainer(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        expected_type: builtins.type[EventT],
+        path: list[str] | None = None,
+    ) -> None:
+        self.expected_type: type[EventT] = expected_type
+        self.path = path or []
+
+    @property
+    def id(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "id"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def type(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "type"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def status(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "status"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def owner(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "owner"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def participants(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "participants"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def title(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "title"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def base_raw_icon_url(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "base_raw_icon_url"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def base_icon_url(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "base_icon_url"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def last_message(self) -> MessageContainer:
+        return MessageContainer(
+            path=[*self.path, "last_message"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def last_event_time(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "last_event_time"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def last_delayed_update_time(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "last_delayed_update_time"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def last_fire_delayed_error_time(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "last_fire_delayed_error_time"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def created(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "created"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def new_messages(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "new_messages"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def link(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "link"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def access(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "access"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def restrictions(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "restrictions"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def pinned_message(self) -> MessageContainer:
+        return MessageContainer(
+            path=[*self.path, "pinned_message"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def participants_count(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "participants_count"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def description(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "description"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def options(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "options"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def join_time(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "join_time"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def invited_by(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "invited_by"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def modified(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "modified"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def messages_count(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "messages_count"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def has_bots(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "has_bots"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def prev_message_id(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "prev_message_id"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def admin_participants(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "admin_participants"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def admins(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "admins"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def cid(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "cid"],
+            expected_type=self.expected_type,
+        )
+
+    def __call__(self, event: Any, client: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        return bool(self.resolve_path(event))

pymax/filters/containers/message.py

@@ -0,0 +1,147 @@
+from __future__ import annotations
+
+import builtins
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from pymax.filters.base import ClientT, Filter, PathResolver
+from pymax.filters.refs import FieldRef, NumberFieldRef, TextFieldRef
+
+if TYPE_CHECKING:
+    pass
+
+EventT = TypeVar("EventT")
+
+
+class MessageContainer(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        expected_type: builtins.type[EventT],
+        path: list[str] | None = None,
+    ) -> None:
+        self.expected_type: type[EventT] = expected_type
+        self.path = path or []
+
+    @property
+    def text(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "text"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def id(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "id"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def chat_id(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "chat_id"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def sender(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "sender"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def time(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "time"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def type(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "type"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def cid(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "cid"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def attaches(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "attaches"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def stats(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "stats"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def status(self) -> TextFieldRef:
+        return TextFieldRef(
+            path=[*self.path, "status"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def reaction_info(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "reaction_info"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def options(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "options"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def prev_message_id(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "prev_message_id"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def ttl(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "ttl"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def unread(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "unread"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def mark(self) -> NumberFieldRef:
+        return NumberFieldRef(
+            path=[*self.path, "mark"],
+            expected_type=self.expected_type,
+        )
+
+    @property
+    def elements(self) -> FieldRef:
+        return FieldRef(
+            path=[*self.path, "elements"],
+            expected_type=self.expected_type,
+        )
+
+    def __call__(self, event: Any, client: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        return bool(self.resolve_path(event))