maxapi-python 1.2.5__py3-none-any.whl → 2.0.0__py3-none-any.whl

pymax/mixins/channel.py

@@ -1,130 +0,0 @@
-from pymax.exceptions import Error, ResponseError, ResponseStructureError
-from pymax.payloads import (
-    GetGroupMembersPayload,
-    JoinChatPayload,
-    ResolveLinkPayload,
-    SearchGroupMembersPayload,
-)
-from pymax.protocols import ClientProtocol
-from pymax.static.constant import (
-    DEFAULT_CHAT_MEMBERS_LIMIT,
-    DEFAULT_MARKER_VALUE,
-)
-from pymax.static.enum import Opcode
-from pymax.types import Channel, Member
-from pymax.utils import MixinsUtils
-
-
-class ChannelMixin(ClientProtocol):
-    async def resolve_channel_by_name(self, name: str) -> Channel | None:
-        """
-        Получает информацию о канале по его имени
-
-        :param name: Имя канала
-        :type name: str
-        :return: Объект Channel или None, если канал не найден
-        :rtype: Channel | None
-        """
-        payload = ResolveLinkPayload(
-            link=f"https://max.ru/{name}",
-        ).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.LINK_INFO, payload=payload)
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        channel = Channel.from_dict(data.get("payload", {}).get("chat", {}))
-        if channel not in self.channels:
-            self.channels.append(channel)
-        return channel
-
-    async def join_channel(self, link: str) -> Channel | None:
-        """
-        Присоединяется к каналу по ссылке
-
-        :param link: Ссылка на канал
-        :type link: str
-        :return: Объект канала, если присоединение прошло успешно, иначе None
-        :rtype: Channel | None
-        """
-        payload = JoinChatPayload(
-            link=link,
-        ).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_JOIN, payload=payload)
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        channel = Channel.from_dict(data.get("payload", {}).get("chat", {}))
-        if channel not in self.channels:
-            self.channels.append(channel)
-        return channel
-
-    async def _query_members(
-        self, payload: GetGroupMembersPayload | SearchGroupMembersPayload
-    ) -> tuple[list[Member], int | None]:
-        data = await self._send_and_wait(
-            opcode=Opcode.CHAT_MEMBERS,
-            payload=payload.model_dump(by_alias=True, exclude_none=True),
-        )
-        response_payload = data.get("payload", {})
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-        marker = response_payload.get("marker")
-        if isinstance(marker, str):
-            marker = int(marker)
-        elif isinstance(marker, int):
-            pass
-        elif marker is None:
-            # маркер может отсутствовать
-            pass
-        else:
-            raise ResponseStructureError("Invalid marker type in response")
-        members = response_payload.get("members")
-        member_list = []
-        if isinstance(members, list):
-            for item in members:
-                if not isinstance(item, dict):
-                    raise ResponseStructureError("Invalid member structure in response")
-                member_list.append(Member.from_dict(item))
-        else:
-            raise ResponseStructureError("Invalid members type in response")
-        return member_list, marker
-
-    async def load_members(
-        self,
-        chat_id: int,
-        marker: int | None = DEFAULT_MARKER_VALUE,
-        count: int = DEFAULT_CHAT_MEMBERS_LIMIT,
-    ) -> tuple[list[Member], int | None]:
-        """
-        Загружает членов канала
-
-        :param chat_id: Идентификатор канала
-        :type chat_id: int
-        :param marker: Маркер для пагинации. По умолчанию DEFAULT_MARKER_VALUE
-        :type marker: int | None
-        :param count: Количество членов для загрузки. По умолчанию DEFAULT_CHAT_MEMBERS_LIMIT.
-        :type count: int
-        :return: Список участников канала и маркер для следующей страницы
-        :rtype: tuple[list[Member], int | None]
-        """
-
-        payload = GetGroupMembersPayload(chat_id=chat_id, marker=marker, count=count)
-        return await self._query_members(payload)
-
-    async def find_members(self, chat_id: int, query: str) -> tuple[list[Member], int | None]:
-        """
-        Поиск участников канала по строке
-        Внимание! веб-клиент всегда возвращает только определённое количество пользователей,
-        тоесть пагинация здесь не реализована!
-
-        :param chat_id: Идентификатор канала
-        :type chat_id: int
-        :param query: Строка для поиска участников
-        :type query: str
-        :return: Список участников канала
-        :rtype: tuple[list[Member], int | None]
-        """
-        payload = SearchGroupMembersPayload(chat_id=chat_id, query=query)
-        return await self._query_members(payload)

