pyrobale 0.2.8__py3-none-any.whl → 0.2.8.1__py3-none-any.whl

{pyrobale-0.2.8.dist-info → pyrobale-0.2.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyrobale
-Version: 0.2.8
+Version: 0.2.8.1
 Summary: A python wrapper for bale api
 Project-URL: github, https://github.com/pyrobale/pyrobale
 Project-URL: website, https://pyrobale.github.io

pyrobale-0.2.8.1.dist-info/RECORD

@@ -0,0 +1,5 @@
+pyrobale.py,sha256=3X2xUh7sdHBCZkp-JcXJ6xL0ggW4BdIcDpmhSR09v7A,89846
+pyrobale-0.2.8.1.dist-info/METADATA,sha256=np0kt27IkkZVub_7t8_n44e9bKOygIuMNaSV0l0v1co,43670
+pyrobale-0.2.8.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pyrobale-0.2.8.1.dist-info/licenses/LICENSE,sha256=XQsvifC5HE85232_XgRHpeQ35MfeL2YcpU4yPiAgmW4,35269
+pyrobale-0.2.8.1.dist-info/RECORD,,

pyrobale.py

@@ -20,8 +20,15 @@ import re
 import sys
 import os
 from urllib.parse import urlparse, unquote
+from io import BytesIO
+import mimetypes
+from os import PathLike
+from pathlib import Path
+import uuid
+from typing import Iterator
+
+__version__ = '0.2.8.1'
 
-__version__ = '0.2.8'
 
 class ChatActions:
     """Represents different chat action states that can be sent to Bale"""
@@ -29,8 +36,10 @@ class ChatActions:
     PHOTO: str = 'upload_photo'
     VIDEO: str = 'record_video'
     CHOOSE_STICKER: str = 'choose_sticker'
+
+
 class conditions:
-    
+
     """
     A class for defining conditions for message handling.
     """
@@ -352,18 +361,24 @@ class LabeledPrice:
 
 class Document:
     def __init__(self, data: dict):
+        print(data)
         if data:
             self.file_id = data.get('file_id')
             self.file_unique_id = data.get('file_unique_id')
             self.file_name = data.get('file_name')
             self.mime_type = data.get('mime_type')
             self.file_size = data.get('file_size')
+            self.input_file = InputFile(self.file_id)
         else:
             self.file_id = None
             self.file_unique_id = None
             self.file_name = None
             self.mime_type = None
             self.file_size = None
+            self.input_file = None
+
+    def __bool__(self):
+        return bool(self.file_id)
 
 
 class Invoice:
@@ -545,39 +560,130 @@ class InlineKeyboardMarkup:
         return {"inline_keyboard": self.inline_keyboard}
 
 
+class InputMedia:
+    """Base class for input media types"""
+
+    def __init__(self, media: str, caption: str = None):
+        self.media = media
+        self.caption = caption
+
+    @property
+    def media_dict(self) -> dict:
+        media_dict = {
+            'media': self.media,
+            'type': self.type
+        }
+        if self.caption:
+            media_dict['caption'] = self.caption
+        return media_dict
+
+
+class InputMediaPhoto(InputMedia):
+    """Represents a photo to be sent"""
+    type = 'photo'
+
+
+class InputMediaVideo(InputMedia):
+    """Represents a video to be sent"""
+    type = 'video'
+
+    def __init__(self, media: str, caption: str = None, width: int = None,
+                 height: int = None, duration: int = None):
+        super().__init__(media, caption)
+        self.width = width
+        self.height = height
+        self.duration = duration
+
+    @property
+    def media_dict(self) -> dict:
+        media_dict = super().media_dict
+        if self.width:
+            media_dict['width'] = self.width
+        if self.height:
+            media_dict['height'] = self.height
+        if self.duration:
+            media_dict['duration'] = self.duration
+        return media_dict
+
+
+class InputMediaAnimation(InputMedia):
+    """Represents an animation to be sent"""
+    type = 'animation'
+
+    def __init__(self, media: str, caption: str = None, width: int = None,
+                 height: int = None, duration: int = None):
+        super().__init__(media, caption)
+        self.width = width
+        self.height = height
+        self.duration = duration
+
+    @property
+    def media_dict(self) -> dict:
+        media_dict = super().media_dict
+        if self.width:
+            media_dict['width'] = self.width
+        if self.height:
+            media_dict['height'] = self.height
+        if self.duration:
+            media_dict['duration'] = self.duration
+        return media_dict
+
+
 class InputFile:
-    """Represents a file to be uploaded"""
+    """Represents a file to be sent"""
 
-    def __init__(self,
-                 client: 'Client',
-                 file: Union[str,
-                             bytes],
-                 filename: Optional[str] = None):
-        self.client = client
-        self.filename = filename
-        if isinstance(file, str):
-            if file.startswith(('http://', 'https://')):
-                r = requests.get(file)
-                r.raise_for_status()
-                self.file = r.content
-            else:
-                try:
-                    self.file = open(file, 'rb')
-                except IOError:
-                    raise BaleException(
-                        f"Failed to open file: {traceback.format_exc()}")
-        else:
-            self.file = file
+    def __init__(self, file: Union[str, bytes] = None, file_id: str = None):
+        if file and file_id:
+            raise ValueError(
+                "Either file or file_id should be provided, not both")
+        elif not file and not file_id:
+            raise ValueError("Either file or file_id must be provided")
 
-    def __del__(self):
-        if hasattr(self, 'file') and hasattr(self.file, 'close'):
-            self.file.close()
+        self.file = file
+        self.file_id = file_id
+
+    @property
+    def file_type(self) -> str:
+        if self.file_id:
+            return "id"
+        if isinstance(self.file, bytes):
+            return "bytes"
+        if self.file.startswith(('http://', 'https://')):
+            return "url"
+        return "path"
+
+    def __str__(self) -> str:
+        if self.file_id:
+            return self.file_id
+        return str(self.file)
+
+
+class InputMediaAudio(InputMedia):
+    """Represents an audio file to be sent"""
+    type = 'audio'
+
+    def __init__(self, media: str, caption: str = None, duration: int = None,
+                 performer: str = None, title: str = None):
+        super().__init__(media, caption)
+        self.duration = duration
+        self.performer = performer
+        self.title = title
 
     @property
-    def to_bytes(self):
-        if hasattr(self.file, 'read'):
-            return self.file.read()
-        return self.file
+    def media_dict(self) -> dict:
+        media_dict = super().media_dict
+        if self.duration:
+            media_dict['duration'] = self.duration
+        if self.performer:
+            media_dict['performer'] = self.performer
+        if self.title:
+            media_dict['title'] = self.title
+        return media_dict
+
+
+class InputMediaDocument(InputMedia):
+    """Represents a document to be sent"""
+    type = 'document'
 
 
 class CallbackQuery:
@@ -857,8 +963,8 @@ class Chat:
             photo_url,
             reply_to_message,
             reply_markup)
