maxapi-python 0.1.3__py3-none-any.whl → 1.0.1__py3-none-any.whl

pymax/navigation.py

@@ -0,0 +1,185 @@
+import random
+
+
+class Navigation:
+    SCREENS_GRAPH = {  # noqa: RUF012
+        "chats_list_tab": [
+            "chat",
+            "contacts_tab",
+            "call_history_tab",
+            "settings_tab",
+            "create_chat",
+            "chat_attachments_voices",
+        ],
+        "chat": [
+            "chats_list_tab",
+            "chat_attachments_media",
+        ],
+        "contacts_tab": [
+            "call_history_tab",
+            "chats_list_tab",
+            "settings_tab",
+            "create_chat",
+        ],
+        "call_history_tab": [
+            "chats_list_tab",
+            "settings_tab",
+            "contacts_tab",
+        ],
+        "settings_tab": [
+            "settings_folders",
+            "settings_privacy",
+            "settings_notifications",
+            "settings_chat_decoration",
+            "call_history_tab",
+            "contacts_tab",
+            "chats_list_tab",
+        ],
+        "settings_folders": [
+            "settings_tab",
+            "chats_list_tab",
+            "contacts_tab",
+            "call_history_tab",
+        ],
+        "settings_privacy": [
+            "settings_tab",
+            "chats_list_tab",
+            "contacts_tab",
+            "call_history_tab",
+        ],
+        "settings_notifications": [
+            "settings_tab",
+            "contacts_tab",
+            "call_history_tab",
+            "chats_list_tab",
+        ],
+        "settings_chat_decoration": [
+            "settings_tab",
+            "chats_list_tab",
+            "contacts_tab",
+            "call_history_tab",
+        ],
+        "create_chat": [
+            "chats_list_tab",
+            "contacts_tab",
+        ],
+        "chat_attachments_media": [
+            "chat_attachments_files",
+            "chat_attachments_voices",
+            "chat_attachments_links",
+            "chat",
+        ],
+        "chat_attachments_files": [
+            "chat_attachments_voices",
+            "chat_attachments_media",
+            "chat_attachments_links",
+            "chat",
+        ],
+        "chat_attachments_voices": [
+            "chat_attachments_links",
+            "chat_attachments_media",
+            "chat_attachments_files",
+            "chat",
+        ],
+        "chat_attachments_links": [
+            "chat_attachments_media",
+            "chat_attachments_files",
+            "chat_attachments_voices",
+            "chat",
+        ],
+    }
+    SCREENS = {  # noqa: RUF012
+        "application_background": 1,
+        "auth_sign_method": 50,
+        "auth_phone_login": 51,
+        "auth_otp": 52,
+        "auth_empty_profile": 53,
+        "auth_avatars": 54,
+        "contacts_tab": 100,
+        "contacts_search": 102,
+        "contacts_search_by_phone": 103,
+        "chats_list_tab": 150,
+        "chats_list_search_initial": 151,
+        "chats_list_search_result": 152,
+        "create_chat": 200,
+        "create_chat_members_picker": 201,
+        "create_chat_info": 202,
+        "avatar_picker_gallery": 250,
+        "avatar_picker_crop": 251,
+        "avatar_picker_camera": 252,
+        "avatar_viewer": 253,
+        "call_history_tab": 300,
+        "call_new_call": 302,
+        "call_create_group_link": 303,
+        "call_add_participants": 304,
+        "call": 305,
+        "chat": 350,
+        "chat_attach_picker": 351,
+        "chat_attach_picker_media_viewer": 352,
+        "chat_attach_picker_camera": 353,
+        "chat_share_location": 354,
+        "chat_share_contact": 355,
+        "chat_forward": 357,
+        "chat_media_viewer": 358,
+        "chat_system_file_viewer": 359,
+        "chat_location_viewer": 360,
+        "chat_info": 400,
+        "chat_info_all_participants": 401,
+        "chat_info_editing": 402,
+        "chat_info_add_participants": 403,
+        "chat_info_administrators": 404,
+        "chat_info_add_administrator": 405,
+        "chat_info_blocked_participants": 406,
+        "chat_info_change_owner": 407,
+        "chat_attachments_media": 408,
+        "chat_attachments_files": 409,
+        "chat_attachments_links": 410,
+        "chat_info_invite_link": 411,
+        "chat_attachments_voices": 412,
+        "settings_tab": 450,
+        "settings_profile_editing": 451,
+        "settings_shortname_change": 452,
+        "settings_phone_change": 453,
+        "settings_notifications": 454,
+        "settings_notifications_system": 455,
+        "settings_folders": 456,
+        "settings_privacy": 457,
+        "settings_privacy_block_list": 458,
+        "settings_media": 459,
+        "settings_messages": 460,
+        "settings_stickers": 461,
+        "settings_chat_decoration": 462,
+        "settings_phone_change_phone_input": 463,
+        "settings_phone_change_phone_otp": 464,
+        "settings_cache": 465,
+        "settings_profile_avatars": 466,
+        "settings_about_application": 467,
+        "settings_privacy_sensitive_content": 479,
+        "miniapp": 500,
+    }
+
+    @classmethod
+    def get_screen_id(cls, screen_name: str) -> int:
+        screen_id = cls.SCREENS.get(screen_name)
+
+        if screen_id is None:
+            raise ValueError(f"Unknown screen name: {screen_name}")
+
+        return screen_id
+
+    @classmethod
+    def can_navigate(cls, from_screen: str, to_screen: str) -> bool:
+        if from_screen == to_screen:
+            return True
+        return to_screen in cls.SCREENS_GRAPH.get(from_screen, [])
+
+    @classmethod
+    def get_random_navigation(cls, screen_name: str) -> str:
+        return random.choice(cls.SCREENS_GRAPH.get(screen_name, []))  # noqa: S311
+
+    @classmethod
+    def get_screen_name(cls, screen_id: int) -> str | None:
+        for name, id_ in cls.SCREENS.items():
+            if id_ == screen_id:
+                return name
+        return None

