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

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))

pymax/filters/filters.py

@@ -0,0 +1,211 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, TypeVar
+
+from .base import ClientT, Filter, PathResolver
+
+if TYPE_CHECKING:
+    pass
+
+EventT = TypeVar("EventT")
+
+
+class ContainsFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        text: str,
+        ignore_case: bool,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.text = text
+        self.ignore_case = ignore_case
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, str):
+            return False
+
+        if self.ignore_case:
+            return self.text.casefold() in value.casefold()
+
+        return self.text in value
+
+
+class StartsWithFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        text: str,
+        ignore_case: bool,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.text = text
+        self.ignore_case = ignore_case
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, str):
+            return False
+
+        if self.ignore_case:
+            return value.casefold().startswith(self.text.casefold())
+
+        return value.startswith(self.text)
+
+
+class EndsWithFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        text: str,
+        ignore_case: bool,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.text = text
+        self.ignore_case = ignore_case
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, str):
+            return False
+
+        if self.ignore_case:
+            return value.casefold().endswith(self.text.casefold())
+
+        return value.endswith(self.text)
+
+
+class LessThanFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        expected: float,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.expected = expected
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, int | float) or isinstance(value, bool):
+            return False
+
+        return value < self.expected
+
+
+class LessThanOrEqualFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        expected: float,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.expected = expected
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, int | float) or isinstance(value, bool):
+            return False
+
+        return value <= self.expected
+
+
+class GreaterThanFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        expected: float,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.expected = expected
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, int | float) or isinstance(value, bool):
+            return False
+
+        return value > self.expected
+
+
+class GreaterThanOrEqualFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        expected: float,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.expected = expected
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, int | float) or isinstance(value, bool):
+            return False
+
+        return value >= self.expected
+
+
+class BetweenFilter(Filter[EventT, ClientT], PathResolver):
+    def __init__(
+        self,
+        path: list[str],
+        expected_type: type[EventT],
+        minimum: float,
+        maximum: float,
+    ) -> None:
+        self.path = path
+        self.expected_type = expected_type
+        self.minimum = minimum
+        self.maximum = maximum
+
+    def __call__(self, event: EventT, _: ClientT) -> bool:
+        if not isinstance(event, self.expected_type):
+            return False
+
+        value = self.resolve_path(event)
+
+        if not isinstance(value, int | float) or isinstance(value, bool):
+            return False
+
+        return self.minimum <= value <= self.maximum

pymax/filters/magic.py

@@ -0,0 +1,13 @@
+from pymax.types import Chat, Message
+
+from .containers import ChatContainer, MessageContainer
+
+
+class MagicFilter:
+    @property
+    def message(self) -> MessageContainer:
+        return MessageContainer(Message)
+
+    @property
+    def chat(self) -> ChatContainer:
+        return ChatContainer(Chat)

pymax/filters/presets.py

@@ -0,0 +1,88 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from pymax.types import LinkType, Message
+
+from .base import Filter
+
+if TYPE_CHECKING:
+    from pymax.base import BaseClient
+    from pymax.types.domain.message import Attachment
+
+
+class Command(Filter[Message, "BaseClient"]):
+    """Проверяет команду в первом слове текста сообщения.
+
+    Args:
+        command: Имя команды, например ``"start"`` для ``/start``.
+        prefix: Префикс команды или кортеж допустимых префиксов.
+
+    Аргументы после команды не влияют на совпадение.
+    """
+
+    def __init__(self, command: str, prefix: str | tuple[str, ...] = "/") -> None:
+        self.prefixes = (prefix,) if isinstance(prefix, str) else prefix
+
+        for prefix_ in self.prefixes:
+            self.command = command.removeprefix(prefix_)
+
+    def __call__(self, message: Message, _: BaseClient) -> bool:
+        if not message.text:
+            return False
+
+        parts = message.text.split(maxsplit=1)
+        if not parts:
+            return False
+
+        dirty_command = parts[0]
+
+        for prefix in self.prefixes:
+            if message.text.startswith(prefix) and dirty_command[len(prefix) :] == self.command:
+                return True
+
+        return False
+
+
+class Reply(Filter[Message, "BaseClient"]):
+    """Проверяет, является ли сообщение ответом на другое сообщение."""
+
+    def __call__(self, message: Message, _: BaseClient) -> bool:
+        if not message.link:
+            return False
+
+        return message.link.type == LinkType.REPLY
+
+
+class Forwarded(Filter[Message, "BaseClient"]):
+    """Проверяет, является ли сообщение пересланным."""
+
+    def __call__(self, message: Message, _: BaseClient) -> bool:
+        if not message.link:
+            return False
+
+        return message.link.type == LinkType.FORWARD
+
+
+class FromMe(Filter[Message, "BaseClient"]):
+    """Проверяет, отправлено ли сообщение текущим пользователем клиента."""
+
+    def __call__(self, message: Message, client: BaseClient) -> bool:
+        if not message.sender or not client.me:
+            return False
+
+        return message.sender == client.me.contact.id
+
+
+class HasAttachment(Filter[Message, "BaseClient"]):
+    """Проверяет наличие хотя бы одного вложения указанных типов.
+
+    Передавайте классы входящих вложений из ``pymax.types``, например
+    ``PhotoAttachment``, а не классы файлов для отправки.
+    """
+
+    def __init__(self, *attachment_types: type[Attachment]) -> None:
+        self.expected_types = attachment_types
+
+    def __call__(self, message: Message, _: BaseClient) -> bool:
+        return any(isinstance(attach, self.expected_types) for attach in message.attaches)