pyrobale 0.2.9.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

pyrobale/objects/chatmember.py

@@ -0,0 +1,162 @@
+from typing import TYPE_CHECKING
+from typing import Optional, Union
+
+if TYPE_CHECKING:
+    from .utils import build_api_url
+    from .user import User
+from .chat import Chat
+from .user import User
+from .enums import ChatType
+from ..client import Client
+
+
+class ChatMember:
+    """Represents a chat member in the Bale messenger, including their user
+    information and status."""
+
+    def __init__(
+        self,
+        user: "User",
+        status: str,
+        custom_title: Optional[str] = None,
+        is_anonymous: Optional[bool] = None,
+        can_be_edited: Optional[bool] = None,
+        can_manage_chat: Optional[bool] = None,
+        can_delete_messages: Optional[bool] = None,
+        can_edit_messages: Optional[bool] = None,
+        can_post_messages: Optional[bool] = None,
+        can_restrict_members: Optional[bool] = None,
+        can_promote_members: Optional[bool] = None,
+        can_change_info: Optional[bool] = None,
+        can_invite_users: Optional[bool] = None,
+        can_pin_messages: Optional[bool] = None,
+        can_manage_topics: Optional[bool] = None,
+        until_date: Optional[int] = None,
+        is_member: Optional[bool] = None,
+        can_send_messages: Optional[bool] = None,
+        can_send_audios: Optional[bool] = None,
+        can_send_documents: Optional[bool] = None,
+        can_send_photos: Optional[bool] = None,
+        can_send_videos: Optional[bool] = None,
+        can_send_video_notes: Optional[bool] = None,
+        can_send_voice_notes: Optional[bool] = None,
+        can_send_polls: Optional[bool] = None,
+        can_send_other_messages: Optional[bool] = None,
+        can_add_web_page_previews: Optional[bool] = None,
+        **kwargs
+    ):
+        self.user = user
+        self.status = status
+        self.custom_title = custom_title
+        self.is_anonymous = is_anonymous
+        self.can_be_edited = can_be_edited
+        self.can_manage_chat = can_manage_chat
+        self.can_delete_messages = can_delete_messages
+        self.can_edit_messages = can_edit_messages
+        self.can_post_messages = can_post_messages
+        self.can_restrict_members = can_restrict_members
+        self.can_promote_members = can_promote_members
+        self.can_change_info = can_change_info
+        self.can_invite_users = can_invite_users
+        self.can_pin_messages = can_pin_messages
+        self.can_manage_topics = can_manage_topics
+        self.until_date = until_date
+        self.is_member = is_member
+        self.can_send_messages = can_send_messages
+        self.can_send_audios = can_send_audios
+        self.can_send_documents = can_send_documents
+        self.can_send_photos = can_send_photos
+        self.can_send_videos = can_send_videos
+        self.can_send_video_notes = can_send_video_notes
+        self.can_send_voice_notes = can_send_voice_notes
+        self.can_send_polls = can_send_polls
+        self.can_send_other_messages = can_send_other_messages
+        self.can_add_web_page_previews = can_add_web_page_previews
+        self.client: Client = kwargs.get("kwargs", {}).get("client")
+        self.chat: Chat = Chat(**kwargs.get("kwargs", {}).get("chat"))
+
+    async def ban(self) -> bool:
+        """Bans the chat member from the chat.
+
+        :param chat_id: The ID of the chat.
+
+        :return: True if the member was banned successfully, False otherwise.
+        """
+        data = await self.chat.ban(self.user.id)
+        return data
+
+    async def unban(self) -> bool:
+        """Unbans the chat member from the chat.
+
+        :return: True if the member was unbanned successfully, False otherwise.
+        """
+        data = await self.chat.unban(self.user.id)
+        return data
+
+    async def promote(
+        self,
+        can_change_info: Optional[bool] = None,
+        can_post_messages: Optional[bool] = None,
+        can_edit_messages: Optional[bool] = None,
+        can_delete_messages: Optional[bool] = None,
+        can_invite_users: Optional[bool] = None,
+        can_restrict_members: Optional[bool] = None,
+        can_pin_messages: Optional[bool] = None,
+        can_promote_members: Optional[bool] = None,
+    ):
+        """Promote or demote a user in a chat.
+
+        Parameters:
+            can_change_info (Optional[bool]): Pass True if the user can change chat title, photo and other settings
+            can_post_messages (Optional[bool]): Pass True if the user can post messages in channels
+            can_edit_messages (Optional[bool]): Pass True if the user can edit messages in channels
+            can_delete_messages (Optional[bool]): Pass True if the user can delete messages
+            can_invite_users (Optional[bool]): Pass True if the user can invite new users
+            can_restrict_members (Optional[bool]): Pass True if the user can restrict, ban or unban chat members
+            can_pin_messages (Optional[bool]): Pass True if the user can pin messages
+            can_promote_members (Optional[bool]): Pass True if the user can add new administrators
+
+        Returns:
+            bool: True on success
+        """
+        return await self.chat.promote(
+            chat_id=self.chat.id,
+            user_id=self.user.id,
+            can_change_info=can_change_info,
+            can_post_messages=can_post_messages,
+            can_edit_messages=can_edit_messages,
+            can_delete_messages=can_delete_messages,
+            can_invite_users=can_invite_users,
+            can_restrict_members=can_restrict_members,
+            can_pin_messages=can_pin_messages,
+            can_promote_members=can_promote_members,
+        )
+
+    async def restrict(
+        self,
+        can_send_messages: Optional[bool] = None,
+        can_send_media_messages: Optional[bool] = None,
+        can_send_other_messages: Optional[bool] = None,
+        can_add_web_page_previews: Optional[bool] = None,
+        until_date: Optional[int] = None,
+    ) -> bool:
+        """Restrict a user in a chat.
+
+        Parameters:
+            can_send_messages (Optional[bool]): Pass True if the user can send text messages
+            can_send_media_messages (Optional[bool]): Pass True if the user can send media messages
+            can_send_other_messages (Optional[bool]): Pass True if the user can send other types of messages
+            can_add_web_page_previews (Optional[bool]): Pass True if the user can add web page previews
+            until_date (Optional[int]): Date when restrictions will be lifted for the user
+
+        Returns:
+            bool: True on success
+        """
+        return await self.chat.restrict(
+            user_id=self.user.id,
+            can_send_messages=can_send_messages,
+            can_send_media_messages=can_send_media_messages,
+            can_send_other_messages=can_send_other_messages,
+            can_add_web_page_previews=can_add_web_page_previews,
+            until_date=until_date,
+        )

