maxapi-python 2.0.1__py3-none-any.whl → 2.1.0__py3-none-any.whl

pymax/api/chats/service.py

@@ -12,9 +12,9 @@ from pymax.api.response import (
 from pymax.exceptions import PyMaxError
 from pymax.logging import get_logger
 from pymax.protocol import Opcode
-from pymax.types.domain import Chat, Message
+from pymax.types.domain import Chat, Member, Message
 
-from .enums import ChatLinkPrefix, ChatPayloadKey
+from .enums import ChatLinkPrefix, ChatMemberOperation, ChatPayloadKey
 from .payloads import (
     ChangeGroupProfilePayload,
     ChangeGroupSettingsOptions,
@@ -23,9 +23,11 @@ from .payloads import (
     CreateGroupMessage,
     CreateGroupPayload,
     FetchChatsPayload,
+    FetchJoinRequests,
     GetChatInfoPayload,
     InviteUsersPayload,
     JoinChatPayload,
+    JoinRequestActionPayload,
     LeaveChatPayload,
     LinkInfoPayload,
     RemoveUsersPayload,
@@ -70,7 +72,9 @@ class ChatService:
         if self.app.chats is None:
             return
 
-        self.app.chats = [chat for chat in self.app.chats if chat.id != chat_id]
+        self.app.chats = [
+            chat for chat in self.app.chats if chat.id != chat_id
+        ]
 
     @staticmethod
     def _process_chat_join_link(link: str) -> str | None:
@@ -108,7 +112,9 @@ class ChatService:
             return None
 
         chat = self._cache_chat(chat)
-        message = require_payload_model(response, Message).bind(self.app.api.messages)
+        message = require_payload_model(response, Message).bind(
+            self.app.api.messages
+        )
         return chat, message
 
     async def invite_users_to_group(
@@ -139,7 +145,9 @@ class ChatService:
         user_ids: list[int],
         show_history: bool = True,
     ) -> Chat | None:
-        return await self.invite_users_to_group(chat_id, user_ids, show_history)
+        return await self.invite_users_to_group(
+            chat_id, user_ids, show_history
+        )
 
     async def remove_users_from_group(
         self,
@@ -183,7 +191,9 @@ class ChatService:
             ),
         )
 
-        response = await self.app.invoke(Opcode.CHAT_UPDATE, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.CHAT_UPDATE, frame.to_payload()
+        )
         chat = parse_payload_item_model(response, ChatPayloadKey.CHAT, Chat)
         if chat:
             self._cache_chat(chat)
@@ -200,7 +210,9 @@ class ChatService:
             description=description,
         )
 
-        response = await self.app.invoke(Opcode.CHAT_UPDATE, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.CHAT_UPDATE, frame.to_payload()
+        )
         chat = parse_payload_item_model(response, ChatPayloadKey.CHAT, Chat)
         if chat:
             self._cache_chat(chat)
@@ -230,7 +242,9 @@ class ChatService:
 
     async def rework_invite_link(self, chat_id: int) -> Chat:
         frame = ReworkInviteLinkPayload(chat_id=chat_id)
-        response = await self.app.invoke(Opcode.CHAT_UPDATE, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.CHAT_UPDATE, frame.to_payload()
+        )
         chat = require_payload_item_model(response, ChatPayloadKey.CHAT, Chat)
         return self._cache_chat(chat)
 
@@ -240,12 +254,18 @@ class ChatService:
             for chat_id in chat_ids
             if (chat := self._get_cached_chat(chat_id)) is not None
         }
-        missed_chat_ids = [chat_id for chat_id in chat_ids if chat_id not in cached]
+        missed_chat_ids = [
+            chat_id for chat_id in chat_ids if chat_id not in cached
+        ]
 
         if missed_chat_ids:
             frame = GetChatInfoPayload(chat_ids=missed_chat_ids)
-            response = await self.app.invoke(Opcode.CHAT_INFO, frame.to_payload())
-            for chat in parse_payload_list(response, ChatPayloadKey.CHATS, Chat):
+            response = await self.app.invoke(
+                Opcode.CHAT_INFO, frame.to_payload()
+            )
+            for chat in parse_payload_list(
+                response, ChatPayloadKey.CHATS, Chat
+            ):
                 chat = self._cache_chat(chat)
                 cached[chat.id] = chat
 