pymax/mixins/group.py

@@ -1,458 +0,0 @@
-import time
-
-from pymax.exceptions import Error
-from pymax.payloads import (
-    ChangeGroupProfilePayload,
-    ChangeGroupSettingsOptions,
-    ChangeGroupSettingsPayload,
-    CreateGroupAttach,
-    CreateGroupMessage,
-    CreateGroupPayload,
-    FetchChatsPayload,
-    GetChatInfoPayload,
-    InviteUsersPayload,
-    JoinChatPayload,
-    LeaveChatPayload,
-    RemoveUsersPayload,
-    ReworkInviteLinkPayload,
-)
-from pymax.protocols import ClientProtocol
-from pymax.static.enum import Opcode
-from pymax.types import Chat, Message
-from pymax.utils import MixinsUtils
-
-
-class GroupMixin(ClientProtocol):
-    async def create_group(
-        self,
-        name: str,
-        participant_ids: list[int] | None = None,
-        notify: bool = True,
-    ) -> tuple[Chat, Message] | None:
-        """
-        Создает группу
-
-        Args:
-            name (str): Название группы.
-            participant_ids (list[int] | None, optional): Список идентификаторов участников. Defaults to None.
-            notify (bool, optional): Флаг оповещения. Defaults to True.
-
-        Returns:
-            tuple[Chat, Message] | None: Объект Chat и Message или None при ошибке.
-        """
-        payload = CreateGroupPayload(
-            message=CreateGroupMessage(
-                cid=int(time.time() * 1000),
-                attaches=[
-                    CreateGroupAttach(
-                        _type="CONTROL",
-                        title=name,
-                        user_ids=(participant_ids if participant_ids else []),
-                    )
-                ],
-            ),
-            notify=notify,
-        ).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.MSG_SEND, payload=payload)
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"]["chat"])
-        message = Message.from_dict(data["payload"])
-
-        if chat:
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-        return chat, message
-
-    async def invite_users_to_group(
-        self,
-        chat_id: int,
-        user_ids: list[int],
-        show_history: bool = True,
-    ) -> Chat | None:
-        """
-        Приглашает пользователей в группу
-
-        Args:
-            chat_id (int): ID группы.
-            user_ids (list[int]): Список идентификаторов пользователей.
-            show_history (bool, optional): Флаг оповещения. Defaults to True.
-
-        Returns:
-            Chat | None: Объект Chat или None при ошибке.
-        """
-        payload = InviteUsersPayload(
-            chat_id=chat_id,
-            user_ids=user_ids,
-            show_history=show_history,
-            operation="add",
-        ).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_MEMBERS_UPDATE, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"]["chat"])
-        if chat:
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-        return chat
-
-    async def invite_users_to_channel(
-        self,
-        chat_id: int,
-        user_ids: list[int],
-        show_history: bool = True,
-    ) -> Chat | None:
-        """
-        Приглашает пользователей в канал
-
-        Args:
-            chat_id (int): ID канала.
-            user_ids (list[int]): Список идентификаторов пользователей.
-            show_history (bool, optional): Флаг оповещения. Defaults to True.
-
-        Returns:
-            Chat | None: Объект Chat или None при ошибке.
-        """
-        return await self.invite_users_to_group(chat_id, user_ids, show_history)
-
-    async def remove_users_from_group(
-        self,
-        chat_id: int,
-        user_ids: list[int],
-        clean_msg_period: int,
-    ) -> bool:
-        """
-        Удаляет пользователей из группы
-
-        Args:
-            chat_id (int): ID группы.
-            user_ids (list[int]): Список идентификаторов пользователей.
-            clean_msg_period (int): Период очистки сообщений.
-
-        Returns:
-            bool: True, если удаление прошло успешно, иначе False.
-        """
-        payload = RemoveUsersPayload(
-            chat_id=chat_id,
-            user_ids=user_ids,
-            clean_msg_period=clean_msg_period,
-        ).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_MEMBERS_UPDATE, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"]["chat"])
-        if chat:
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-        return True
-
-    async def change_group_settings(
-        self,
-        chat_id: int,
-        all_can_pin_message: bool | None = None,
-        only_owner_can_change_icon_title: bool | None = None,
-        only_admin_can_add_member: bool | None = None,
-        only_admin_can_call: bool | None = None,
-        members_can_see_private_link: bool | None = None,
-    ) -> None:
-        """
-        Изменяет настройки группы
-
-        Args:
-            chat_id (int): ID группы.
-            all_can_pin_message (bool | None, optional): Все могут закреплять сообщения. Defaults to None.
-            only_owner_can_change_icon_title (bool | None, optional): Только владелец может менять иконку и название. Defaults to None.
-            only_admin_can_add_member (bool | None, optional): Только администраторы могут добавлять участников. Defaults to None.
-            only_admin_can_call (bool | None, optional): Только администраторы могут звонить. Defaults to None.
-            members_can_see_private_link (bool | None, optional): Участники могут видеть приватную ссылку. Defaults to None.
-        Returns:
-            None
-        """
-        payload = ChangeGroupSettingsPayload(
-            chat_id=chat_id,
-            options=ChangeGroupSettingsOptions(
-                ALL_CAN_PIN_MESSAGE=all_can_pin_message,
-                ONLY_OWNER_CAN_CHANGE_ICON_TITLE=only_owner_can_change_icon_title,
-                ONLY_ADMIN_CAN_ADD_MEMBER=only_admin_can_add_member,
-                ONLY_ADMIN_CAN_CALL=only_admin_can_call,
-                MEMBERS_CAN_SEE_PRIVATE_LINK=members_can_see_private_link,
-            ),
-        ).model_dump(by_alias=True, exclude_none=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_UPDATE, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"]["chat"])
-        if chat:
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-    async def change_group_profile(
-        self,
-        chat_id: int,
-        name: str | None,
-        description: str | None = None,
-    ) -> None:
-        """
-        Изменяет профиль группы
-
-        Args:
-            chat_id (int): ID группы.
-            name (str | None): Название группы.
-            description (str | None, optional): Описание группы. Defaults to None.
-
-        Returns:
-            None
-        """
-        payload = ChangeGroupProfilePayload(
-            chat_id=chat_id,
-            theme=name,
-            description=description,
-        ).model_dump(by_alias=True, exclude_none=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_UPDATE, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"]["chat"])
-        if chat:
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-    def _process_chat_join_link(self, link: str) -> str | None:
-        idx = link.find("join/")
-        return link[idx:] if idx != -1 else None
-
-    async def join_group(self, link: str) -> Chat:
-        """
-        Вступает в группу по ссылке
-
-        Args:
-            link (str): Ссылка на группу.
-
-        Returns:
-            Chat: Объект чата группы
-        """
-        proceed_link = self._process_chat_join_link(link)
-        if proceed_link is None:
-            raise ValueError("Invalid group link")
-
-        payload = JoinChatPayload(link=proceed_link).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_JOIN, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"]["chat"])
-        if chat:
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-        return chat
-
-    async def resolve_group_by_link(self, link: str) -> Chat | None:
-        """
-        Разрешает группу по ссылке
-
-        Args:
-            link (str): Ссылка на группу.
-
-        Returns:
-            Chat | None: Объект чата группы или None, если не найдено.
-        """
-        proceed_link = self._process_chat_join_link(link)
-        if proceed_link is None:
-            raise ValueError("Invalid group link")
-
-        data = await self._send_and_wait(
-            opcode=Opcode.LINK_INFO,
-            payload={
-                "link": proceed_link,
-            },
-        )
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"].get("chat", {}))
-        return chat
-
-    async def rework_invite_link(self, chat_id: int) -> Chat:
-        """
-        Пересоздает ссылку для приглашения в группу
-
-        Args:
-            chat_id (int): ID группы.
-
-        Returns:
-            Chat: Обновленный объект чата с новой ссылкой.
-        """
-        payload = ReworkInviteLinkPayload(chat_id=chat_id).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_UPDATE, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chat = Chat.from_dict(data["payload"].get("chat"))
-        if not chat:
-            raise Error("no_chat", "Chat data missing in response", "Chat Error")
-
-        return chat
-
-    async def get_chats(self, chat_ids: list[int]) -> list[Chat]:
-        """
-        Получает информацию о группах по их ID
-
-        :param chat_ids: Список идентификаторов групп.
-        :type chat_ids: list[int]
-        :return: Список объектов Chat.
-        :rtype: list[Chat]
-        """
-        missed_chat_ids = [
-            chat_id for chat_id in chat_ids if await self._get_chat(chat_id) is None
-        ]
-        if missed_chat_ids:
-            payload = GetChatInfoPayload(chat_ids=missed_chat_ids).model_dump(by_alias=True)
-        else:
-            chats: list[Chat] = [
-                chat for chat_id in chat_ids if (chat := await self._get_chat(chat_id)) is not None
-            ]
-            return chats
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_INFO, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chats_data = data["payload"].get("chats", [])
-        chats: list[Chat] = []
-        for chat_dict in chats_data:
-            chat = Chat.from_dict(chat_dict)
-            chats.append(chat)
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-        return chats
-
-    async def get_chat(self, chat_id: int) -> Chat:
-        """
-        Получает информацию о группе по ее ID
-
-        Args:
-            chat_id (int): Идентификатор группы.
-
-        Returns:
-            Chat: Объект Chat.
-        """
-        chats = await self.get_chats([chat_id])
-        if not chats:
-            raise Error("no_chat", "Chat not found in response", "Chat Error")
-        return chats[0]
-
-    async def leave_group(self, chat_id: int) -> None:
-        """
-        Покидает группу
-
-        :param chat_id: Идентификатор группы.
-        :type chat_id: int
-        :return: None
-        :rtype: None
-        """
-        payload = LeaveChatPayload(chat_id=chat_id).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHAT_LEAVE, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        cached_chat = await self._get_chat(chat_id)
-        if cached_chat is not None:
-            self.chats.remove(cached_chat)
-
-    async def leave_channel(self, chat_id: int) -> None:
-        """
-        Покидает канал
-
-        :param chat_id: Идентификатор канала.
-        :type chat_id: int
-        :return: None
-        :rtype: None
-        """
-        await self.leave_group(chat_id)
-
-    async def fetch_chats(self, marker: int | None = None) -> list[Chat]:
-        """
-        Загружает список чатов
-
-        :param marker: Маркер для пагинации, по умолчанию None
-        :type marker: int | None
-        :return: Список объектов Chat
-        :rtype: list[Chat]
-        """
-        if marker is None:
-            marker = int(time.time() * 1000)
-
-        payload = FetchChatsPayload(marker=marker).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.CHATS_LIST, payload=payload)
-
-        if data.get("payload", {}).get("error"):
-            MixinsUtils.handle_error(data)
-
-        chats_data = data["payload"].get("chats", [])
-        chats: list[Chat] = []
-        for chat_dict in chats_data:
-            chat = Chat.from_dict(chat_dict)
-            chats.append(chat)
-            cached_chat = await self._get_chat(chat.id)
-            if cached_chat is None:
-                self.chats.append(chat)
-            else:
-                idx = self.chats.index(cached_chat)
-                self.chats[idx] = chat
-
-        return chats