pyrobale/objects/chatphoto.py

@@ -0,0 +1,18 @@
+from typing import Optional
+
+
+class ChatPhoto:
+    """This object represents a chat photo."""
+
+    def __init__(
+        self,
+        small_file_id: Optional[str] = None,
+        small_file_unique_id: Optional[str] = None,
+        big_file_id: Optional[str] = None,
+        big_file_unique_id: Optional[str] = None,
+        **kwargs
+    ):
+        self.small_file_id = small_file_id
+        self.small_file_unique_id = small_file_unique_id
+        self.big_file_id = big_file_id
+        self.big_file_unique_id = big_file_unique_id

pyrobale/objects/contact.py

@@ -0,0 +1,16 @@
+from typing import Optional
+
+
+class Contact:
+    def __init__(
+        self,
+        phone_number: Optional[str] = None,
+        first_name: Optional[str] = None,
+        last_name: Optional[str] = None,
+        user_id: Optional[int] = None,
+        **kwargs
+    ):
+        self.phone_number = phone_number
+        self.first_name = first_name
+        self.last_name = last_name
+        self.user_id = user_id

pyrobale/objects/copytextbutton.py

@@ -0,0 +1,5 @@
+class CopyTextButton:
+    """Represents a copy text button."""
+
+    def __init__(self, text: str, **kwargs):
+        self.text = text

pyrobale/objects/document.py

@@ -0,0 +1,26 @@
+from typing import TYPE_CHECKING, Optional
+
+if TYPE_CHECKING:
+    from .utils import build_api_url
+    from .photosize import PhotoSize
+
+
+class Document:
+    """Represents a general file to be sent without any special properties."""
+
+    def __init__(
+        self,
+        file_id: str,
+        file_unique_id: str,
+        thumbnail: "PhotoSize",
+        file_name: str,
+        mime_type: str,
+        file_size: int,
+        **kwargs
+    ):
+        self.file_id = file_id
+        self.file_unique_id = file_unique_id
+        self.thumbnail = thumbnail
+        self.file_name = file_name
+        self.mime_type = mime_type
+        self.file_size = file_size