-    
-    def send_action(self, action: str, how_many_times = 1) -> bool:
+
+    def send_action(self, action: str, how_many_times=1) -> bool:
         """Send a chat action"""
         return self.client.send_chat_action(self.id, action, how_many_times)
 
@@ -1259,6 +1365,7 @@ class User:
             photo_url,
             reply_to_message,
             reply_markup)
+
     def send_action(self, action: str, how_many_times: int = 1):
         """Send a chat action to this user"""
         return self.client.send_chat_action(self.id, action, how_many_times)
@@ -1286,33 +1393,65 @@ class Message:
                     'chat', {})})
         self.text = result.get('text')
         self.caption = result.get('caption')
-        self.document = Document(result.get('document'))
-        self.photo = Document(result.get('photo'))
-        self.video = Document(result.get('video'))
-        self.audio = Document(result.get('audio'))
-        self.voice = Voice(result.get('voice'))
-        self.animation = result.get('animation')
-        self.contact = Contact(result.get('contact'))
-        self.location = Location(result.get('location'))
+
+        # Media handling
+        self.document = Document(
+            result.get(
+                'document',
+                {})) if result.get('document') else None
+        # Handle photo array properly
+        photos = result.get('photo', [])
+        self.photo = [Document(photo) for photo in photos] if photos else None
+        self.largest_photo = Document(photos[-1]) if photos else None
+
+        self.video = Document(
+            result.get('video')) if result.get('video') else None
+        self.audio = Document(
+            result.get('audio')) if result.get('audio') else None
+        self.voice = Voice(
+            result.get('voice')) if result.get('voice') else None
+        self.animation = Document(
+            result.get('animation')) if result.get('animation') else None
+        self.sticker = Document(
+            result.get('sticker')) if result.get('sticker') else None
+        self.video_note = Document(
+            result.get('video_note')) if result.get('video_note') else None
+
+        self.media_group_id = result.get('media_group_id')
+        self.has_media = any([self.document,
+                              self.photo,
+                              self.video,
+                              self.audio,
+                              self.voice,
+                              self.animation,
+                              self.sticker,
+                              self.video_note])
+
+        self.contact = Contact(
+            result.get('contact')) if result.get('contact') else None
+        self.location = Location(
+            result.get('location')) if result.get('location') else None
         self.forward_from = User(client, {'ok': True, 'result': result.get(
             'forward_from', {})}) if result.get('forward_from') else None
         self.forward_from_message_id = result.get('forward_from_message_id')
