RubigramClient 1.7.1__py3-none-any.whl → 1.7.3__py3-none-any.whl

rubigram/types/object.py

@@ -0,0 +1,112 @@
+import sys
+from typing import Union, TypeVar, Type, get_origin, get_args, get_type_hints
+from dataclasses import dataclass, fields
+from json import dumps
+
+T = TypeVar("T", bound="Object")
+
+
+@dataclass
+class Object:
+
+    def asdict(self):
+        data = {}
+        for f in fields(self):
+            value = getattr(self, f.name)
+            if isinstance(value, Object):
+                inner = value.asdict()
+                data[f.name] = {"_": value.__class__.__name__, **inner}
+            elif isinstance(value, list):
+                data[f.name] = [
+                    {"_": v.__class__.__name__, **v.asdict()} if isinstance(v, Object) else v for v in value
+                ]
+            else:
+                data[f.name] = value
+        return data
+
+    def jsonify(self, exclude_none: bool = False) -> str:
+        def clear(obj):
+            if isinstance(obj, dict):
+                return {k: clear(v) for k, v in obj.items() if v is not None}
+            elif isinstance(obj, list):
+                return [clear(i) for i in obj if i is not None]
+            else:
+                return obj
+
+        data = self.asdict()
+        data.pop("client", None)
+        if exclude_none:
+            data = clear(data)
+        return dumps(
+            {"_": self.__class__.__name__, **data},
+            ensure_ascii=False,
+            indent=4
+        )
+
+    @classmethod
+    def parse(cls: Type[T], data: dict) -> T:
+        data = data or {}
+        init_data = {}
+
+        try:
+            module = sys.modules[cls.__module__]
+            type_hints = get_type_hints(
+                cls, globalns=module.__dict__, localns=None)
+        except Exception:
+            type_hints = {}
+
+        for field in fields(cls):
+            raw_value = data.get(field.name)
+            if isinstance(raw_value, dict) and "_" in raw_value:
+                raw_value = {k: v for k, v in raw_value.items() if k != "_"}
+
+            field_type = type_hints.get(field.name, field.type)
+            origin = get_origin(field_type)
+
+            if origin is Union:
+                args = get_args(field_type)
+                non_none_args = [arg for arg in args if arg is not type(None)]
+                if len(non_none_args) == 1:
+                    field_type = non_none_args[0]
+                    origin = get_origin(field_type)
+                else:
+                    pass
+
+            if isinstance(raw_value, dict) and isinstance(field_type, type) and issubclass(field_type, Object):
+                init_data[field.name] = field_type.parse(raw_value)
+
+            elif origin == list:
+                inner_type = get_args(field_type)[0]
+                inner_origin = get_origin(inner_type)
+                if inner_origin is Union:
+                    inner_args = [a for a in get_args(
+                        inner_type) if a is not type(None)]
+                    inner_type = inner_args[0] if inner_args else inner_type
+
+                if isinstance(raw_value, list):
+                    if isinstance(inner_type, type) and issubclass(inner_type, Object):
+                        init_data[field.name] = [
+                            inner_type.parse(v) if isinstance(v, dict) else v for v in raw_value
+                        ]
+                    else:
+                        init_data[field.name] = raw_value
+                else:
+                    init_data[field.name] = [
+                    ] if raw_value is None else raw_value
+
+            elif origin is Union:
+                args = get_args(field_type)
+                obj_arg = next((arg for arg in args if isinstance(
+                    arg, type) and issubclass(arg, Object)), None)
+                if obj_arg and isinstance(raw_value, dict):
+                    init_data[field.name] = obj_arg.parse(raw_value)
+                else:
+                    init_data[field.name] = raw_value
+
+            else:
+                init_data[field.name] = raw_value
+
+        return cls(**init_data)
+
+    def __str__(self):
+        return self.jsonify(True)

rubigram/types/types.py

