RubigramClient 1.6.4__py3-none-any.whl → 1.6.5__py3-none-any.whl

rubigram/{models.py → types.py}

@@ -1,169 +1,233 @@
-from dataclasses import dataclass, fields, is_dataclass
-from typing import Any, Type, Union, TypeVar, Literal, Optional, get_origin, get_args
+from dataclasses import dataclass, is_dataclass, fields
+from typing import Type, TypeVar, Union, Optional, get_origin, get_args
+from json import dumps
+from enum import Enum
+from . import enums
 import rubigram
-import json
 
 
-T = TypeVar("T", bound="Dict")
+T = TypeVar("T", bound="DataManager")
 
 
 @dataclass
-class Dict:
-    def to_dict(self) -> dict[str, Any]:
+class DataManager:
+    def to_dict(self) -> dict:
         data = {}
         for field in fields(self):
             value = getattr(self, field.name)
-            if is_dataclass(value):
+            if isinstance(value, Enum):
+                data[field.name] = value.value
+            elif is_dataclass(value):
                 data[field.name] = value.to_dict()
             elif isinstance(value, list):
-                data[field.name] = [i.to_dict() if is_dataclass(i)else i for i in value]
+                data[field.name] = [i.to_dict() if is_dataclass(i) else i for i in value]
             else:
                 data[field.name] = value
         return data
 
+
     @classmethod
-    def from_dict(cls: Type[T], data: dict[str, Any]) -> Optional[T]:
-        if data is None:
-            data = {}
+    def from_dict(Class: Type[T], data: dict) -> T:
+        data = data or {}
         init_data = {}
-        for field in fields(cls):
+
+        for field in fields(Class):
             value = data.get(field.name)
             field_type = field.type
             origin = get_origin(field_type)
-            if isinstance(value, dict) and isinstance(field_type, type) and issubclass(field_type, Dict):
+
+            if isinstance(value, dict) and isinstance(field_type, type) and issubclass(field_type, DataManager):
                 init_data[field.name] = field_type.from_dict(value)
+
             elif origin == list:
                 inner_type = get_args(field_type)[0]
-                if isinstance(inner_type, type) and issubclass(inner_type, Dict):
+                if isinstance(inner_type, type) and issubclass(inner_type, DataManager):
                     init_data[field.name] = [inner_type.from_dict(
                         v) if isinstance(v, dict) else v for v in (value or [])]
                 else:
                     init_data[field.name] = value or []
-            elif get_origin(field_type) is Union:
+
+            elif origin == Union:
                 args = get_args(field_type)
                 dict_type = next((a for a in args if isinstance(
-                    a, type) and issubclass(a, Dict)), None)
+                    a, type) and issubclass(a, DataManager)), None)
                 if dict_type and isinstance(value, dict):
                     init_data[field.name] = dict_type.from_dict(value)
                 else:
                     init_data[field.name] = value
+
             else:
                 init_data[field.name] = value
-        return cls(**init_data)
 
-    def to_json(self):
-        data = self.to_dict().copy()
+        return Class(**init_data)
+
+    def json(self):
+        def clear(object):
+            if isinstance(object, dict):
+                return {key: clear(value) for key, value in object.items() if value is not None}
+            elif isinstance(object, list):
+                return [clear(i) for i in object if i is not None]
+            else:
+                return object
+
+        data = self.to_dict()
         data.pop("client", None)
-        return json.dumps(data, ensure_ascii=False, indent=4)
+        clean_data = clear(data)
+        return dumps(clean_data, ensure_ascii=False, indent=4)
 
 
 @dataclass