-        self.invoice = Invoice(result.get('invoice'))
+        self.invoice = Invoice(
+            result.get('invoice')) if result.get('invoice') else None
         self.reply_to_message = Message(client, {'ok': True, 'result': result.get(
             'reply_to_message', {})}) if result.get('reply_to_message') else None
         self.reply = self.reply_message
         self.send = lambda text, parse_mode=None, reply_markup=None: self.client.send_message(
             self.chat.id, text, parse_mode, reply_markup, reply_to_message=self)
-        
+
         self.command = None
         self.args = None
-        txt = self.text.split(' ')
-        
-        self.command = txt[0]
-        self.has_slash_command = self.command.startswith('/')
-        self.args = txt[1:]
-        
-        self.start = self.command == '/start'
+        txt = self.text.split(' ') if self.text else []
+
+        self.command = txt[0] if txt else None
+        self.has_slash_command = self.command.startswith(
+            '/') if self.text else None
+        self.args = txt[1:] if self.text else None
+
+        self.start = self.command == '/start' if self.text else None
 
     def edit(self,
              text: str,
@@ -1523,6 +1662,7 @@ class Client:
         self.database_name = database_name
         self.auto_log_start_message = auto_log_start_message
         self._base_url = f"{session}/bot{token}"
+        self._file_url = f"{session}/file/bot{token}"
         self._session = requests.Session()
         self._message_handler = None
         self._message_edit_handler = None
@@ -1601,6 +1741,17 @@ class Client:
         data = self._make_request('GET', 'getMe')
         return User(self, data)
 
+    def get_file(self, file_id: str) -> bytes:
+        """Get file information from Bale API"""
+        data = {
+            'file_id': file_id
+        }
+        response = self._make_request('POST', 'getFile', json=data)
+        file_path = response['result']['file_path']
+        url = f"{self._file_url}/{file_path}"
+        file_response = self._session.get(url)
+        return file_response.content
+
     def set_webhook(self, url: str, certificate: Optional[str] = None,
                     max_connections: Optional[int] = None) -> bool:
         """Set webhook for getting updates"""
@@ -1984,16 +2135,21 @@ class Client:
             'reply_markup': reply_markup.keyboard if reply_markup else None}
         response = self._make_request('POST', 'sendInvoice', json=data)
         return Message(self, response)
-    
-    def send_chat_action(self, chat: Union[int, str, 'Chat'], action: str, how_many_times: int = 1) -> bool:
+
+    def send_chat_action(self,
+                         chat: Union[int,
+                                     str,
+                                     'Chat'],
+                         action: str,
+                         how_many_times: int = 1) -> bool:
         """Send a chat action"""
         if not chat:
             raise ValueError("Chat ID cannot be empty")
-            
+
         data = {
-            'chat_id': str(chat) if isinstance(chat, (int, str)) else str(chat.id),
-            'action': action
-        }
+            'chat_id': str(chat) if isinstance(
+                chat, (int, str)) else str(
+                chat.id), 'action': action}
         res = []
         for _ in range(how_many_times):
             response = self._make_request('POST', 'sendChatAction', json=data)
@@ -2127,7 +2283,8 @@ class Client:
         """Handle different types of messages"""
         msg_data = update.get('message', {})
 
-        if 'new_chat_members' in msg_data and hasattr(self, '_member_join_handler'):
+        if 'new_chat_members' in msg_data and hasattr(
+                self, '_member_join_handler'):
             chat, user = msg_data['chat'], msg_data['new_chat_members'][0]
             self._create_thread(
                 self._member_join_handler,
@@ -2137,7 +2294,8 @@ class Client:
             )
             return
 