pymax/payloads.py

@@ -1,175 +1,195 @@
-from typing import Any, Literal
-
-from pydantic import BaseModel, Field
-
-from pymax.static import AttachType, AuthType
-
-
-def to_camel(string: str) -> str:
-    parts = string.split("_")
-    return parts[0] + "".join(word.capitalize() for word in parts[1:])
-
-
-class CamelModel(BaseModel):
-    model_config = {
-        "alias_generator": to_camel,
-        "populate_by_name": True,
-    }
-
-
-class BaseWebSocketMessage(BaseModel):
-    ver: int = 11
-    cmd: int
-    seq: int
-    opcode: int
-    payload: dict[str, Any]
-
-
-class RequestCodePayload(CamelModel):
-    phone: str
-    type: AuthType = AuthType.START_AUTH
-    language: str = "ru"
-
-
-class SendCodePayload(CamelModel):
-    token: str
-    verify_code: str
-    auth_token_type: AuthType = AuthType.CHECK_CODE
-
-
-class SyncPayload(CamelModel):
-    interactive: bool = True
-    token: str
-    chats_sync: int = 0
-    contacts_sync: int = 0
-    presence_sync: int = 0
-    drafts_sync: int = 0
-    chats_count: int = 40
-
-
-class ReplyLink(CamelModel):
-    type: str = "REPLY"
-    message_id: str
-
-
-class UploadPhotoPayload(CamelModel):
-    count: int = 1
-
-
-class AttachPhotoPayload(CamelModel):
-    type: AttachType = Field(AttachType.PHOTO, alias="_type")
-    photo_token: str
-
-
-class SendMessagePayloadMessage(CamelModel):
-    text: str
-    cid: int
-    elements: list[Any]
-    attaches: list[dict[str, Any]]
-    link: ReplyLink | None = None
-
-
-class SendMessagePayload(CamelModel):
-    chat_id: int
-    message: SendMessagePayloadMessage
-    notify: bool = False
-
-
-class EditMessagePayload(CamelModel):
-    chat_id: int
-    message_id: int
-    text: str
-    elements: list[Any]
-    attaches: list[Any]
-
-
-class DeleteMessagePayload(CamelModel):
-    chat_id: int
-    message_ids: list[int]
-    for_me: bool = False
-
-
-class FetchContactsPayload(CamelModel):
-    contact_ids: list[int]
-
-
-class FetchHistoryPayload(CamelModel):
-    chat_id: int
-    from_time: int = Field(alias="from")
-    forward: int
-    backward: int = 200
-    get_messages: bool = True
-
-
-class ChangeProfilePayload(CamelModel):
-    first_name: str
-    last_name: str | None = None
-    description: str | None = None
-
-
-class ResolveLinkPayload(CamelModel):
-    link: str
-
-
-class PinMessagePayload(CamelModel):
-    chat_id: int
-    notify_pin: bool
-    pin_message_id: int
-
-
-class CreateGroupAttach(CamelModel):
-    type: Literal["CONTROL"] = Field("CONTROL", alias="_type")
-    event: str = "new"
-    chat_type: str = "CHAT"
-    title: str
-    user_ids: list[int]
-
-
-class CreateGroupMessage(CamelModel):
-    cid: int
-    attaches: list[CreateGroupAttach]
-
-
-class CreateGroupPayload(CamelModel):
-    message: CreateGroupMessage
-    notify: bool = True
-
-
-class InviteUsersPayload(CamelModel):
-    chat_id: int
-    user_ids: list[int]
-    show_history: bool
-    operation: str = "add"
-
-
-class RemoveUsersPayload(CamelModel):
-    chat_id: int
-    user_ids: list[int]
-    operation: str = "remove"
-    clean_msg_period: int
-
-
-class ChangeGroupSettingsOptions(BaseModel):
-    ONLY_OWNER_CAN_CHANGE_ICON_TITLE: bool | None
-    ALL_CAN_PIN_MESSAGE: bool | None
-    ONLY_ADMIN_CAN_ADD_MEMBER: bool | None
-    ONLY_ADMIN_CAN_CALL: bool | None
-    MEMBERS_CAN_SEE_PRIVATE_LINK: bool | None
-
-
-class ChangeGroupSettingsPayload(CamelModel):
-    chat_id: int
-    options: ChangeGroupSettingsOptions
-
-
-class ChangeGroupProfilePayload(CamelModel):
-    chat_id: int
-    theme: str | None
-    description: str | None
-
-
-class GetGroupMembersPayload(CamelModel):
-    type: str = "MEMBER"
-    marker: int
-    chat_id: int
-    count: int
+from typing import Any, Literal
+
+from pydantic import BaseModel, Field
+
+from pymax.static import AttachType, AuthType
+
+
+def to_camel(string: str) -> str:
+    parts = string.split("_")
+    return parts[0] + "".join(word.capitalize() for word in parts[1:])
+
+
+class CamelModel(BaseModel):
+    model_config = {
+        "alias_generator": to_camel,
+        "populate_by_name": True,
+    }
+
+
+class BaseWebSocketMessage(BaseModel):
+    ver: int = 11
+    cmd: int
+    seq: int
+    opcode: int
+    payload: dict[str, Any]
+
+
+class RequestCodePayload(CamelModel):
+    phone: str
+    type: AuthType = AuthType.START_AUTH
+    language: str = "ru"
+
+
+class SendCodePayload(CamelModel):
+    token: str
+    verify_code: str
+    auth_token_type: AuthType = AuthType.CHECK_CODE
+
+
+class SyncPayload(CamelModel):
+    interactive: bool = True
+    token: str
+    chats_sync: int = 0
+    contacts_sync: int = 0
+    presence_sync: int = 0
+    drafts_sync: int = 0
+    chats_count: int = 40
+
+
+class ReplyLink(CamelModel):
+    type: str = "REPLY"
+    message_id: str
+
+
+class UploadPhotoPayload(CamelModel):
+    count: int = 1
+
+
+class AttachPhotoPayload(CamelModel):
+    type: AttachType = Field(AttachType.PHOTO, alias="_type")
+    photo_token: str
+
+
+class SendMessagePayloadMessage(CamelModel):
+    text: str
+    cid: int
+    elements: list[Any]
+    attaches: list[dict[str, Any]]
+    link: ReplyLink | None = None
+
+
+class SendMessagePayload(CamelModel):
+    chat_id: int
+    message: SendMessagePayloadMessage
+    notify: bool = False
+
+
+class EditMessagePayload(CamelModel):
+    chat_id: int
+    message_id: int
+    text: str
+    elements: list[Any]
+    attaches: list[Any]
+
+
+class DeleteMessagePayload(CamelModel):
+    chat_id: int
+    message_ids: list[int]
+    for_me: bool = False
+
+
+class FetchContactsPayload(CamelModel):
+    contact_ids: list[int]
+
+
+class FetchHistoryPayload(CamelModel):
+    chat_id: int
+    from_time: int = Field(alias="from")
+    forward: int
+    backward: int = 200
+    get_messages: bool = True
+
+
+class ChangeProfilePayload(CamelModel):
+    first_name: str
+    last_name: str | None = None
+    description: str | None = None
+
+
+class ResolveLinkPayload(CamelModel):
+    link: str
+
+
+class PinMessagePayload(CamelModel):
+    chat_id: int
+    notify_pin: bool
+    pin_message_id: int
+
+
+class CreateGroupAttach(CamelModel):
+    type: Literal["CONTROL"] = Field("CONTROL", alias="_type")
+    event: str = "new"
+    chat_type: str = "CHAT"
+    title: str
+    user_ids: list[int]
+
+
+class CreateGroupMessage(CamelModel):
+    cid: int
+    attaches: list[CreateGroupAttach]
+
+
+class CreateGroupPayload(CamelModel):
+    message: CreateGroupMessage
+    notify: bool = True
+
+
+class InviteUsersPayload(CamelModel):
+    chat_id: int
+    user_ids: list[int]
+    show_history: bool
+    operation: str = "add"
+
+
+class RemoveUsersPayload(CamelModel):
+    chat_id: int
+    user_ids: list[int]
+    operation: str = "remove"
+    clean_msg_period: int
+
+
+class ChangeGroupSettingsOptions(BaseModel):
+    ONLY_OWNER_CAN_CHANGE_ICON_TITLE: bool | None
+    ALL_CAN_PIN_MESSAGE: bool | None
+    ONLY_ADMIN_CAN_ADD_MEMBER: bool | None
+    ONLY_ADMIN_CAN_CALL: bool | None
+    MEMBERS_CAN_SEE_PRIVATE_LINK: bool | None
+
+
+class ChangeGroupSettingsPayload(CamelModel):
+    chat_id: int
+    options: ChangeGroupSettingsOptions
+
+
+class ChangeGroupProfilePayload(CamelModel):
+    chat_id: int
+    theme: str | None
+    description: str | None
+
+
+class GetGroupMembersPayload(CamelModel):
+    type: str = "MEMBER"
+    marker: int
+    chat_id: int
+    count: int
+
+
+class NavigationEventParams(BaseModel):
+    action_id: int
+    screen_to: int
+    screen_from: int | None = None
+    source_id: int
+    session_id: int
+
+
+class NavigationEventPayload(CamelModel):
+    event: str
+    time: int
+    type: str = "NAV"
+    user_id: int
+    params: NavigationEventParams
+
+
+class NavigationPayload(CamelModel):
+    events: list[NavigationEventPayload]