python-trueconf-bot 1.0.0__py3-none-any.whl

trueconf/methods/__init__.py

@@ -0,0 +1,59 @@
+from .add_participant_to_chat import AddChatParticipant
+from .auth import AuthMethod
+from .base import MessageIdCounter
+from .base import ReturnResolver
+from trueconf.methods.base import TrueConfMethod
+from .create_channel import CreateChannel
+from .create_group_chat import CreateGroupChat
+from .create_p2p_chat import CreateP2PChat
+from .edit_message import EditMessage
+from .edit_survey import EditSurvey
+from .forward_message import ForwardMessage
+from .get_chat_by_id import GetChatByID
+from .get_chat_history import GetChatHistory
+from .get_chat_participants import GetChatParticipants
+from .get_chats import GetChats
+from .get_file_info import GetFileInfo
+from .get_message_by_id import GetMessageById
+from .get_user_display_name import GetUserDisplayName
+from .has_chat_participant import HasChatParticipant
+from .remove_chat import RemoveChat
+from .remove_message import RemoveMessage
+from .remove_participant_from_chat import RemoveChatParticipant
+from .send_file import SendFile
+from .send_message import SendMessage
+from .send_survey import SendSurvey
+from .subscribe_file_progress import SubscribeFileProgress
+from .unsubscribe_file_progress import UnsubscribeFileProgress
+from .upload_file import UploadFile
+
+__all__ = [
+    'CreateGroupChat',
+    'AuthMethod',
+    'EditSurvey',
+    'GetFileInfo',
+    'RemoveChatParticipant',
+    'GetUserDisplayName',
+    'ForwardMessage',
+    'SendMessage',
+    'CreateChannel',
+    'EditMessage',
+    'GetChatHistory',
+    'GetChatParticipants',
+    'HasChatParticipant',
+    'UnsubscribeFileProgress',
+    'SendSurvey',
+    'RemoveMessage',
+    'GetChats',
+    'GetMessageById',
+    'SubscribeFileProgress',
+    'SendFile',
+    'UploadFile',
+    'CreateP2PChat',
+    'RemoveChat',
+    'ReturnResolver',
+    'MessageIdCounter',
+    'TrueConfMethod',
+    'GetChatByID',
+    'AddChatParticipant'
+]

trueconf/methods/add_participant_to_chat.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.add_chat_participant_response import AddChatParticipantResponse
+
+
+@dataclass
+class AddChatParticipant(TrueConfMethod[AddChatParticipantResponse]):
+    __api_method__ = "addChatParticipant"
+    __returning__ = AddChatParticipantResponse
+
+    chat_id: str
+    user_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "userId": self.user_id
+        }

trueconf/methods/auth.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.auth_response_payload import AuthResponsePayload
+
+
+@dataclass
+class AuthMethod(TrueConfMethod[AuthResponsePayload]):
+    __api_method__ = "auth"
+    __returning__ = AuthResponsePayload
+
+    token: str
+    tokenType: str = "JWT"
+    receive_unread_messages: bool = False
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self) -> dict:
+        return {
+            "token": self.token,
+            "tokenType": self.tokenType,
+            "receiveUnread": self.receive_unread_messages
+
+        }

trueconf/methods/base.py

@@ -0,0 +1,107 @@
+from __future__ import annotations
+import logging
+from asyncio import get_running_loop
+from abc import ABC, abstractmethod
+from typing import TypeVar, Generic, TYPE_CHECKING, ClassVar, Protocol, runtime_checkable
+from trueconf.exceptions import ApiError as ApiErrorException
+from trueconf.types.responses.api_error import ApiError
+
+logger = logging.getLogger("chat_bot")
+
+T = TypeVar("T")
+
+
+@runtime_checkable
+class ReturnResolver(Protocol[T]):
+    @staticmethod
+    def parse(resp: dict) -> T: ...
+
+
+class MessageIdCounter:
+    _counter = 0
+
+    @classmethod
+    def get_next_id(cls) -> int:
+        cls._counter += 1
+        return cls._counter
+
+
+class TrueConfMethod(ABC, Generic[T]):
+    def __init__(self):
+        self.id = MessageIdCounter.get_next_id()
+
+    if TYPE_CHECKING:
+        __api_method__: ClassVar[str]
+        __returning__: ClassVar[type[T]]
+    else:
+        @property
+        @abstractmethod
+        def __api_method__(self) -> str:
+            ...
+
+        @property
+        @abstractmethod
+        def __returning__(self) -> type[T]:
+            ...
+
+    @abstractmethod
+    def payload(self) -> dict:
+        ...
+
+    def _parse_return(self, resp: dict) -> T:
+        ret = self.__returning__
+
+        payload = (resp.get("payload") or {})
+
+        # 2) Ошибка API имеет приоритет
+        if isinstance(payload, dict) and payload.get("errorCode", 0) != 0:
+            return ApiError.from_dict(payload)  # type: ignore[return-value] # type: ignore[return-value]
+
+        # 1) Кастомный парсер (если класс его предоставляет)
+        if hasattr(ret, "parse"):
+            return ret.parse(resp)  # type: ignore[return-value]
+
+        # 3) Если просят dict — отдать как есть
+        if ret is dict:
+            return payload  # type: ignore[return-value]
+
+        # 4) Поддержка mashumaro: есть from_dict -> используем alias-ы
+        if hasattr(ret, "from_dict"):
+            if isinstance(payload, list):
+                return ret.from_dict({"chats": payload})  # type: ignore[return-value]
+            return ret.from_dict(payload)  # type: ignore[return-value]
+
+        # 5) Фоллбек: прямой конструктор (без alias-ов)
+        return ret(**payload)  # type: ignore[misc]
+
+    async def __call__(self, bot: "ChatBot") -> T:
+        loop = get_running_loop()
+        future = loop.create_future()
+        bot._register_future(self.id, future)
+
+        try:
+            message = {
+                "type": 1,
+                "id": self.id,
+                "method": self.__api_method__,
+                "payload": self.payload(),
+            }
+
+        except AttributeError:
+            raise RuntimeError(
+                f"{type(self).__name__} must define __api_method__ and __returning__"
+            )
+
+        logger.debug(f"📤 Sending message: {message}")
+
+        await bot._send_ws_payload(message)
+
+        data = await future
+        logger.debug(f"✅ Received response for {self.__api_method__}: {data}")
+
+        result = self._parse_return(data)
+
+        if isinstance(result, ApiError):
+            raise ApiErrorException(str(result))
+
+        return result