-        if 'left_chat_member' in msg_data and hasattr(self, '_member_leave_handler'):
+        if 'left_chat_member' in msg_data and hasattr(
+                self, '_member_leave_handler'):
             chat, user = msg_data['chat'], msg_data['left_chat_member']
             self._create_thread(
                 self._member_leave_handler,
@@ -2151,17 +2309,23 @@ class Client:
             text = msg_data['text']
             for command, handler in self._text_handlers.items():
                 if text.startswith(command):
-                    conds = conditions(self, message.author, message, None, message.chat)
+                    conds = conditions(
+                        self, message.author, message, None, message.chat)
                     params = inspect.signature(handler).parameters
                     args = (message, conds) if len(params) > 1 else (message,)
                     self._create_thread(handler, *args)
                     return
 
         if self._message_handler:
-            conds = conditions(self, message.author, message, None, message.chat)
+            conds = conditions(
+                self,
+                message.author,
+                message,
+                None,
+                message.chat)
             params = inspect.signature(self._message_handler).parameters
-            args = ((message, update, conds) if len(params) > 2 else 
-                   (message, update) if len(params) > 1 else (message,))
+            args = ((message, update, conds) if len(params) > 2 else
+                    (message, update) if len(params) > 1 else (message,))
             self._create_thread(self._message_handler, *args)
 
     def _handle_update(self, update):
@@ -2176,17 +2340,21 @@ class Client:
 
         for update_type, (cls, handler) in message_types.items():
             if update_type in update:
-                message = cls(self, {'ok': True, 'result': update[update_type]})
+                message = cls(
+                    self, {
+                        'ok': True, 'result': update[update_type]})
                 handler(message, update)
                 return
 
         if 'callback_query' in update and self._callback_handler:
-            obj = CallbackQuery(self, {'ok': True, 'result': update['callback_query']})
+            obj = CallbackQuery(
+                self, {
+                    'ok': True, 'result': update['callback_query']})
             params = inspect.signature(self._callback_handler).parameters
             conds = conditions(self, obj.author, None, obj, obj.chat)
             args = (obj, update, conds) if len(params) > 2 else (obj, update)
             self._create_thread(self._callback_handler, *args)
-            
+
     def _handle_tick_events(self, current_time):
         """Handle periodic tick events"""
         if hasattr(self, '_tick_handlers'):
@@ -2198,7 +2366,7 @@ class Client:
     def run(self, debug=False):
         """Start p-olling for new messages"""
         try:
-            self.get_me()
+            self.user = User(self, {"ok": True, "result": self.get_me()})
         except BaseException:
             raise BaleTokenNotFoundError("token not found")
 
@@ -2222,7 +2390,7 @@ class Client:
                         print("Source file changed, restarting...")
                         python = sys.executable
                         os.execl(python, python, *sys.argv)
-                    
+
                 updates = self.get_updates(offset=offset, timeout=30)
                 for update in updates:
                     update_id = update['update_id']
@@ -2231,7 +2399,8 @@ class Client:
                         offset = update_id + 1
                         past_updates.add(update_id)
                         if len(past_updates) > 100:
-                            past_updates = set(sorted(list(past_updates))[-50:])
+                            past_updates = set(
+                                sorted(list(past_updates))[-50:])
 
                 current_time = time.time()
                 self._handle_tick_events(current_time)
@@ -2250,9 +2419,10 @@ class Client:
             return False
 
     def get_updates(self, offset: Optional[int] = None,
-                   limit: Optional[int] = None, 
-                   timeout: Optional[int] = None) -> List[Dict[str, Any]]:
-        params = {k: v for k, v in locals().items() if k != 'self' and v is not None}
+                    limit: Optional[int] = None,
+                    timeout: Optional[int] = None) -> List[Dict[str, Any]]:
+        params = {k: v for k, v in locals().items() if k !=
+                  'self' and v is not None}
         response = self._make_request('GET', 'getUpdates', params=params)
         return response.get('result', [])
 
@@ -2264,6 +2434,6 @@ class Client:
         self._threads.clear()
         if hasattr(self, '_close_handler'):
             self._close_handler()
-    
+
     def create_ref_link(self, data: str):
-        return f"https://ble.ir/{self.get_me().username}?start={data}"
+        return f"https://ble.ir/{self.get_me().username}?start={data}"

pyrobale-0.2.8.dist-info/RECORD

@@ -1,5 +0,0 @@
-pyrobale.py,sha256=Nb4HcO5aydJbc6Ofwz60PtZpLAh0Obx_IT4YydUeFgw,84787
-pyrobale-0.2.8.dist-info/METADATA,sha256=DjHaCavEk_o0UZI_eLqTRzDzHSTV7V85LI3ifOWOrT0,43668
-pyrobale-0.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pyrobale-0.2.8.dist-info/licenses/LICENSE,sha256=XQsvifC5HE85232_XgRHpeQ35MfeL2YcpU4yPiAgmW4,35269
-pyrobale-0.2.8.dist-info/RECORD,,