-class Location(Dict):
-    longitude: Optional[str] = None
-    latitude: Optional[str] = None
+class Chat(DataManager):
+    chat_id: Optional[str] = None
+    chat_type: Optional[enums.ChatType] = None
+    user_id: Optional[str] = None
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+    title: Optional[str] = None
+    username: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class OpenChatData(Dict):
-    object_guid: Optional[str] = None
-    object_type: Optional[Literal["User", "Bot", "Group", "Channel"]] = None
+class File(DataManager):
+    file_id: Optional[str] = None
+    file_name: Optional[str] = None
+    size: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class JoinChannelData(Dict):
-    username: Optional[str] = None
-    ask_join: Optional[bool] = False
+class ForwardedFrom(DataManager):
+    type_from: Optional[enums.ForwardedFrom] = None
+    message_id: Optional[str] = None
+    from_chat_id: Optional[str] = None
+    from_sender_id: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonLink(Dict):
-    type: Optional[Literal["joinchannel", "url"]] = None
-    link_url: Optional[str] = None
-    joinchannel_data: Optional[JoinChannelData] = None
-    open_chat_data: Optional[OpenChatData] = None
+class PaymentStatus(DataManager):
+    payment_id: Optional[str] = None
+    status: Optional[enums.PaymentStatus] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonSelectionItem(Dict):
+class MessageTextUpdate(DataManager):
+    message_id: Optional[str] = None
     text: Optional[str] = None
-    image_url: Optional[str] = None
-    type: Optional[Literal["TextOnly", "TextImgThu", "TextImgBig"]] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonTextbox(Dict):
-    type_line: Optional[Literal["SingleLine", "MultiLine"]] = None
-    type_keypad: Optional[Literal["String", "Number"]] = None
-    place_holder: Optional[str] = None
-    title: Optional[str] = None
-    default_value: Optional[str] = None
+class Bot(DataManager):
+    bot_id: Optional[str] = None
+    bot_title: Optional[str] = None
+    avatar: Optional[File] = None
+    description: Optional[str] = None
+    username: Optional[str] = None
+    start_message: Optional[str] = None
+    share_url: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonLocation(Dict):
-    default_pointer_location: Optional[Location] = None
-    default_map_location: Optional[Location] = None
-    type: Optional[Literal["Picker", "View"]] = None
-    title: Optional[str] = None
-    location_image_url: Optional[str] = None
+class BotCommand(DataManager):
+    command: Optional[str] = None
+    description: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonStringPicker(Dict):
-    items: Optional[list[str]] = None
-    default_value: Optional[str] = None
-    title: Optional[str] = None
+class Sticker(DataManager):
+    sticker_id: Optional[str] = None
+    file: Optional[File] = None
+    emoji_character: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonNumberPicker(Dict):
-    min_value: Optional[str] = None
-    max_value: Optional[str] = None
-    default_value: Optional[str] = None
-    title: Optional[str] = None
+class ContactMessage(DataManager):
+    phone_number: Optional[str] = None
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonCalendar(Dict):
-    default_value: Optional[str] = None
-    type: Optional[Literal["DatePersian", "DateGregorian"]] = None
-    min_year: Optional[str] = None
-    max_year: Optional[str] = None
-    title: Optional[str] = None
+class PollStatus(DataManager):
+    state: Optional[enums.PollStatus] = None
+    selection_index: Optional[int] = None
+    percent_vote_options: Optional[list[int]] = None
+    total_vote: Optional[int] = None
+    show_total_votes: Optional[bool] = None
+
+    def __repr__(self):
+        return self.json()
+
+
+@dataclass
+class Poll(DataManager):
+    question: Optional[str] = None
+    options: Optional[list[str]] = None
+    poll_status: Optional[PollStatus] = None
+
+    def __repr__(self):
+        return self.json()
+
+
+@dataclass
+class Location(DataManager):
+    longitude: Optional[str] = None
+    latitude: Optional[str] = None
+
+    def __repr__(self):
+        return self.json()
+
+
+@dataclass
+class LiveLocation(DataManager):
+    start_time: Optional[str] = None
+    live_period: Optional[int] = None
+    current_location: Optional[Location] = None
+    user_id: Optional[str] = None
+    status: Optional[enums.LiveLocationStatus] = None
+    last_update_time: Optional[str] = None
+
+    def __repr__(self):
+        return self.json()
+
+
+@dataclass
+class ButtonSelectionItem(DataManager):
+    text: Optional[str] = None
+    image_url: Optional[str] = None
+    type: Optional[enums.ButtonSelectionType] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ButtonSelection(Dict):
+class ButtonSelection(DataManager):
     selection_id: Optional[str] = None
     search_type: Optional[str] = None
     get_type: Optional[str] = None
@@ -173,149 +237,152 @@ class ButtonSelection(Dict):
     title: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class Button(Dict):