@@ -0,0 +1,211 @@
+from typing import Optional, Union
+from dataclasses import dataclass, field
+from .. import enums
+from .object import Object
+
+
+@dataclass
+class Chat(Object):
+    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
+
+
+@dataclass
+class File(Object):
+    file_id: Optional[str] = None
+    file_name: Optional[str] = None
+    size: Optional[int] = None
+
+
+@dataclass
+class ForwardedFrom(Object):
+    type_from: Optional[enums.ForwardedFrom] = None
+    message_id: Optional[str] = None
+    from_chat_id: Optional[str] = None
+    from_sender_id: Optional[str] = None
+
+
+@dataclass
+class PaymentStatus(Object):
+    payment_id: Optional[str] = None
+    status: Optional[Union[str, enums.PaymentStatus]] = None
+
+
+@dataclass
+class Bot(Object):
+    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
+
+
+@dataclass
+class BotCommand(Object):
+    command: Optional[str] = None
+    description: Optional[str] = None
+
+
+@dataclass
+class Sticker(Object):
+    sticker_id: Optional[str] = None
+    file: Optional[File] = None
+    emoji_character: Optional[str] = None
+
+
+@dataclass
+class ContactMessage(Object):
+    phone_number: Optional[str] = None
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+
+
+@dataclass
+class PollStatus(Object):
+    state: Optional[Union[str, 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
+
+
+@dataclass
+class Poll(Object):
+    question: Optional[str] = None
+    options: Optional[list[str]] = None
+    poll_status: Optional[PollStatus] = None
+
+
+@dataclass
+class Location(Object):
+    longitude: Optional[str] = None
+    latitude: Optional[str] = None
+
+
+@dataclass
+class LiveLocation(Object):
+    start_time: Optional[str] = None
+    live_period: Optional[int] = None
+    current_location: Optional[Location] = None
+    user_id: Optional[str] = None
+    status: Optional[Union[str, enums.LiveLocationStatus]] = None
+    last_update_time: Optional[str] = None
+
+
+@dataclass
+class ButtonSelectionItem(Object):
+    text: Optional[str] = None
+    image_url: Optional[str] = None
+    type: Optional[Union[str, enums.ButtonSelectionType]] = None
+
+
+@dataclass
+class ButtonSelection(Object):
+    selection_id: Optional[str] = None
+    search_type: Optional[str] = None
+    get_type: Optional[str] = None
+    items: Optional[list[ButtonSelectionItem]] = None
+    is_multi_selection: Optional[bool] = None
+    columns_count: Optional[str] = None
+    title: Optional[str] = None
+
+
+@dataclass
+class ButtonCalendar(Object):
+    default_value: Optional[str] = None
+    type: Optional[Union[str, enums.ButtonCalendarType]] = None
+    min_year: Optional[str] = None
+    max_year: Optional[str] = None
+    title: Optional[str] = None
+
+
+@dataclass
+class ButtonNumberPicker(Object):
+    min_value: Optional[str] = None
+    max_value: Optional[str] = None
+    default_value: Optional[str] = None
+    title: Optional[str] = None
+
+
+@dataclass
+class ButtonStringPicker(Object):
+    items: Optional[list[str]] = None
+    default_value: Optional[str] = None
+    title: Optional[str] = None
+
+
+@dataclass
+class ButtonTextbox(Object):
+    type_line: Optional[Union[str, enums.ButtonTextboxTypeLine]] = None
+    type_keypad: Optional[Union[str, enums.ButtonTextboxTypeKeypad]] = None
+    place_holder: Optional[str] = None
+    title: Optional[str] = None
+    default_value: Optional[str] = None
+
+
+@dataclass
+class ButtonLocation(Object):
+    default_pointer_location: Optional[Location] = None
+    default_map_location: Optional[Location] = None
+    type: Optional[Union[str, enums.ButtonLocationType]] = None
+    title: Optional[str] = None
+    location_image_url: Optional[str] = None
+
+
+@dataclass
+class OpenChatData(Object):
+    Object_guid: Optional[str] = None
+    Object_type: Optional[Union[str, enums.ChatType]] = None
+
+
+@dataclass
+class JoinChannelData(Object):
+    username: Optional[str] = None
+    ask_join: bool = False
+
+
+@dataclass
+class ButtonLink(Object):
+    type: Optional[Union[str, enums.ButtonLinkType]] = None
+    link_url: Optional[str] = None
+    joinchannel_data: Optional[JoinChannelData] = None
+    open_chat_data: Optional[OpenChatData] = None
+
+
+@dataclass
+class AuxData(Object):
+    start_id: Optional[str] = None
+    button_id: Optional[str] = None
+
+
+@dataclass
+class Button(Object):
+    id: Optional[str] = None
+    button_text: Optional[str] = None
+    type: Optional[Union[str, 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(Object):
+    buttons: list[Button] = field(default_factory=list)
+
+
+@dataclass
+class Keypad(Object):
+    rows: list[KeypadRow] = field(default_factory=list)
+    resize_keyboard: bool = True
+    on_time_keyboard: bool = False