trueconf/methods/create_channel.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.create_channel_response import CreateChannelResponse
+
+
+@dataclass
+class CreateChannel(TrueConfMethod[CreateChannelResponse]):
+    __api_method__ = "createChannel"
+    __returning__ = CreateChannelResponse
+
+    title: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "title": self.title,
+        }

trueconf/methods/create_group_chat.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.create_group_chat_response import CreateGroupChatResponse
+
+
+@dataclass
+class CreateGroupChat(TrueConfMethod[CreateGroupChatResponse]):
+    __api_method__ = "createGroupChat"
+    __returning__ = CreateGroupChatResponse
+
+    title: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "title": self.title,
+        }

trueconf/methods/create_p2p_chat.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.create_p2p_chat_response import CreateP2PChatResponse
+
+
+@dataclass
+class CreateP2PChat(TrueConfMethod[CreateP2PChatResponse]):
+    __api_method__ = "createP2PChat"
+    __returning__ = CreateP2PChatResponse
+
+    user_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "userId": self.user_id,
+        }

trueconf/methods/edit_message.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.edit_message_response import EditMessageResponse
+
+
+@dataclass
+class EditMessage(TrueConfMethod[EditMessageResponse]):
+    __api_method__ = "editMessage"
+    __returning__ = EditMessageResponse
+    message_id: str
+    text: str
+    parse_mode: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "messageId": self.message_id,
+            "content": {
+                "text": self.text,
+                "parseMode": self.parse_mode,
+            },
+        }

trueconf/methods/edit_survey.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.edit_survey_response import EditSurveyResponse
+
+
+@dataclass
+class EditSurvey(TrueConfMethod[EditSurveyResponse]):
+    __api_method__ = "editSurvey"
+    __returning__ = EditSurveyResponse
+
+    message_id: str
+    server: str
+    path: str
+    title: str
+    description: str
+    button_text: str = "{{Go to survey}}"
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "messageId": self.message_id,
+                "content": {
+                    "path": self.path,
+                    "title": self.title,
+                    "description": self.description,
+                    "buttonText": self.button_text,
+                    "alt": f"📊 <a href='https://{self.server}/webtools/survey?id={self.path}&error=autologin_not_supported'>{self.title}</a>"
+                }
+        }

trueconf/methods/forward_message.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.forward_message_response import ForwardMessageResponse
+
+
+@dataclass
+class ForwardMessage(TrueConfMethod[ForwardMessageResponse]):
+    __api_method__ = "forwardMessage"
+    __returning__ = ForwardMessageResponse
+
+    message_id: str
+    chat_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "messageId": self.message_id,
+            "chatId": self.chat_id,
+        }

trueconf/methods/get_chat_by_id.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_chat_by_id_response import GetChatByIdResponse
+
+
+@dataclass
+class GetChatByID(TrueConfMethod[GetChatByIdResponse]):
+    __api_method__ = "getChatByID"
+    __returning__ = GetChatByIdResponse
+
+    chat_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id
+        }

trueconf/methods/get_chat_history.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_chat_history_response import GetChatHistoryResponse
+
+
+@dataclass
+class GetChatHistory(TrueConfMethod[GetChatHistoryResponse]):
+    __api_method__ = "getChatHistory"
+    __returning__ = GetChatHistoryResponse
+
+    chat_id: str
+    count: int
+    from_message_id: str | None = None
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "count": self.count,
+            "fromMessageId": self.from_message_id,
+        }

