PySPlusthon 0.5.3__py3-none-any.whl

PySPlusthon/objects/photo.py

@@ -0,0 +1,31 @@
+from . import Object
+
+
+class Photo(Object):
+    attribute_names = [
+        ("id", "file_id"),
+        ("size", "file_size")
+    ]
+
+    @classmethod
+    def from_protobuf(cls, protobuf_data):
+        return cls(
+            id=protobuf_data.file_id,
+            width=protobuf_data.ext.document_ex_photo.w,
+            height=protobuf_data.ext.document_ex_photo.h,
+            size=protobuf_data.file_size
+        )
+
+    def __init__(
+            self,
+            id: str = None,
+            width: int = None,
+            height: int = None,
+            size: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.width: int = width
+        self.height: int = height
+        self.size: int = size

PySPlusthon/objects/poll.py

@@ -0,0 +1,34 @@
+from typing import List
+
+from PySPlusthon import enums
+from . import Object
+from .poll_option import PollOption
+
+
+class Poll(Object):
+
+    def __init__(
+            self,
+            id: str = None,
+            question: str = None,
+            options: List[PollOption] = None,
+            total_voter_count: int = None,
+            is_closed: bool = None,
+            is_anonymous: bool = None,
+            type: "enums.PollType" = None,
+            allows_multiple_answers: bool = None,
+            allows_revoting: bool = None,
+            members_only: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.question: str = question
+        self.options: List[PollOption] = options
+        self.total_voter_count: int = total_voter_count
+        self.is_closed: bool = is_closed
+        self.is_anonymous: bool = is_anonymous
+        self.type: "enums.PollType" = type
+        self.allows_multiple_answers: bool = allows_multiple_answers
+        self.allows_revoting: bool = allows_revoting
+        self.members_only: bool = members_only

PySPlusthon/objects/poll_option.py

@@ -0,0 +1,16 @@
+from . import Object
+
+
+class PollOption(Object):
+
+    def __init__(
+            self,
+            persistent_id: str = None,
+            text: str = None,
+            voter_count: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.persistent_id: str = persistent_id
+        self.text: str = text
+        self.voter_count: int = voter_count

PySPlusthon/objects/reply_keyboard.py

@@ -0,0 +1,79 @@
+from typing import Union, List
+from copy import deepcopy
+
+from . import ReplyMarkup, ReplyKeyboardButton
+from .list import Object, List as PySplusthonList
+from PySPlusthon import objects
+
+
+class ReplyKeyboard(ReplyMarkup):
+    attribute_names = [
+        ("resize", "resize_keyboard"),
+        ("one_time", "one_time_keyboard"),
+        ("remove", "remove_keyboard")
+    ]
+
+    def __init__(
+            self,
+            *rows: List[Union["objects.ReplyKeyboardButton", str]],
+            resize: bool = None,
+            one_time: bool = None,
+            selective: bool = None,
+            remove: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        if "keyboard" not in kwargs:
+            self.keyboard: List[List["objects.ReplyKeyboardButton"]] = PySplusthonList()
+        for row in rows:
+            self.add_row(*row)
+        self.resize: bool = resize
+        self.one_time: bool = one_time
+        self.selective: bool = selective
+        self.remove: bool = remove
+
+    @classmethod
+    def expected_types(cls):
+        expected_types = super().expected_types()
+        expected_types["keyboard"] = List[List[objects.ReplyKeyboardButton]]
+        return expected_types
+
+    @classmethod
+    def wrap(cls, raw_object):
+        return Object.wrap.__func__(cls, raw_object)
+
+    def add_button(
+            self,
+            button: Union["objects.ReplyKeyboardButton", str],
+            row_index: int = -1,
+            button_index: int = -1
+    ):
+        if isinstance(button, str):
+            button = ReplyKeyboardButton(button)
+        if button_index == -1:
+            self.keyboard[row_index].append(button)
+        elif button_index < 0:
+            self.keyboard[row_index].insert(button_index + 1, button)
+        else:
+            self.keyboard[row_index].insert(button_index, button)
+
+    def add_row(self, *row: Union["objects.ReplyKeyboardButton", str], row_index: int = -1):
+        if row_index == -1:
+            self.keyboard.append(PySplusthonList())
+        elif row_index < 0:
+            self.keyboard.insert(row_index + 1, PySplusthonList())
+        else:
+            self.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.keyboard):
+            for j, button in enumerate(row):
+                markup.keyboard[i][j] = button.format(*args, **kwargs)
+        return markup

PySPlusthon/objects/reply_keyboard_button.py

@@ -0,0 +1,27 @@
+from typing import Union
+from copy import deepcopy
+
+from . import Object
+from PySPlusthon import objects
+
+
+class ReplyKeyboardButton(Object):
+
+    def __init__(
+            self,
+            text: str = None,
+            request_contact: bool = None,
+            request_location: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.text: str = text
+        self.request_contact: bool = request_contact
+        self.request_location: bool = request_location
+        
+
+    def format(self, *args, **kwargs):
+        button = deepcopy(self)
+        if button.text:
+            button.text = button.text.format(*args, **kwargs)
+        return button

PySPlusthon/objects/reply_keyboard_remove.py

@@ -0,0 +1,10 @@
+from . import ReplyKeyboard
+
+
+class ReplyKeyboardRemove(ReplyKeyboard):
+
+    def __init__(self):
+        super().__init__(remove=True)
+
+    def __repr__(self):
+        return f"{type(self).__name__}()"

PySPlusthon/objects/reply_markup.py

@@ -0,0 +1,12 @@
+from . import Object
+
+
+class ReplyMarkup(Object):
+
+    @classmethod
+    def wrap(cls, raw_object):
+        from . import InlineKeyboard, ReplyKeyboard
+        if "inline_keyboard" in raw_object:
+            return InlineKeyboard.wrap(raw_object)
+        elif "keyboard" in raw_object:
+            return ReplyKeyboard.wrap(raw_object)

PySPlusthon/objects/shipping_address.py

@@ -0,0 +1,22 @@
+from .object import Object
+
+
+class ShippingAddress(Object):
+
+    def __init__(
+            self,
+            country_code: str = None,
+            stat: str = None,
+            city: str = None,
+            street_line1: str = None,
+            street_line2: str = None,
+            post_code: str = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.country_code: str = country_code
+        self.stat: str = stat
+        self.city: str = city
+        self.street_line1: str = street_line1
+        self.street_line2: str = street_line2
+        self.post_code: str = post_code

PySPlusthon/objects/shipping_query.py

@@ -0,0 +1,20 @@
+from . import Object
+from PySPlusthon import objects
+
+
+class ShippingQuery(Object):
+    attribute_names = [
+        ("author", "from")
+    ]
+
+    def __init__(
+            self,
+            id: str = None,
+            author: "objects.User" = None,
+            shipping_address: "objects.ShippingAddress" = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.author: "objects.User" = author
+        self.shipping_address: "objects.ShippingAddress" = shipping_address

PySPlusthon/objects/sticker.py

@@ -0,0 +1,54 @@
+from PySPlusthon import objects
+from . import Object
+
+
+class Sticker(Object):
+    attribute_names = [
+        ("id", "file_id"),
+        ("unique_id", "file_unique_id"),
+        ("size", "file_size")
+    ]
+
+    @classmethod
+    def from_protobuf(cls, protobuf_data):
+        return cls(
+            id=protobuf_data.image512.file_location.file_id,
+            unique_id=str(protobuf_data.sticker_id.value),
+            width=protobuf_data.image512.width,
+            height=protobuf_data.image512.height,
+            size=protobuf_data.image512.file_size,
+            thumb=objects.Photo(
+                id=protobuf_data.image256.file_location.file_id,
+                width=protobuf_data.image256.width,
+                height=protobuf_data.image256.height,
+                size=protobuf_data.image256.file_size
+            )
+        )
+
+    def __init__(
+            self,
+            id: str = None,
+            unique_id: str = None,
+            width: int = None,
+            height: int = None,
+            size: int = None,
+            thumb: "objects.Photo" = None,
+            emoji: str = None,
+            set_name: str = None,
+            type: str = None,
+            is_animated: bool = None,
+            is_video: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.unique_id: str = unique_id
+        self.width: int = width
+        self.height: int = height
+        self.size: int = size
+        self.thumb: "objects.Photo" = thumb
+        self.emoji: str = emoji
+        self.set_name: str = set_name
+        self.type: str = type
+        self.is_animated: bool = is_animated
+        self.is_video: bool = is_video

PySPlusthon/objects/sticker_set.py

@@ -0,0 +1,29 @@
+from typing import List
+
+from . import Object
+from PySPlusthon import objects
+
+
+class StickerSet(Object):
+
+    def __init__(
+            self,
+            name: str = None,
+            title: str = None,
+            sticker_type: str = None,
+            is_animated: bool = None,
+            is_video: bool = None,
+            contains_masks: bool = None,
+            stickers: List["objects.Sticker"] = None,
+            thumb: "objects.Photo" = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.name: str = name
+        self.title: str = title
+        self.sticker_type: str = sticker_type
+        self.is_animated: bool = is_animated
+        self.is_video: bool = is_video
+        self.contains_masks: bool = contains_masks
+        self.stickers: List["objects.Sticker"] = stickers
+        self.thumb: "objects.Photo" = thumb

PySPlusthon/objects/transaction.py

@@ -0,0 +1,25 @@
+from . import Object
+from PySPlusthon import enums
+
+
+class Transaction(Object):
+    attribute_names = [
+        ("user_id", "userID"),
+        ("created_at", "createdAt"),
+    ]
+
+    def __init__(
+            self,
+            id: str = None,
+            status: "enums.TransactionStatus" = None,
+            user_id: int = None,
+            amount: int = None,
+            created_at: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.status: "enums.TransactionStatus" = status
+        self.user_id: int = user_id
+        self.amount: int = amount
+        self.created_at: int = created_at

PySPlusthon/objects/update.py

@@ -0,0 +1,33 @@
+from . import Object
+from PySPlusthon import objects
+
+
+class Update(Object):
+    attribute_names = [
+        ("id", "update_id")
+    ]
+
+    def __init__(
+            self,
+            id: int = None,
+            message: "objects.Message" = None,
+            edited_message: "objects.EditedMessage" = None,
+            channel_post: "objects.Message" = None,
+            edited_channel_post: "objects.EditedMessage" = None,
+            callback_query: "objects.CallbackQuery" = None,
+            shipping_query: "objects.ShippingQuery" = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: int = id
+        self.message: "objects.Message" = message
+        self.edited_message: "objects.EditedMessage" = edited_message
+        self.channel_post: "objects.Message" = channel_post
+        self.edited_channel_post: "objects.EditedMessage" = edited_channel_post
+        self.callback_query: "objects.CallbackQuery" = callback_query
+        self.shipping_query: "objects.ShippingQuery" = shipping_query
+
+    def get_effective_update(self):
+        for value in self.__dict__.values():
+            if isinstance(value, Object):
+                return value

PySPlusthon/objects/user.py

@@ -0,0 +1,60 @@
+from typing import Union
+
+from . import Object
+from ..states import StateMachine
+from ..sync_support import add_sync_support_to_object
+
+
+@add_sync_support_to_object
+class User(Object):
+
+    @classmethod
+    def from_protobuf(cls, id: int):
+        return cls(id=id)
+
+    def __init__(
+            self,
+            id: int = None,
+            username: str = None,
+            first_name: str = None,
+            last_name: str = None,
+            language_code: str = None,
+            is_bot: bool = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: int = id
+        self.username: str = username
+        self.first_name: str = first_name
+        self.last_name: str = last_name
+        self.language_code: str = language_code
+        self.is_bot: bool = is_bot
+
+    def mention(self, text=None):
+        text = text or self.full_name
+        return f"[{text}](uid:{self.id})"
+
+    @property
+    def link(self):
+        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
+        return ""
+
+    def set_state(self, state):
+        StateMachine.global_state_machine[self.id] = state
+
+    def get_state(self):
+        return StateMachine.global_state_machine[self.id]
+
+    def del_state(self):
+        del StateMachine.global_state_machine[self.id]
+
+    async def invite(self, chat_id: Union[int, str]):
+        return await self.client.invite_user(chat_id, self.id)

PySPlusthon/objects/video.py

@@ -0,0 +1,45 @@
+from PySPlusthon import objects
+from . import Object
+
+
+class Video(Object):
+    attribute_names = [
+        ("id", "file_id"),
+        ("size", "file_size")
+    ]
+
+    @classmethod
+    def from_protobuf(cls, protobuf_data):
+        return cls(
+            id=protobuf_data.file_id,
+            width=protobuf_data.ext.document_ex_video.w,
+            height=protobuf_data.ext.document_ex_video.h,
+            duration=protobuf_data.ext.document_ex_video.duration,
+            thumb=objects.Photo(
+                width=protobuf_data.thumb.w,
+                height=protobuf_data.thumb.h,
+                file_size=len(protobuf_data.thumb.thumb)
+            ),
+            mime_type=protobuf_data.mime_type,
+            size=protobuf_data.file_size
+        )
+
+    def __init__(
+            self,
+            id: str = None,
+            width: int = None,
+            height: int = None,
+            duration: int = None,
+            thumb: "objects.Photo" = None,
+            mime_type: str = None,
+            size: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.width: int = width
+        self.height: int = height
+        self.duration: int = duration
+        self.thumb: "objects.Photo" = thumb
+        self.mime_type: str = mime_type
+        self.size: int = size

PySPlusthon/objects/voice.py

@@ -0,0 +1,31 @@
+from . import Object
+
+
+class Voice(Object):
+    attribute_names = [
+        ("id", "file_id"),
+        ("size", "file_size")
+    ]
+
+    @classmethod
+    def from_protobuf(cls, protobuf_data):
+        return cls(
+            id=protobuf_data.file_id,
+            duration=protobuf_data.ext.document_ex_voice.duration,
+            mime_type=protobuf_data.mime_type,
+            size=protobuf_data.file_size
+        )
+
+    def __init__(
+            self,
+            id: str = None,
+            duration: int = None,
+            mime_type: str = None,
+            size: int = None,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.id: str = id
+        self.duration: int = duration
+        self.mime_type: str = mime_type
+        self.size: int = size

PySPlusthon/objects/webhook_info.py

@@ -0,0 +1,16 @@
+from . import Object
+
+
+class WebhookInfo(Object):
+
+    def __init__(
+            self,
+            url: str,
+            has_custom_certificate: bool,
+            pending_update_count: int,
+            **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.url: str = url
+        self.has_custom_certificate: bool = has_custom_certificate
+        self.pending_update_count: int = pending_update_count

PySPlusthon/proto/__init__.py

@@ -0,0 +1,2 @@
+from . import enums_pb2 as enums
+from . import ws_pb2 as ws

PySPlusthon/proto/enums_pb2.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: enums.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'enums.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0b\x65nums.proto\x12\x05\x65nums*\xec\x01\n\x08SendType\x12\x15\n\x11SEND_TYPE_UNKNOWN\x10\x00\x12\x13\n\x0fSEND_TYPE_PHOTO\x10\x01\x12\x13\n\x0fSEND_TYPE_VIDEO\x10\x02\x12\x13\n\x0fSEND_TYPE_VOICE\x10\x03\x12\x11\n\rSEND_TYPE_GIF\x10\x04\x12\x13\n\x0fSEND_TYPE_AUDIO\x10\x05\x12\x16\n\x12SEND_TYPE_DOCUMENT\x10\x06\x12\x15\n\x11SEND_TYPE_STICKER\x10\x07\x12\x1a\n\x16SEND_TYPE_CROWDFUNDING\x10\x08\x12\x17\n\x13SEND_TYPE_SPONSORED\x10\t*>\n\x0bRestriction\x12\x17\n\x13RESTRICTION_PRIVATE\x10\x00\x12\x16\n\x12RESTRICTION_PUBLIC\x10\x01*U\n\tGroupType\x12\x14\n\x10GROUP_TYPE_GROUP\x10\x00\x12\x16\n\x12GROUP_TYPE_CHANNEL\x10\x01\x12\x1a\n\x16GROUP_TYPE_SUPER_GROUP\x10\x02*\xbe\x01\n\nExPeerType\x12\x18\n\x14\x45X_PEER_TYPE_UNKNOWN\x10\x00\x12\x18\n\x14\x45X_PEER_TYPE_PRIVATE\x10\x01\x12\x16\n\x12\x45X_PEER_TYPE_GROUP\x10\x02\x12\x18\n\x14\x45X_PEER_TYPE_CHANNEL\x10\x03\x12\x14\n\x10\x45X_PEER_TYPE_BOT\x10\x04\x12\x1b\n\x17\x45X_PEER_TYPE_SUPERGROUP\x10\x05\x12\x17\n\x13\x45X_PEER_TYPE_THREAD\x10\x06*\x9b\x03\n\nTypingType\x12\x17\n\x13TYPING_TYPE_UNKNOWN\x10\x00\x12\x14\n\x10TYPING_TYPE_TEXT\x10\x01\x12\x1f\n\x1bTYPING_TYPE_VOICE_RECORDING\x10\x02\x12\x1d\n\x19TYPING_TYPE_SENDING_VOICE\x10\x03\x12\x1c\n\x18TYPING_TYPE_SENDING_FILE\x10\x04\x12\x1d\n\x19TYPING_TYPE_SENDING_PHOTO\x10\x05\x12\x1d\n\x19TYPING_TYPE_SENDING_VIDEO\x10\x06\x12\x1d\n\x19TYPING_TYPE_SENDING_MUSIC\x10\x07\x12 \n\x1cTYPING_TYPE_CHOOSING_STICKER\x10\x08\x12\x1c\n\x18TYPING_TYPE_CHOOSING_GIF\x10\t\x12$\n TYPING_TYPE_CREATING_GIFT_PACKET\x10\n\x12\x1d\n\x19TYPING_TYPE_SENDING_ALBUM\x10\x0b\x12\x1e\n\x1aTYPING_TYPE_CHOOSING_EMOJI\x10\x0c*4\n\x03Sex\x12\x0f\n\x0bSEX_UNKNOWN\x10\x00\x12\x0c\n\x08SEX_MALE\x10\x01\x12\x0e\n\nSEX_FEMALE\x10\x02\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'enums_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_SENDTYPE']._serialized_start=23
+  _globals['_SENDTYPE']._serialized_end=259
+  _globals['_RESTRICTION']._serialized_start=261
+  _globals['_RESTRICTION']._serialized_end=323
+  _globals['_GROUPTYPE']._serialized_start=325
+  _globals['_GROUPTYPE']._serialized_end=410
+  _globals['_EXPEERTYPE']._serialized_start=413
+  _globals['_EXPEERTYPE']._serialized_end=603
+  _globals['_TYPINGTYPE']._serialized_start=606
+  _globals['_TYPINGTYPE']._serialized_end=1017
+  _globals['_SEX']._serialized_start=1019
+  _globals['_SEX']._serialized_end=1071
+# @@protoc_insertion_point(module_scope)