-    id: Optional[str] = None
-    button_text: Optional[str] = None
-    type: Literal[
-        "Simple", "Selection", "Calendar", "NumberPicker", "StringPicker", "Location", "Payment",
-        "CameraImage", "CameraVideo", "GalleryImage", "GalleryVideo", "File", "Audio", "RecordAudio",
-        "MyPhoneNumber", "MyLocation", "Textbox", "Link", "AskMyPhoneNumber", "AskLocation", "Barcode"
-    ] = "Simple"
-    button_selection: Optional[ButtonSelection] = None
-    button_calendar: Optional[ButtonCalendar] = None
-    button_number_picker: Optional[ButtonNumberPicker] = None
-    button_string_picker: Optional[ButtonStringPicker] = None
-    button_location: Optional[ButtonLocation] = None
-    button_textbox: Optional[ButtonTextbox] = None
-    button_link: Optional[ButtonLink] = None
+class ButtonCalendar(DataManager):
+    default_value: Optional[str] = None
+    type: Optional[enums.ButtonCalendarType] = None
+    min_year: Optional[str] = None
+    max_year: Optional[str] = None
+    title: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class KeypadRow(Dict):
-    buttons: list[Button]
+class ButtonNumberPicker(DataManager):
+    min_value: Optional[str] = None
+    max_value: Optional[str] = None
+    default_value: Optional[str] = None
+    title: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class Keypad(Dict):
-    rows: list[KeypadRow]
-    resize_keyboard: bool = True
-    on_time_keyboard: bool = False
+class ButtonStringPicker(DataManager):
+    items: Optional[list[str]] = None
+    default_value: Optional[str] = None
+    title: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class PollStatus(Dict):
-    state: Optional[Literal["Open", "Closed"]] = None
-    selection_index: Optional[int] = None
-    percent_vote_options: Optional[list[int]] = None
-    total_vote: Optional[int] = None
-    show_total_votes: Optional[bool] = None
+class ButtonTextbox(DataManager):
+    type_line: Optional[enums.ButtonTextboxTypeLine] = None
+    type_keypad: Optional[enums.ButtonTextboxTypeKeypad] = None
+    place_holder: Optional[str] = None
+    title: Optional[str] = None
+    default_value: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class File(Dict):
-    file_id: Optional[str] = None
-    file_name: Optional[str] = None
-    size: Optional[str] = None
+class ButtonLocation(DataManager):
+    default_pointer_location: Optional[Location] = None
+    default_map_location: Optional[Location] = None
+    type: Optional[enums.ButtonLocationType] = None
+    title: Optional[str] = None
+    location_image_url: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class LiveLocation(Dict):
-    start_time: Optional[str] = None
-    live_period: Optional[int] = None
-    current_location: Optional[Location] = None
-    user_id: Optional[str] = None
-    status: Optional[Literal["Stopped", "Live"]] = None
-    last_update_time: Optional[str] = None
+class OpenChatData(DataManager):
+    object_guid: Optional[str] = None
+    object_type: Optional[enums.ChatType] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class Poll(Dict):
-    question: Optional[str] = None
-    options: Optional[list[str]] = None
-    poll_status: Optional[PollStatus] = None
+class JoinChannelData(DataManager):
+    username: Optional[str] = None
+    ask_join: bool = False
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ContactMessage(Dict):
-    phone_number: Optional[str] = None
-    first_name: Optional[str] = None
-    last_name: Optional[str] = None
+class ButtonLink(DataManager):
+    type: Optional[enums.ButtonLinkType] = None
+    link_url: Optional[str] = None
+    joinchannel_data: Optional[JoinChannelData] = None
+    open_chat_data: Optional[OpenChatData] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class Sticker(Dict):
-    sticker_id: Optional[str] = None
-    file: Optional[File] = None
-    emoji_character: Optional[str] = None
+class AuxData(DataManager):
+    start_id: Optional[str] = None
+    button_id: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class ForwardedFrom(Dict):
-    type_from: Optional[Literal["User", "Channel", "Bot"]] = None
-    message_id: Optional[str] = None
-    from_chat_id: Optional[str] = None
-    from_sender_id: Optional[str] = None
+class Button(DataManager):
+    id: Optional[str] = None
+    button_text: Optional[str] = None
+    type: enums.ButtonType = enums.ButtonType.Simple
+    button_selection: Optional[ButtonSelection] = None
+    button_calendar: Optional[ButtonCalendar] = None
+    button_number_picker: Optional[ButtonNumberPicker] = None
+    button_string_picker: Optional[ButtonStringPicker] = None
+    button_location: Optional[ButtonLocation] = None
+    button_textbox: Optional[ButtonTextbox] = None
+    button_link: Optional[ButtonLink] = None
+
+
+@dataclass
+class KeypadRow(DataManager):
+    buttons: list[Button]
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class AuxData(Dict):
-    start_id: Optional[str] = None
-    button_id: Optional[str] = None
+class Keypad(DataManager):
+    rows: list[KeypadRow]
+    resize_keyboard: bool = True
+    on_time_keyboard: bool = False
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class PaymentStatus(Dict):
-    payment_id: Optional[str] = None
-    status: Optional[Literal["Paid", "NotPaid"]] = None
+class MessageKeypadUpdate(DataManager):
+    message_id: Optional[str] = None
+    inline_keypad: Optional[Keypad] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class Message(Dict):
+class Message(DataManager):
     message_id: Optional[str] = None
     text: Optional[str] = None
     time: Optional[str] = None
     is_edited: Optional[bool] = None