@@ -272,6 +292,86 @@ class ChatService:
 
         chats = [
             self._cache_chat(chat)
-            for chat in parse_payload_list(response, ChatPayloadKey.CHATS, Chat)
+            for chat in parse_payload_list(
+                response, ChatPayloadKey.CHATS, Chat
+            )
         ]
         return chats
+
+    async def get_join_requests(
+        self, chat_id: int, count: int = 100
+    ) -> list[Member]:
+        frame = FetchJoinRequests(chat_id=chat_id, count=count)
+
+        response = await self.app.invoke(
+            Opcode.CHAT_MEMBERS, frame.to_payload()
+        )
+
+        return parse_payload_list(response, ChatPayloadKey.MEMBERS, Member)
+
+    async def confirm_join_requests(
+        self,
+        chat_id: int,
+        user_ids: list[int],
+        show_history: bool = True,
+    ) -> Chat | None:
+        frame = JoinRequestActionPayload(
+            chat_id=chat_id,
+            user_ids=user_ids,
+            show_history=show_history,
+            operation=ChatMemberOperation.ADD,
+        )
+
+        response = await self.app.invoke(
+            Opcode.CHAT_MEMBERS_UPDATE, frame.to_payload()
+        )
+
+        chat = parse_payload_item_model(response, ChatPayloadKey.CHAT, Chat)
+        if chat:
+            return self._cache_chat(chat)
+
+        return None
+
+    async def confirm_join_request(
+        self,
+        chat_id: int,
+        user_id: int,
+        show_history: bool = True,
+    ) -> Chat | None:
+        return await self.confirm_join_requests(
+            chat_id=chat_id,
+            user_ids=[user_id],
+            show_history=show_history,
+        )
+
+    async def decline_join_requests(
+        self,
+        chat_id: int,
+        user_ids: list[int],
+    ) -> Chat | None:
+        frame = JoinRequestActionPayload(
+            chat_id=chat_id,
+            user_ids=user_ids,
+            show_history=None,
+            operation=ChatMemberOperation.REMOVE,
+        )
+
+        response = await self.app.invoke(
+            Opcode.CHAT_MEMBERS_UPDATE, frame.to_payload()
+        )
+
+        chat = parse_payload_item_model(response, ChatPayloadKey.CHAT, Chat)
+        if chat:
+            return self._cache_chat(chat)
+
+        return None
+
+    async def decline_join_request(
+        self,
+        chat_id: int,
+        user_id: int,
+    ) -> Chat | None:
+        return await self.decline_join_requests(
+            chat_id=chat_id,
+            user_ids=[user_id],
+        )

pymax/api/facade.py

@@ -5,6 +5,7 @@ from typing import TYPE_CHECKING
 from pymax.logging import get_logger
 
 from .auth import AuthService
+from .bots import BotsService
 from .chats import ChatService
 from .messages import MessageService
 from .self import SelfService
@@ -25,6 +26,7 @@ class ApiFacade:
         self.messages = MessageService(app)
         self.chats = ChatService(app)
         self.users = UserService(app)
+        self.bots = BotsService(app)
         self.account = SelfService(app)
         self.session = SessionService(app)
         self.auth = AuthService(app)

pymax/api/messages/payloads.py

@@ -3,7 +3,11 @@ from typing import Any
 from pydantic import Field
 
 from pymax.api.models import CamelModel
-from pymax.api.uploads.payloads import AttachFilePayload, AttachPhotoPayload, VideoAttachPayload
+from pymax.api.uploads.payloads import (
+    AttachFilePayload,
+    AttachPhotoPayload,
+    VideoAttachPayload,
+)
 
 from .enums import ItemType, ReadAction
 

pymax/api/messages/service.py

@@ -9,7 +9,11 @@ from pymax.api.response import (
     payload_item,
     require_payload_model,
 )
-from pymax.api.uploads.payloads import AttachFilePayload, AttachPhotoPayload, VideoAttachPayload
+from pymax.api.uploads.payloads import (
+    AttachFilePayload,
+    AttachPhotoPayload,
+    VideoAttachPayload,
+)
 from pymax.exceptions import UploadError
 from pymax.files import File, Photo, Video
 from pymax.formatting.markdown import Formatter
