pyrobale 0.2.9.2__tar.gz → 0.2.9.4.1__tar.gz

{pyrobale-0.2.9.2 → pyrobale-0.2.9.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyrobale
-Version: 0.2.9.2
+Version: 0.2.9.4.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.9.2 → pyrobale-0.2.9.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pyrobale"
-version = "0.2.9.2"
+version = "0.2.9.4.1"
 authors = [
     { name = "Ali Safamanesh", email = "darg.q.a.a@gmail.com" },
 ]

{pyrobale-0.2.9.2 → pyrobale-0.2.9.4.1}/pyrobale.py

@@ -26,8 +26,11 @@ from typing import Union, Optional, Dict, Any, List, Tuple, Callable
 import sqlite3
 import json
 import traceback
+import asyncio
+import queue
+
+__version__ = '0.2.9.4.1'
 
-__version__ = '0.2.9.2'
 
 class BaleException(Exception):
     """Base exception for Bale API errors"""
@@ -54,6 +57,8 @@ class BaleException(Exception):
 class BaleAPIError(BaleException):
     """Exception raised when Bale API returns an error response"""
     pass
+
+
 class BaleNetworkError(BaleException):
     """Exception raised when network-related issues occur during API calls"""
     pass
@@ -103,6 +108,7 @@ class BaleUnknownError(BaleException):
     """Exception raised for unexpected or unknown errors"""
     pass
 
+
 class ChatActions:
     """Represents different chat action states that can be sent to Bale"""
     TYPING: str = 'typing'
@@ -110,6 +116,26 @@ class ChatActions:
     VIDEO: str = 'record_video'
     CHOOSE_STICKER: str = 'choose_sticker'
 
+
+class Sticker:
+    def __init__(self, data: dict):
+        self.file_id = data.get('file_id')
+        self.file_unique_id = data.get('file_unique_id')
+        self.type = data.get('type')
+        self.width = data.get('width')
+        self.height = data.get('height')
+        self.file_size = data.get('file_size')
+
+
+class StickerSet:
+    def __init__(self, data: dict):
+        self.name = data.get('name')
+        self.title = data.get('title')
+        self.stickers = [Sticker(sticker)
+                         for sticker in data.get('stickers', [])]
+        self.thumbnail = data.get('thumbnail')
+
+
 class DataBase:
 
     """
@@ -483,7 +509,7 @@ class Chat:
                      reply_to_message: Union[int,
                                              str,
                                              'Message'] = None,
-                     reply_markup: Union['MenuKeyboardMarkup' , 'InlineKeyboardMarkup'] = None):
+                     reply_markup: Union['MenuKeyboardMarkup', 'InlineKeyboardMarkup'] = None):
         return self.client.send_invoice(
             self.id,
             title,
@@ -892,7 +918,7 @@ class User:
                      reply_to_message: Union[int,
                                              str,
                                              'Message'] = None,
-                     reply_markup: Union['MenuKeyboardMarkup' , 'InlineKeyboardMarkup'] = None):
+                     reply_markup: Union['MenuKeyboardMarkup', 'InlineKeyboardMarkup'] = None):
         return self.client.send_invoice(
             self.id,
             title,
@@ -1170,7 +1196,7 @@ class Message:
                       provider_token: str,
                       prices: list,
                       photo_url: Optional[str] = None,
-                      reply_markup: Union['MenuKeyboardMarkup' , 'InlineKeyboardMarkup'] = None):
+                      reply_markup: Union['MenuKeyboardMarkup', 'InlineKeyboardMarkup'] = None):
         return self.client.send_invoice(
             self.chat.id,
             title,
@@ -1182,6 +1208,13 @@ class Message:
             self,
             reply_markup)
 
+    def forward(self,
+                chat_id: Union[str, int]) -> 'Message':
+        """Forward a message to this user"""
+        return self.client.forward_message(
+            chat_id,
+            self.chat.id,
+            self.message_id)
 
 
 class LabeledPrice:
@@ -1547,7 +1580,7 @@ class InputMediaDocument(InputMedia):
 
 
 class CallbackQuery:
-    
+
     """Represents a callback query from a callback button"""
 
     def __init__(self, client: 'Client', data: Dict[str, Any]):
@@ -1587,6 +1620,7 @@ class CallbackQuery:
             reply_markup=reply_markup,
             reply_to_message=self.message.id)
 
+
 class Client:
     """Main client class for interacting with Bale API"""
 
@@ -1613,6 +1647,10 @@ class Client:
         self._threads = []
         self._polling = False
         self.user = None
+        self.event_handlers = []
+        self.message_queue = queue.Queue()
+        self.lock = threading.Lock()
+        self.waiting_events = {}
 
     def set_state(self,
                   chat_or_user_id: Union[Chat,
@@ -2149,6 +2187,44 @@ class Client:
         response = self._make_request('GET', 'getChatMembersCount', json=data)
         return response['result']
 
+    def upload_sticker_file(self, user_id: int, sticker: 'InputFile'):
+        """Upload a file for future use in sticker sets"""
+        data = {
+            'user_id': user_id
+        }
+        files = {
+            'sticker': sticker
+        }
+        response = self._make_request('POST', 'uploadStickerFile', data=data, files=files)
+        return response['result']
+
+    def create_new_sticker_set(
+            self, user_id: int, name: str, title: str, stickers: List['InputFile']) -> bool:
+        """Create a new sticker set owned by a user"""
+        data = {
+            'user_id': user_id,
+            'name': name,
+            'title': title
+        }
+        files = {}
+        for i, sticker in enumerate(stickers):
+            files[f'stickers[{i}]'] = sticker
+        response = self._make_request('POST', 'createNewStickerSet', data=data, files=files)
+        return response['result']
+
+    def add_sticker_to_set(self, user_id: int, name: str,
+                           sticker: 'InputFile') -> bool:
+        """Add a new sticker to a set created by the bot"""
+        data = {
+            'user_id': user_id,
+            'name': name
+        }
+        files = {
+            'sticker': sticker
+        }
+        response = self._make_request('POST', 'addStickerToSet', data=data, files=files)
+        return response['result']
+
     def is_joined(self, user: Union[User, int, str],
                   chat: Union[Chat, int, str]) -> bool:
         """Check if user is a member of the chat"""
@@ -2234,6 +2310,9 @@ class Client:
         """Handle different types of messages"""
         msg_data = update.get('message', {})
 
+        if 'message' in update:
+            message = Message(self, {'ok': True, 'result': update['message']})
+
         if 'new_chat_members' in msg_data and hasattr(
                 self, '_member_join_handler'):
             chat, user = msg_data['chat'], msg_data['new_chat_members'][0]
@@ -2347,6 +2426,9 @@ class Client:
 
         self._polling = True
         self._threads = []
+        self.message_queue = queue.Queue()
+        self.event_handlers = []
+        self.lock = threading.Lock()
         offset = 0
         past_updates = collections.deque(maxlen=100)
         source_file = inspect.getfile(self.__class__)
@@ -2371,6 +2453,9 @@ class Client:
                     update_id = update['update_id']
                     if update_id not in past_updates:
                         past_updates.append(update_id)
+                        if 'message' in update:
+                            message = Message(
+                                self, {'ok': True, 'result': update['message']})
                         self._handle_update(update)
                         offset = update_id + 1
 
@@ -2413,11 +2498,15 @@ class Client:
                 self._close_handler()
             except Exception as e:
                 print(f"Error in close handler: {e}")
+    
+    def wait_for_message(self, checker=Any):
+        self.waiting_events[checker]        
 
     def create_ref_link(self, data: str) -> str:
         """Create a reference link for the bot"""
         return f"https://ble.ir/{self.get_me().username}?start={data}"
 
+
 def run_multiple_bots(bots: List[Client]) -> List[Client]:
     """
     Run multiple bots concurrently in separate threads.