PySPlusthon 0.5.3__py3-none-any.whl

PySPlusthon/objects/chat.py

@@ -0,0 +1,118 @@
+from typing import Union, BinaryIO
+
+from PySPlusthon import enums, objects
+from . import Object
+from ..sync_support import add_sync_support_to_object
+
+
+@add_sync_support_to_object
+class Chat(Object):
+
+    @classmethod
+    def from_protobuf(cls, protobuf_data):
+        return cls(
+            id=f"{protobuf_data.id}|{protobuf_data.type}"
+        )
+
+    def __init__(
+            self,
+            id: int = None,
+            type: "enums.ChatType" = None,
+            title: str = None,
+            username: str = None,
+            first_name: str = None,
+            last_name: str = None,
+            all_members_are_administrators: bool = None,
+            bio: str = None,
+            description: str = None,
+            invite_link: str = None,
+            linked_chat_id: int = None,
+            pinned_message: "objects.Message" = None,
+            sticker_set_name: str = None,
+            can_set_sticker_set: bool = None,
+            photo: "objects.ChatPhoto" = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: int = id
+        self.type: "enums.ChatType" = type
+        self.title: str = title
+        self.username: str = username
+        self.first_name: str = first_name
+        self.last_name: str = last_name
+        self.all_members_are_administrators: bool = all_members_are_administrators
+        self.bio: str = bio
+        self.description: str = description
+        self.invite_link: str = invite_link
+        self.linked_chat_id: int = linked_chat_id
+        self.pinned_message: "objects.Message" = pinned_message
+        self.sticker_set_name: str = sticker_set_name
+        self.can_set_sticker_set: bool = can_set_sticker_set
+        self.photo: "objects.ChatPhoto" = photo
+
+    def __eq__(self, other):
+        return self.id == other.id
+
+    def mention(self, text=None):
+        text = text or self.full_name
+        return f"[{text}](uid:{self.id})"
+
+    @property
+    def link(self):
+        if self.invite_link:
+            return f"https://{self.invite_link}"
+        if self.username:
+            return f"{self.client.http_connection.short_url}/{self.username}"
+
+    @property
+    def full_name(self):
+        if self.first_name and self.last_name:
+            return f"{self.first_name} {self.last_name}"
+        if self.first_name:
+            return self.first_name
+        if self.title:
+            return self.title
+        return ""
+
+    async def ban_member(self, user_id: Union[int, str]):
+        return await self.client.ban_chat_member(self.id, user_id)
+
+    async def delete_photo(self):
+        return await self.client.delete_chat_photo(self.id)
+
+    async def get(self):
+        return await self.client.get_chat(self.id)
+
+    async def get_administrators(self):
+        return await self.client.get_chat_administrators(self.id)
+
+    async def get_member(self, user_id: Union[int, str]):
+        return await self.client.get_chat_member(self.id, user_id)
+
+    async def get_members_count(self):
+        return await self.client.get_chat_members_count(self.id)
+
+    async def leave(self):
+        return await self.client.leave_chat(self.id)
+
+    async def pin_message(self, message_id: int):
+        return await self.client.pin_chat_message(self.id, message_id)
+
+    async def promote_member(self, user_id: Union[int, str], *args, **kwargs):
+        return await self.client.promote_chat_member(self.id, user_id, *args, **kwargs)
+
+
+    async def set_description(self, description: str):
+        return await self.client.set_chat_description(self.id, description)
+
+    async def set_photo(self, photo: Union[str, bytes, BinaryIO, "objects.InputMedia"]):
+        return await self.client.set_chat_photo(self.id, photo)
+
+    async def set_title(self, title: str):
+        return await self.client.set_chat_title(self.id, title)
+
+    async def unban_member(self, user_id: Union[int, str], only_if_banned: bool = True):
+        return await self.client.unban_chat_member(self.id, user_id, only_if_banned)
+
+    async def unpin_message(self, message_id: int = None):
+        return await self.client.unpin_chat_message(self.id, message_id)

PySPlusthon/objects/chat_member.py

@@ -0,0 +1,53 @@
+from . import Object
+from PySPlusthon import enums, objects
+
+
+class ChatMember(Object):
+
+    def __init__(
+            self,
+            is_member: bool = None,
+            can_send_media: bool = None,
+            can_send_gif_stickers: bool = None,
+            can_forward_message_from: bool = None,
+            can_send_gift_packet: bool = None,
+            can_send_link_message: bool = None,
+            can_send_forwarded_message: bool = None,
+            can_kick_user: bool = None,
+            can_send_message: bool = None,
+            can_see_members: bool = None,
+            can_add_story: bool = None,
+            user: "objects.User" = None,
+            status: "enums.ChatMemberStatus" = None,
+            can_change_info: bool = None,
+            can_delete_messages: bool = None,
+            can_invite_users: bool = None,
+            can_restrict_members: bool = None,
+            can_pin_messages: bool = None,
+            can_promote_members: bool = None,
+            can_send_messages: bool = None,
+            can_send_media_messages: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.is_member: bool = is_member
+        self.can_send_media: bool = can_send_media
+        self.can_send_gif_stickers: bool = can_send_gif_stickers
+        self.can_forward_message_from: bool = can_forward_message_from
+        self.can_send_gift_packet: bool = can_send_gift_packet
+        self.can_send_link_message: bool = can_send_link_message
+        self.can_send_forwarded_message: bool = can_send_forwarded_message
+        self.can_kick_user: bool = can_kick_user
+        self.can_send_message: bool = can_send_message
+        self.can_see_members: bool = can_see_members
+        self.can_add_story: bool = can_add_story
+        self.user: "objects.User" = user
+        self.status: "enums.ChatMemberStatus" = status
+        self.can_change_info: bool = can_change_info
+        self.can_delete_messages: bool = can_delete_messages
+        self.can_invite_users: bool = can_invite_users
+        self.can_restrict_members: bool = can_restrict_members
+        self.can_pin_messages: bool = can_pin_messages
+        self.can_promote_members: bool = can_promote_members
+        self.can_send_messages: bool = can_send_messages
+        self.can_send_media_messages: bool = can_send_media_messages

PySPlusthon/objects/chat_permissions.py

@@ -0,0 +1,26 @@
+from . import Object
+
+
+class ChatPermissions(Object):
+
+    def __init__(
+            self,
+            can_send_messages: bool = None,
+            can_send_audios: bool = None,
+            can_send_documents: bool = None,
+            can_send_photos: bool = None,
+            can_send_videos: bool = None,
+            can_change_info: bool = None,
+            can_invite_users: bool = None,
+            can_pin_messages: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.can_send_messages: bool = can_send_messages
+        self.can_send_audios: bool = can_send_audios
+        self.can_send_documents: bool = can_send_documents
+        self.can_send_photos: bool = can_send_photos
+        self.can_send_videos: bool = can_send_videos
+        self.can_change_info: bool = can_change_info
+        self.can_invite_users: bool = can_invite_users
+        self.can_pin_messages: bool = can_pin_messages

PySPlusthon/objects/chat_photo.py

@@ -0,0 +1,24 @@
+from . import Object
+
+
+class ChatPhoto(Object):
+    attribute_names = [
+        ("small_id", "small_file_id"),
+        ("small_unique_id", "small_file_unique_id"),
+        ("big_id", "big_file_id"),
+        ("big_unique_id", "big_file_unique_id")
+    ]
+
+    def __init__(
+            self,
+            small_id: str = None,
+            small_unique_id: str = None,
+            big_id: str = None,
+            big_unique_id: str = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.small_id: str = small_id
+        self.small_unique_id: str = small_unique_id
+        self.big_id: str = big_id
+        self.big_unique_id: str = big_unique_id

PySPlusthon/objects/contact.py

@@ -0,0 +1,18 @@
+from . import Object
+
+
+class Contact(Object):
+
+    def __init__(
+            self,
+            phone_number: str = None,
+            first_name: str = None,
+            last_name: str = None,
+            user_id: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.phone_number: str = phone_number
+        self.first_name: str = first_name
+        self.last_name: str = last_name
+        self.user_id: int = user_id

PySPlusthon/objects/date.py

@@ -0,0 +1,35 @@
+from datetime import datetime
+
+from . import Object
+
+
+class Date(Object, datetime):
+
+    @classmethod
+    def wrap(cls, raw_object):
+        date_time = datetime.fromtimestamp(raw_object)
+        return super().__new__(cls, *date_time.timetuple()[:6])
+
+    @classmethod
+    def from_protobuf(cls, timestamp_ms: int):
+        return cls.wrap(timestamp_ms // 1000)
+
+    def __new__(
+            cls,
+            *args,
+            **kwargs
+    ):
+        return super().__new__(cls, *args, *kwargs)
+
+    def __init__(
+            self,
+            *args,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+
+    def __repr__(self):
+        return str(self)
+
+    def unwrap(self):
+        return int(super().timestamp())

PySPlusthon/objects/document.py

@@ -0,0 +1,36 @@
+from PySPlusthon import objects
+
+from . import Object
+
+
+class Document(Object):
+    attribute_names = [
+        ("id", "file_id"),
+        ("size", "file_size"),
+        ("name", "file_name")
+    ]
+
+    @classmethod
+    def from_protobuf(cls, protobuf_data):
+        return cls(
+            id=protobuf_data.file_id,
+            name=protobuf_data.name,
+            size=protobuf_data.file_size,
+            mime_type=protobuf_data.mime_type
+        )
+
+    def __init__(
+            self,
+            id: str = None,
+            thumb: "objects.Photo" = None,
+            name: str = None,
+            mime_type: str = None,
+            size: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.thumb: None = thumb
+        self.name: str = name
+        self.mime_type: str = mime_type
+        self.size: int = size

PySPlusthon/objects/edited_message.py

@@ -0,0 +1,27 @@
+from . import Object
+from PySPlusthon import objects
+
+
+class EditedMessage(Object):
+    attribute_names = [
+        ("id", "message_id"),
+        ("author", "from")
+    ]
+
+    def __init__(
+            self,
+            id: int = None,
+            author: "objects.User" = None,
+            date: "objects.Date" = None,
+            chat: "objects.Chat" = None,
+            edit_date: "objects.Date" = None,
+            text: str = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: int = id
+        self.author: "objects.User" = author
+        self.date: "objects.Date" = date
+        self.chat: "objects.Chat" = chat
+        self.edit_date: "objects.Date" = edit_date
+        self.text: str = text

PySPlusthon/objects/entity.py

@@ -0,0 +1,21 @@
+from . import Object
+from PySPlusthon import objects
+
+
+class Entity(Object):
+
+    def __init__(
+            self,
+            type: str = None,
+            offset: int = None,
+            length: int = None,
+            url: str = None,
+            user: "objects.User" = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.type: str = type
+        self.offset: int = offset
+        self.length: int = length
+        self.url: str = url
+        self.user: "objects.User" = user

PySPlusthon/objects/file.py

@@ -0,0 +1,21 @@
+from . import Object
+
+
+class File(Object):
+    attribute_names = [
+        ("id", "file_id"),
+        ("size", "file_size"),
+        ("path", "file_path")
+    ]
+
+    def __init__(
+            self,
+            id: str = None,
+            size: int = None,
+            path: str = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.size: int = size
+        self.path: str = path

PySPlusthon/objects/forward_origin.py

@@ -0,0 +1,22 @@
+from PySPlusthon import enums
+from . import Object
+from .. import objects
+
+
+class ForwardOrigin(Object):
+
+    def __init__(
+            self,
+            type: "enums.PeerType" = None,
+            date: "objects.Date" = None,
+            sender_user: "objects.User" = None,
+            chat: "objects.Chat" = None,
+            message_id: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.type: "enums.PeerType" = type
+        self.date: "objects.Date" = date
+        self.sender_user: "objects.User" = sender_user
+        self.chat: "objects.Chat" = chat
+        self.message_id: int = message_id

PySPlusthon/objects/inline_keyboard.py

@@ -0,0 +1,65 @@
+from typing import Union, List, Tuple
+from copy import deepcopy
+
+from . import Object, ReplyMarkup, InlineKeyboardButton
+from .list import List as PySplusthonList
+from PySPlusthon import objects
+
+
+class InlineKeyboard(ReplyMarkup):
+    def __init__(
+            self,
+            *rows: List[Union["objects.InlineKeyboardButton", Tuple[str, str]]],
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        if "inline_keyboard" not in kwargs:
+            self.inline_keyboard: List[List["objects.InlineKeyboardButton"]] = PySplusthonList()
+        for row in rows:
+            self.add_row(*row)
+
+    @classmethod
+    def expected_types(cls):
+        expected_types = super().expected_types()
+        expected_types["inline_keyboard"] = List[List[objects.InlineKeyboardButton]]
+        return expected_types
+
+    @classmethod
+    def wrap(cls, raw_object):
+        return Object.wrap.__func__(cls, raw_object)
+
+    def add_button(
+            self,
+            button: Union["objects.InlineKeyboardButton", Tuple[str, str]],
+            row_index: int = -1,
+            button_index: int = -1
+    ):
+        if isinstance(button, tuple):
+            button = InlineKeyboardButton(*button)
+        if button_index == -1:
+            self.inline_keyboard[row_index].append(button)
+        elif button_index < 0:
+            self.inline_keyboard[row_index].insert(button_index + 1, button)
+        else:
+            self.inline_keyboard[row_index].insert(button_index, button)
+
+    def add_row(self, *row: Union["objects.InlineKeyboardButton", Tuple[str, str]], row_index: int = -1):
+        if row_index == -1:
+            self.inline_keyboard.append(PySplusthonList())
+        elif row_index < 0:
+            self.inline_keyboard.insert(row_index + 1, PySplusthonList())
+        else:
+            self.inline_keyboard.insert(row_index, PySplusthonList())
+        for button in row:
+            self.add_button(button, row_index)
+
+    def on_click(self, row_index: int, button_index: int):
+        from ..conditions import click
+        return click(self, row_index, button_index)
+
+    def format(self, *args, **kwargs):
+        markup = deepcopy(self)
+        for i, row in enumerate(markup.inline_keyboard):
+            for j, button in enumerate(row):
+                markup.inline_keyboard[i][j] = button.format(*args, **kwargs)
+        return markup

PySPlusthon/objects/inline_keyboard_button.py

@@ -0,0 +1,30 @@
+from typing import Union
+from copy import deepcopy
+
+from . import Object
+from PySPlusthon import objects
+
+
+class InlineKeyboardButton(Object):
+
+    def __init__(
+            self,
+            text: str = None,
+            callback_data: str = None,
+            url: str = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.text: str = text
+        self.callback_data: str = callback_data
+        self.url: str = url
+
+    def format(self, *args, **kwargs):
+        button = deepcopy(self)
+        if button.text:
+            button.text = button.text.format(*args, **kwargs)
+        if button.callback_data:
+            button.callback_data = button.callback_data.format(*args, **kwargs)
+        if button.url:
+            button.url = button.url.format(*args, **kwargs)
+        return button

PySPlusthon/objects/input_media.py

@@ -0,0 +1,44 @@
+from typing import Union, BinaryIO
+from os.path import isfile
+
+from sqlalchemy import Boolean
+
+from . import Object
+
+
+class InputMedia(Object):
+
+    def __init__(
+            self,
+            type: str = None,
+            media: Union[str, bytes, BinaryIO] = None,
+            caption: str = None,
+            parse_mode : str = None,
+            disable_content_type_detection : Boolean = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.type: str = type
+        self.media: Union[str, bytes, BinaryIO] = resolve_media(media)
+        self.caption: str = caption
+
+    @property
+    def is_json_serializable(self):
+        return isinstance(self.media, str)
+
+
+def resolve_media(media):
+    if isinstance(media, InputMedia):
+        return media.media
+
+    if isinstance(media, Object):
+        if getattr(media, "id", None):
+            return media.id
+
+    try:
+        if isfile(media):
+            return open(media, "rb")
+    except TypeError:
+        pass
+
+    return media

PySPlusthon/objects/input_media_photo.py

@@ -0,0 +1,14 @@
+from typing import Union, BinaryIO
+
+from . import InputMedia
+
+
+class InputMediaPhoto(InputMedia):
+
+    def __init__(
+            self,
+            media: Union[str, bytes, BinaryIO] = None,
+            caption: str = None,
+            **kwargs
+    ):
+        super().__init__("photo", media, caption, **kwargs)

PySPlusthon/objects/input_media_video.py

@@ -0,0 +1,14 @@
+from typing import Union, BinaryIO
+
+from . import InputMedia
+
+
+class InputMediaVideo(InputMedia):
+
+    def __init__(
+            self,
+            media: Union[str, bytes, BinaryIO] = None,
+            caption: str = None,
+            **kwargs
+    ):
+        super().__init__("video", media, caption, **kwargs)

PySPlusthon/objects/invite_link.py

@@ -0,0 +1,32 @@
+from . import Object
+from PySPlusthon import objects
+
+
+class InviteLink(Object):
+    attribute_names = [
+        ("link", "invite_link")
+    ]
+
+    def __init__(
+            self,
+            link: str = None,
+            creator: "objects.User" = None,
+            creates_join_request: bool = None,
+            is_primary: bool = None,
+            is_revoked: bool = None,
+            name: str = None,
+            expire_date: int = None,
+            member_limit: int = None,
+            pending_join_request_count: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.link: str = link
+        self.creator: "objects.User" = creator
+        self.creates_join_request: bool = creates_join_request
+        self.is_primary: bool = is_primary
+        self.is_revoked: bool = is_revoked
+        self.name: str = name
+        self.expire_date: int = expire_date
+        self.member_limit: int = member_limit
+        self.pending_join_request_count: int = pending_join_request_count

PySPlusthon/objects/list.py

@@ -0,0 +1,24 @@
+from . import Object, unwrap
+
+
+class List(list, Object):
+    @classmethod
+    def wrap(cls, raw_object):
+        return cls(raw_object)
+
+    def bind(self, client):
+        self.client = client
+        for element in self:
+            if isinstance(element, Object):
+                element.bind(client)
+
+    def unwrap(self):
+        return unwrap(self)
+
+    def __repr__(self):
+        if not self:
+            return "[]"
+
+        elements = ",\n".join(f"{repr(element)}" for element in self)
+        elements = "\n".join(" "*4 + line for line in elements.splitlines())
+        return f"[\n{elements}\n]"

PySPlusthon/objects/location.py

@@ -0,0 +1,14 @@
+from . import Object
+
+
+class Location(Object):
+
+    def __init__(
+            self,
+            longitude: float = None,
+            latitude: float = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.longitude: float = longitude
+        self.latitude: float = latitude