-    sender_type: Optional[Literal["User", "Bot"]] = None
+    sender_type: Optional[enums.MessageSender] = None
     sender_id: Optional[str] = None
     aux_data: Optional[AuxData] = None
     file: Optional[File] = None
@@ -329,98 +396,111 @@ class Message(Dict):
     live_location: Optional[LiveLocation] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
 
 @dataclass
-class InlineMessage(Dict):
-    client: Optional["rubigram.Client"] = None
-    sender_id: Optional[str] = None
-    text: Optional[str] = None
+class MessageId(DataManager):
     message_id: Optional[str] = None
+    file_id: Optional[str] = None
     chat_id: Optional[str] = None
-    file: Optional[File] = None
-    location: Optional[Location] = None
-    aux_data: Optional[AuxData] = None
+    client: Optional["rubigram.Client"] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
 
+    async def delete(self):
+        return await self.client.delete_message(self.chat_id, self.message_id)
 
-@dataclass
-class Bot(Dict):
-    bot_id: Optional[str] = None
-    bot_title: Optional[str] = None
-    avatar: Optional[File] = None
-    description: Optional[str] = None
-    username: Optional[str] = None
-    start_message: Optional[str] = None
-    share_url: Optional[str] = None
+    async def edit_text(self, text: str):
+        return await self.client.edit_message_text(self.chat_id, self.message_id, text)
 
-    def __repr__(self):
-        return self.to_json()
+    async def edit_message_keypad(self, keypad: Keypad):
+        return await self.client.edit_message_keypad(self.chat_id, self.message_id, keypad)
 
-
-@dataclass
-class Chat(Dict):
-    chat_id: Optional[str] = None
-    chat_type: Optional[Literal["User", "Bot", "Group", "Channel"]] = None
-    user_id: Optional[str] = None
-    first_name: Optional[str] = None
-    last_name: Optional[str] = None
-    title: Optional[str] = None
-    username: Optional[str] = None
-
-    def __repr__(self):
-        return self.to_json()
+    async def forward(self, chat_id: str):
+        return await self.client.forward_message(self.chat_id, self.message_id, chat_id)
 
 
 @dataclass
-class MessageId(Dict):
-    message_id: Optional[str] = None
-
-    def __repr__(self):
-        return self.to_json()
-
-
-@dataclass
-class Update(Dict):
+class Update(DataManager):
     client: Optional["rubigram.Client"] = None
-    type: Optional[Literal["NewMessage", "UpdatedMessage", "RemovedMessage", "StartedBot", "StoppedBot", "UpdatedPayment"]] = None
+    type: Optional[enums.UpdateType] = None
     chat_id: Optional[str] = None
     removed_message_id: Optional[str] = None
     new_message: Optional[Message] = None
     updated_message: Optional[Message] = None
     updated_payment: Optional[PaymentStatus] = None
 