trueconf/methods/get_chat_participants.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_chat_participants_response import GetChatParticipantsResponse
+
+
+@dataclass
+class GetChatParticipants(TrueConfMethod[GetChatParticipantsResponse]):
+    __api_method__ = "getChatParticipants"
+    __returning__ = GetChatParticipantsResponse
+
+    chat_id: str
+    page_size: int
+    page_number: int
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "pageSize": self.page_size,
+            "pageNumber": self.page_number,
+        }

trueconf/methods/get_chats.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_chats_response import GetChatsResponse
+
+
+@dataclass
+class GetChats(TrueConfMethod[GetChatsResponse]):
+    __api_method__ = "getChats"
+    __returning__ = GetChatsResponse
+
+    count: int
+    page: int
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "count": self.count,
+            "page": self.page,
+        }

trueconf/methods/get_file_info.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_file_info_response import GetFileInfoResponse
+
+
+@dataclass
+class GetFileInfo(TrueConfMethod[GetFileInfoResponse]):
+    __api_method__ = "getFileInfo"
+    __returning__ = GetFileInfoResponse
+
+    file_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "fileId": self.file_id
+        }

trueconf/methods/get_message_by_id.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_message_by_id_response import GetMessageByIdResponse
+
+
+@dataclass
+class GetMessageById(TrueConfMethod[GetMessageByIdResponse]):
+    __api_method__ = "getMessageById"
+    __returning__ = GetMessageByIdResponse
+    message_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "messageId": self.message_id
+        }

trueconf/methods/get_user_display_name.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.get_user_display_name_response import GetUserDisplayNameResponse
+
+
+@dataclass
+class GetUserDisplayName(TrueConfMethod[GetUserDisplayNameResponse]):
+    __api_method__ = "getUserDisplayName"
+    __returning__ = GetUserDisplayNameResponse
+
+    user_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "userId": self.user_id,
+        }

trueconf/methods/has_chat_participant.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.has_chat_participant_response import HasChatParticipantResponse
+
+
+@dataclass
+class HasChatParticipant(TrueConfMethod[HasChatParticipantResponse]):
+    __api_method__ = "hasChatParticipant"
+    __returning__ = HasChatParticipantResponse
+
+    chat_id: str
+    user_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "userId": self.user_id
+        }

trueconf/methods/remove_chat.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.remove_chat_response import RemoveChatResponse
+
+
+@dataclass
+class RemoveChat(TrueConfMethod[RemoveChatResponse]):
+    __api_method__ = "removeChat"
+    __returning__ = RemoveChatResponse
+
+    chat_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+        }

trueconf/methods/remove_message.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.remove_message_response import RemoveMessageResponse
+
+
+@dataclass
+class RemoveMessage(TrueConfMethod[RemoveMessageResponse]):
+    __api_method__ = "removeMessage"
+    __returning__ = RemoveMessageResponse
+
+    message_id: str
+    for_all: bool
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "messageId": self.message_id,
+            "forAll": self.for_all
+        }

trueconf/methods/remove_participant_from_chat.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.remove_chat_participant_response import RemoveChatParticipantResponse
+
+
+@dataclass
+class RemoveChatParticipant(TrueConfMethod[RemoveChatParticipantResponse]):
+    __api_method__ = "removeChatParticipant"
+    __returning__ = RemoveChatParticipantResponse
+
+    chat_id: str
+    user_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "userId": self.user_id
+        }

trueconf/methods/send_file.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+from __future__ import annotations
+from dataclasses import dataclass
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.send_file_response import SendFileResponse
+
+
+@dataclass
+class SendFile(TrueConfMethod[SendFileResponse]):
+    __api_method__ = "sendFile"
+    __returning__ = SendFileResponse
+
+    chat_id: str
+    temporal_file_id: str
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "content": {
+                "temporalFileId": self.temporal_file_id,
+            }
+        }

trueconf/methods/send_message.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+from __future__ import annotations
+from dataclasses import dataclass
+from typing import Optional
+from trueconf.methods.base import TrueConfMethod
+from trueconf.types.responses.send_message_response import SendMessageResponse
+
+
+@dataclass
+class SendMessage(TrueConfMethod[SendMessageResponse]):
+    __api_method__ = "sendMessage"
+    __returning__ = SendMessageResponse
+    chat_id: str
+    text: str
+    parse_mode: str
+    reply_message_id: Optional[str] = None
+
+    def __post_init__(self):
+        super().__init__()
+
+    def payload(self):
+        return {
+            "chatId": self.chat_id,
+            "replyMessageId": self.reply_message_id,
+            "content": {
+                "text": self.text,
+                "parseMode": self.parse_mode,
+            },
+        }