pyrobale/objects/enums.py

@@ -0,0 +1,28 @@
+from enum import Enum
+
+
+class UpdatesTypes(Enum):
+    """Types of updates."""
+
+    MESSAGE = "message"
+    MESSAGE_EDITED = "message_edited"
+    CALLBACK_QUERY = "callback_query"
+    PRE_CHECKOUT_QUERY = "pre_checkout_query"
+    MEMBER_JOINED = "member_joined"
+    MEMBER_LEFT = "member_left"
+
+
+class ChatAction(Enum):
+    """Actions of a user in a chat."""
+
+    TYPING = "typing"
+    PHOTO = "upload_photo"
+    VIDEO = "upload_video"
+
+
+class ChatType(Enum):
+    """Types of chats."""
+
+    PRIVATE = "private"
+    GROUP = "group"
+    CHANNEL = "channel"

pyrobale/objects/file.py

@@ -0,0 +1,15 @@
+class File:
+    """A class representing a file object."""
+
+    def __init__(
+        self,
+        file_id: str,
+        file_unique_id: str,
+        file_size: int,
+        file_path: str,
+        **kwargs
+    ):
+        self.file_id = file_id
+        self.file_unique_id = file_unique_id
+        self.file_size = file_size
+        self.file_path = file_path

pyrobale/objects/inlinekeyboardbutton.py

@@ -0,0 +1,24 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .webappdata import WebAppData
+    from .copytextbutton import CopyTextButton
+
+
+class InlineKeyboardButton:
+    """Represents a button in an inline keyboard."""
+
+    def __init__(
+        self,
+        text: str,
+        url: str = None,
+        callback_data: str = None,
+        web_app: "WebAppData" = None,
+        copy_text_button: "CopyTextButton" = None,
+        **kwargs
+    ):
+        self.text = text
+        self.url = url
+        self.callback_data = callback_data
+        self.web_app = web_app
+        self.copy_text_button = copy_text_button

pyrobale/objects/inlinekeyboardmarkup.py

@@ -0,0 +1,84 @@
+from typing import TYPE_CHECKING, Union, Optional
+
+if TYPE_CHECKING:
+    from .webappinfo import WebAppInfo
+    from .copytextbutton import CopyTextButton
+
+
+class InlineKeyboardMarkup:
+    def __init__(self) -> None:
+        self.inline_keyboard: list[list[dict]] = []
+
+    def add_button(
+        self,
+        text: str,
+        callback_data: Optional[str] = None,
+        url: Optional[str] = None,
+        web_app: Optional[Union["WebAppInfo", str]] = None,
+        copy_text_button: Optional["CopyTextButton"] = None,
+        **kwargs
+    ) -> "InlineKeyboardMarkup":
+        """Adds a button to the inline keyboard.
+
+        Parameters:
+            text (str): The text to display on the button.
+            callback_data (str, optional): The callback data to send when the button is clicked.
+            url (str, optional): The URL to open when the button is clicked.
+            web_app (WebAppInfo, optional): The web app to open when the button is clicked.
+            copy_text_button (CopyTextButton, optional): The copy text button to add to the button.
+
+        Returns:
+            InlineKeyboardMarkup: The updated InlineKeyboardMarkup object.
+        """
+        button = {"text": text}
+
+        field_count = sum(
+            field is not None
+            for field in [callback_data, url, web_app, copy_text_button]
+        )
+        if field_count == 0:
+            raise ValueError(
+                "At least one of callback_data, url, web_app, or copy_text must be provided."
+            )
+        if field_count > 1:
+            raise ValueError(
+                "Only one of callback_data, url, web_app, or copy_text can be provided."
+            )
+
+        if callback_data:
+            button["callback_data"] = callback_data
+        elif url:
+            button["url"] = url
+        elif web_app:
+            if isinstance(web_app, str):
+                button["web_app"] = {"url": web_app}
+            elif hasattr(web_app, "to_dict"):
+                button["web_app"] = web_app.to_dict()
+            else:
+                raise ValueError(
+                    "web_app must be a string URL or an object with to_dict() method."
+                )
+        elif copy_text_button:
+            button["copy_text"] = {"text": copy_text_button.text}
+
+        if not self.inline_keyboard:
+            self.inline_keyboard.append([])
+        self.inline_keyboard[-1].append(button)
+        return self
+
+    def add_row(self) -> "InlineKeyboardMarkup":
+        """Adds a new row to the inline keyboard.
+
+        Returns:
+            InlineKeyboardMarkup: The updated InlineKeyboardMarkup object.
+        """
+        self.inline_keyboard.append([])
+        return self
+
+    def to_dict(self) -> dict:
+        print(self.inline_keyboard)
+        return {"inline_keyboard": self.inline_keyboard}
+
+    @property
+    def json(self) -> dict:
+        return self.to_dict()