-    def __str__(self):
-        return self.to_json()
-
-    async def send_text(self, text: str,) -> "MessageId":
-        return await self.client.send_message(self.chat_id, text)
+    def __repr__(self):
+        return self.json()
 
-    async def download(self, name: str):
-        return await self.client.download_file(self.new_message.file.file_id, name)
+    async def download(self, file_name: str):
+        return await self.client.download_file(self.new_message.file.file_id, file_name)
 
     async def reply(
         self,
         text: str,
         chat_keypad: Keypad = None,
         inline_keypad: Keypad = None,
-        chat_keypad_type: Literal["New", "Remove"] = None,
+        chat_keypad_type: Optional[enums.ChatKeypadType] = None,
         disable_notification: bool = None,
     ) -> "MessageId":
         return await self.client.send_message(self.chat_id, text, chat_keypad, inline_keypad, chat_keypad_type, disable_notification, self.new_message.message_id)
+    
+    async def reply_poll(
+        self,
+        question: str,
+        options: list[str],
+        chat_keypad: Keypad = None,
+        inline_keypad: Keypad = None,
+        chat_keypad_type: Optional[enums.ChatKeypadType] = None,
+        disable_notification: bool = False
+    ) -> "MessageId":
+        return await self.client.send_poll(self.chat_id, question, options, chat_keypad, inline_keypad, chat_keypad_type, disable_notification, self.new_message.message_id)
+    
+    async def reply_location(
+        self,
+        latitude: str,
+        longitude: str,
+        chat_keypad: Keypad = None,
+        inline_keypad: Keypad = None,
+        chat_keypad_type: Optional[enums.ChatKeypadType] = None,
+        disable_notification: bool = False,
+    ) -> "MessageId":
+        return await self.client.send_location(self.chat_id, latitude, longitude, chat_keypad, inline_keypad, chat_keypad_type, disable_notification, self.new_message.message_id)
+    
+    async def reply_contact(
+        self,
+        first_name: str,
+        last_name: str,
+        phone_number: str,
+        chat_keypad: Keypad = None,
+        inline_keypad: Keypad = None,
+        chat_keypad_type: Optional[enums.ChatKeypadType] = None,
+        disable_notification: bool = False,
+    ) -> "MessageId":
+        return await self.client.send_location(self.chat_id, first_name, last_name, phone_number, chat_keypad, inline_keypad, chat_keypad_type, disable_notification, self.new_message.message_id)
+    
+    async def reply_sticker(
+        self,
+        sticker_id: str,
+        chat_keypad: Keypad = None,
+        inline_keypad: Keypad = None,
+        chat_keypad_type: Optional[enums.ChatKeypadType] = None,
+        disable_notification: bool = False,
+    ) -> "MessageId":
+        return await self.client.send_message(self.chat_id, sticker_id, chat_keypad, inline_keypad, chat_keypad_type, disable_notification, self.new_message.message_id)
 
     async def reply_file(
         self,
         file: str,
         file_name: str,
         caption: str = None,
-        type: Literal["File", "Image", "Voice", "Music", "Gif", "Video"] = "File",
+        type: enums.FileType = enums.FileType.File,
         chat_keypad: Keypad = None,
         inline_keypad: Keypad = None,
-        chat_keypad_type: Literal["New", "Remove"] = None,
+        chat_keypad_type: Optional[enums.ChatKeypadType] = None,
         disable_notification: bool = False,
     ) -> "MessageId":
         return await self.client.send_file(self.chat_id, file, file_name, caption, type, chat_keypad, inline_keypad, chat_keypad_type, disable_notification, self.new_message.message_id)
@@ -445,9 +525,24 @@ class Update(Dict):
 
 
 @dataclass
-class Updates(Dict):
+class Updates(DataManager):
     updates: list[Update] = None
     next_offset_id: Optional[str] = None
 
     def __repr__(self):
-        return self.to_json()
+        return self.json()
+
+
+@dataclass
+class InlineMessage(DataManager):
+    client: Optional["rubigram.Client"] = None
+    sender_id: Optional[str] = None
+    text: Optional[str] = None
+    message_id: Optional[str] = None
+    chat_id: Optional[str] = None
+    file: Optional[File] = None
+    location: Optional[Location] = None
+    aux_data: Optional[AuxData] = None
+
+    def __repr__(self):
+        return self.json()