@@ -65,13 +69,17 @@ class MessageService:
     async def _upload_attachments(
         self, attachments: SendAttachments
     ) -> list[AttachPhotoPayload | VideoAttachPayload | AttachFilePayload]:
-        result: list[AttachPhotoPayload | VideoAttachPayload | AttachFilePayload] = []
+        result: list[
+            AttachPhotoPayload | VideoAttachPayload | AttachFilePayload
+        ] = []
         if not attachments:
             return result
 
         for attachment in attachments:
             if isinstance(attachment, Photo):
-                upload_result = await self.app.api.uploads.upload_photo(attachment)
+                upload_result = await self.app.api.uploads.upload_photo(
+                    attachment
+                )
                 if not upload_result:
                     logger.error("Photo uploading failed")
                     raise UploadError("Photo uploading failed")
@@ -79,7 +87,9 @@ class MessageService:
                 result.append(upload_result)
 
             elif isinstance(attachment, Video):
-                upload_result = await self.app.api.uploads.upload_video(attachment)
+                upload_result = await self.app.api.uploads.upload_video(
+                    attachment
+                )
                 if not upload_result:
                     logger.error("Video uploading failed")
                     raise UploadError("Video uploading failed")
@@ -87,7 +97,9 @@ class MessageService:
                 result.append(upload_result)
 
             elif isinstance(attachment, File):
-                upload_result = await self.app.api.uploads.upload_file(attachment)
+                upload_result = await self.app.api.uploads.upload_file(
+                    attachment
+                )
                 if not upload_result:
                     logger.error("File uploading failed")
                     raise UploadError("File uploading failed")
@@ -105,7 +117,9 @@ class MessageService:
         *,
         notify: bool = True,
     ) -> Message | None:
-        logger.info("sending message chat_id=%s text_len=%s", chat_id, len(text))
+        logger.info(
+            "sending message chat_id=%s text_len=%s", chat_id, len(text)
+        )
 
         clean_text, elements = Formatter.format_markdown(text)
 
@@ -157,7 +171,10 @@ class MessageService:
             Opcode.CHAT_HISTORY,
             payload=frame.to_payload(),
         )
-        return parse_payload_list(response, MessagePayloadKey.MESSAGES, Message) or None
+        return (
+            parse_payload_list(response, MessagePayloadKey.MESSAGES, Message)
+            or None
+        )
 
     async def delete_message(
         self,
@@ -178,7 +195,9 @@ class MessageService:
         )
 
         await self.app.invoke(Opcode.MSG_DELETE, frame.to_payload())
-        logger.info("messages deleted chat_id=%s count=%s", chat_id, len(message_ids))
+        logger.info(
+            "messages deleted chat_id=%s count=%s", chat_id, len(message_ids)
+        )
         return True
 
     async def pin_message(
@@ -200,7 +219,9 @@ class MessageService:
         )
 
         await self.app.invoke(Opcode.CHAT_UPDATE, frame.to_payload())
-        logger.info("message pinned chat_id=%s message_id=%s", chat_id, message_id)
+        logger.info(
+            "message pinned chat_id=%s message_id=%s", chat_id, message_id
+        )
         return True
 
     async def get_video_by_id(
@@ -242,7 +263,9 @@ class MessageService:
             file_id=file_id,
         )
 
-        response = await self.app.invoke(Opcode.FILE_DOWNLOAD, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.FILE_DOWNLOAD, frame.to_payload()
+        )
         return parse_payload_model(response, FileRequest)
 
     async def add_reaction(
@@ -263,7 +286,9 @@ class MessageService:
             reaction=ReactionInfoPayload(id=reaction),
         )
 
-        response = await self.app.invoke(Opcode.MSG_REACTION, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.MSG_REACTION, frame.to_payload()
+        )
         reaction_info = payload_item(response, MessagePayloadKey.REACTION_INFO)
         if reaction_info:
             return ReactionInfo.model_validate(reaction_info)

pymax/api/self/service.py

@@ -37,7 +37,9 @@ class SelfService:
     async def request_profile_photo_upload_url(self) -> str:
         logger.info("requesting profile photo upload url")
         frame = UploadPayload(profile=True)