pyrobale/objects/inputfile.py

@@ -0,0 +1,17 @@
+from typing import List, Union
+
+
+class InputFile:
+    def __init__(self, file: Union[bytes, str], **kwargs):
+        """
+        Initialize an InputFile object that can handle file uploads in 3 ways:
+        1. Using file_id for files already on Bale servers
+        2. Using HTTP URL for remote files (5MB limit for images, 20MB for other content)
+        3. Using multipart/form-data upload (10MB limit for images, 50MB for other content)
+
+        Args:
+            file: Can be either:
+                - bytes: For direct file upload
+                - str: For file_id or HTTP URL
+        """
+        self.file = file

pyrobale/objects/inputmedias.py

@@ -0,0 +1,166 @@
+class InputMedia:
+    """Base class for all input media types.
+
+    This is an abstract class that should not be used directly.
+    """
+
+    def __init__(self, media, caption=None):
+        """Initialize the base InputMedia object.
+
+        Args:
+            media (str): File to send. Can be a file_id, HTTP URL, or "<attach://file_attach_name>".
+            caption (str, optional): Caption for the media, 0-1024 characters.
+        """
+        self.media = media
+        self.caption = caption
+
+
+class InputMediaPhoto(InputMedia):
+    """Represents a photo to be sent.
+
+    This object represents a photo that needs to be sent to Bale.
+    """
+
+    def __init__(self, media, caption=None):
+        """Initialize an InputMediaPhoto object.
+
+        Args:
+            media (str): File to send. Can be:
+                1) A file_id to send a file that exists on Bale servers (recommended)
+                2) An HTTP URL for Bale to get a file from the Internet
+                3) "<attach://file_attach_name>" to upload a new file using multipart/form-data
+            caption (str, optional): Caption for the photo, 0-1024 characters.
+        """
+        super().__init__(media, caption)
+        self.type = "photo"
+
+
+class InputMediaVideo(InputMedia):
+    """Represents a video to be sent.
+
+    This object represents a video that needs to be sent to Bale.
+    """
+
+    def __init__(
+        self,
+        media,
+        caption=None,
+        thumbnail=None,
+        width=None,
+        height=None,
+        duration=None,
+    ):
+        """Initialize an InputMediaVideo object.
+
+        Args:
+            media (str): File to send. Can be:
+                1) A file_id to send a file that exists on Bale servers (recommended)
+                2) An HTTP URL for Bale to get a file from the Internet
+                3) "<attach://file_attach_name>" to upload a new file using multipart/form-data
+            caption (str, optional): Caption for the video, 0-1024 characters.
+            thumbnail (InputFile or str, optional): Thumbnail of the video.
+                Should be in JPEG format and less than 200 KB in size.
+                Width and height should not exceed 320px.
+            width (int, optional): Video width.
+            height (int, optional): Video height.
+            duration (int, optional): Video duration in seconds.
+        """
+        super().__init__(media, caption)
+        self.type = "video"
+        self.thumbnail = thumbnail
+        self.width = width
+        self.height = height
+        self.duration = duration
+
+
+class InputMediaAnimation(InputMedia):
+    """Represents an animation to be sent.
+
+    This object represents an animation file (GIF or H.264/MPEG-4 AVC
+    video without sound) that needs to be sent to Bale.
+    """
+
+    def __init__(
+        self,
+        media,
+        caption=None,
+        thumbnail=None,
+        width=None,
+        height=None,
+        duration=None,
+    ):
+        """Initialize an InputMediaAnimation object.
+
+        Args:
+            media (str): File to send. Can be:
+                1) A file_id to send a file that exists on Bale servers (recommended)
+                2) An HTTP URL for Bale to get a file from the Internet
+                3) "<attach://file_attach_name>" to upload a new file using multipart/form-data
+            caption (str, optional): Caption for the animation, 0-1024 characters.
+            thumbnail (InputFile or str, optional): Thumbnail of the animation.
+                Should be in JPEG format and less than 200 KB in size.
+                Width and height should not exceed 320px.
+            width (int, optional): Animation width.
+            height (int, optional): Animation height.
+            duration (int, optional): Animation duration in seconds.
+        """
+        super().__init__(media, caption)
+        self.type = "animation"
+        self.thumbnail = thumbnail
+        self.width = width
+        self.height = height
+        self.duration = duration
+
+
+class InputMediaAudio(InputMedia):
+    """Represents an audio file to be sent.
+
+    This object represents an audio file that needs to be sent to Bale.
+    The file will be treated as music.
+    """
+
+    def __init__(self, media, caption=None, thumbnail=None, duration=None, title=None):
+        """Initialize an InputMediaAudio object.
+
+        Args:
+            media (str): File to send. Can be:
+                1) A file_id to send a file that exists on Bale servers (recommended)
+                2) An HTTP URL for Bale to get a file from the Internet
+                3) "<attach://file_attach_name>" to upload a new file using multipart/form-data
+            caption (str, optional): Caption for the audio file, 0-1024 characters.
+            thumbnail (InputFile or str, optional): Thumbnail of the audio file.
+                Should be in JPEG format and less than 200 KB in size.
+                Width and height should not exceed 320px.
+            duration (int, optional): Duration of the audio in seconds.
+            title (str, optional): Title of the audio.
+        """
+        super().__init__(media, caption)
+        self.type = "audio"
+        self.thumbnail = thumbnail
+        self.duration = duration
+        self.title = title
+
+
+class InputMediaDocument(InputMedia):
+    """Represents a document to be sent.
+
+    This object represents a general file (document) that needs to be
+    sent to Bale.
+    """
+
+    def __init__(self, media, caption=None, thumbnail=None):
+        """Initialize an InputMediaDocument object.
+
+        Args:
+            media (str): File to send. Can be:
+                1) A file_id to send a file that exists on Bale servers (recommended)
+                2) An HTTP URL for Bale to get a file from the Internet
+                3) "<attach://file_attach_name>" to upload a new file using multipart/form-data
+            caption (str, optional): Caption for the document, 0-1024 characters.
+            thumbnail (InputFile or str, optional): Thumbnail of the document.
+                Should be in JPEG format and less than 200 KB in size.
+                Width and height should not exceed 320px.
+        """
+        super().__init__(media, caption)
+        self.type = "document"
+        self.thumbnail = thumbnail