-        response = await self.app.invoke(Opcode.PHOTO_UPLOAD, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.PHOTO_UPLOAD, frame.to_payload()
+        )
         return str(require_payload_item(response, SelfPayloadKey.URL))
 
     async def change_profile(
@@ -84,13 +86,17 @@ class SelfService:
             include=chat_include,
             filters=filters or [],
         )
-        response = await self.app.invoke(Opcode.FOLDERS_UPDATE, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.FOLDERS_UPDATE, frame.to_payload()
+        )
         return require_payload_model(response, FolderUpdate)
 
     async def get_folders(self, folder_sync: int = 0) -> FolderList:
         logger.info("fetching folders")
         frame = GetFolderPayload(folder_sync=folder_sync)
-        response = await self.app.invoke(Opcode.FOLDERS_GET, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.FOLDERS_GET, frame.to_payload()
+        )
         return require_payload_model(response, FolderList)
 
     async def update_folder(
@@ -109,13 +115,17 @@ class SelfService:
             filters=filters or [],
             options=options or [],
         )
-        response = await self.app.invoke(Opcode.FOLDERS_UPDATE, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.FOLDERS_UPDATE, frame.to_payload()
+        )
         return require_payload_model(response, FolderUpdate)
 
     async def delete_folder(self, folder_id: str) -> FolderUpdate:
         logger.info("deleting folder")
         frame = DeleteFolderPayload(folder_ids=[folder_id])
-        response = await self.app.invoke(Opcode.FOLDERS_DELETE, frame.to_payload())
+        response = await self.app.invoke(
+            Opcode.FOLDERS_DELETE, frame.to_payload()
+        )
         return require_payload_model(response, FolderUpdate)
 
     async def close_all_sessions(self) -> bool:
@@ -129,7 +139,9 @@ class SelfService:
         token = payload_item(response, SelfPayloadKey.TOKEN, str)
 
         if not token:
-            logger.warning("no token received after closing sessions, skipping token update")
+            logger.warning(
+                "no token received after closing sessions, skipping token update"
+            )
             return False
 
         await self.app.store.update_token(self.app.session.token, token)

pymax/api/session/payloads.py

@@ -52,10 +52,17 @@ class MobileUserAgentPayload(CamelModel):
             by_alias=True,
             exclude_none=True,
         )
-        if self.device_type == DeviceType.WEB and "headerUserAgent" not in payload:
+        if (
+            self.device_type == DeviceType.WEB
+            and "headerUserAgent" not in payload
+        ):
             payload["headerUserAgent"] = DEFAULT_WEB_HEADER_USER_AGENT
 
-        return {alias: payload[alias] for alias in WEB_USER_AGENT_ALIASES if alias in payload}
+        return {
+            alias: payload[alias]
+            for alias in WEB_USER_AGENT_ALIASES
+            if alias in payload
+        }
 
 
 class MobileHandshakePayload(CamelModel):

pymax/api/uploads/models.py

@@ -1,9 +1,6 @@
-from typing import Literal
-
-from pydantic import BaseModel, Field
+from pydantic import BaseModel
 
 from pymax.api.models import CamelModel
-from pymax.types import AttachmentType
 
 
 class PhotoPayloadResponse(BaseModel):

pymax/api/uploads/payloads.py

@@ -5,18 +5,24 @@ from pymax.types import AttachmentType
 
 
 class AttachPhotoPayload(CamelModel):
-    type: AttachmentType = Field(default=AttachmentType.PHOTO, serialization_alias="_type")
+    type: AttachmentType = Field(
+        default=AttachmentType.PHOTO, serialization_alias="_type"
+    )
     photo_token: str
 
 
 class VideoAttachPayload(CamelModel):
-    type: AttachmentType = Field(default=AttachmentType.VIDEO, serialization_alias="_type")
+    type: AttachmentType = Field(
+        default=AttachmentType.VIDEO, serialization_alias="_type"
+    )
     video_id: int
     token: str
 
 
 class AttachFilePayload(CamelModel):
-    type: AttachmentType = Field(default=AttachmentType.FILE, serialization_alias="_type")
+    type: AttachmentType = Field(
+        default=AttachmentType.FILE, serialization_alias="_type"
+    )
     file_id: int