pyrobale/objects/invoice.py

@@ -0,0 +1,14 @@
+class Invoice:
+    def __init__(
+        self,
+        title: str,
+        description: str,
+        start_parameter: str,
+        currency: str,
+        total_amount: int,
+    ) -> None:
+        self.title = title
+        self.description = description
+        self.start_parameter = start_parameter
+        self.currency = currency
+        self.total_amount = total_amount

pyrobale/objects/keyboardbutton.py

@@ -0,0 +1,19 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .webappinfo import WebAppInfo
+
+
+class KeyboardButton:
+    def __init__(
+        self,
+        text: str,
+        request_contact: bool = False,
+        request_location: bool = False,
+        web_app: "WebAppInfo" = None,
+        **kwargs
+    ):
+        self.text = text
+        self.request_contact = request_contact
+        self.request_location = request_location
+        self.web_app = web_app

pyrobale/objects/labeledprice.py

@@ -0,0 +1,8 @@
+class LabeledPrice:
+    def __init__(self, label: str, amount: int) -> None:
+        self.label = label
+        self.amount = amount
+
+    @property
+    def json(self):
+        return {"label": self.label, "amount": self.amount}

pyrobale/objects/location.py

@@ -0,0 +1,4 @@
+class Location:
+    def __init__(self, latitude: float, longitude: float):
+        self.latitude = latitude
+        self.longitude = longitude