pyrobale 0.2.8.2__tar.gz → 0.2.9__tar.gz

{pyrobale-0.2.8.2 → pyrobale-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyrobale
-Version: 0.2.8.2
+Version: 0.2.9
 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.2 → pyrobale-0.2.9}/pyproject.toml

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

{pyrobale-0.2.8.2 → pyrobale-0.2.9}/pyrobale.py

@@ -8,26 +8,19 @@ Features:
 - New and Up to date
 - Internal database management
 """
+from typing import Optional, Dict, Any, List, Union
+import os
 import json
 import time
-from typing import Optional, Dict, Any, List, Union
 import threading
 import traceback
 import sqlite3
 import inspect
-import requests
 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
+import requests
 
-__version__ = '0.2.8.2'
+__version__ = '0.2.9'
 
 
 class ChatActions:
@@ -743,6 +736,9 @@ class Chat:
         self.description = result.get('description')
         self.invite_link = result.get('invite_link')
         self.photo = result.get('photo')
+        self.is_channel_chat = self.CHANNEL = self.type == "channel"
+        self.is_group_chat = self.GROUP = self.type == "group"
+        self.is_private_chat = self.PRIVATE = self.type == "private"
 
     def send_photo(self,
                    photo: Union[str,
@@ -968,7 +964,7 @@ class Chat:
         """Send a chat action"""
         return self.client.send_chat_action(self.id, action, how_many_times)
 
-    def banChatMember(
+    def ban_chat_member(
             self,
             user_id: int,
             until_date: Optional[int] = None) -> bool:
@@ -982,7 +978,7 @@ class Chat:
             'POST', 'banChatMember', data=data)
         return response.get('ok', False)
 
-    def unbanChatMember(
+    def unban_chat_member(
             self,
             user_id: int,
             only_if_banned: bool = False) -> bool:
@@ -996,7 +992,7 @@ class Chat:
             'POST', 'unbanChatMember', data=data)
         return response.get('ok', False)
 
-    def promoteChatMember(
+    def promotee_chat_member(
             self,
             user_id: int,
             can_change_info: bool = None,
@@ -1022,7 +1018,7 @@ class Chat:
             'POST', 'promoteChatMember', data=data)
         return response.get('ok', False)
 
-    def setChatPhoto(self, photo: Union[str, bytes, InputFile]) -> bool:
+    def set_chat_photo(self, photo: Union[str, bytes, InputFile]) -> bool:
         """Set a new chat photo"""
         files = None
         data = {'chat_id': self.id}
@@ -1038,26 +1034,26 @@ class Chat:
             'POST', 'setChatPhoto', data=data, files=files)
         return response.get('ok', False)
 
-    def leaveChat(self) -> bool:
+    def leave_chat(self) -> bool:
         """Leave the chat"""
         data = {'chat_id': self.id}
         response = self.client._make_request('POST', 'leaveChat', data=data)
         return response.get('ok', False)
 
-    def getChat(self) -> 'Chat':
+    def get_chat(self) -> 'Chat':
         """Get up to date information about the chat"""
         data = {'chat_id': self.id}
         response = self.client._make_request('GET', 'getChat', data=data)
         return Chat(self.client, response)
 
-    def getChatMembersCount(self) -> int:
+    def get_members_count(self) -> int:
         """Get the number of members in the chat"""
         data = {'chat_id': self.id}
         response = self.client._make_request(
             'POST', 'getChatMembersCount', data=data)
         return response.get('result', 0)
 
-    def pinChatMessage(
+    def pin_message(
             self,
             message_id: int,
             disable_notification: bool = False) -> bool:
@@ -1071,7 +1067,7 @@ class Chat:
             'POST', 'pinChatMessage', data=data)
         return response.get('ok', False)
 
-    def unPinChatMessage(self, message_id: int) -> bool:
+    def unpin_message(self, message_id: int) -> bool:
         """Unpin a message in the chat"""
         data = {
             'chat_id': self.id,
@@ -1081,14 +1077,14 @@ class Chat:
             'POST', 'unpinChatMessage', data=data)
         return response.get('ok', False)
 
-    def unpinAllChatMessages(self) -> bool:
+    def unpin_all_messages(self) -> bool:
         """Unpin all messages in the chat"""
         data = {'chat_id': self.id}
         response = self.client._make_request(
             'POST', 'unpinAllChatMessages', data=data)
         return response.get('ok', False)
 
-    def setChatTitle(self, title: str) -> bool:
+    def set_chat_title(self, title: str) -> bool:
         """Change the title of the chat"""
         data = {
             'chat_id': self.id,
@@ -1097,7 +1093,7 @@ class Chat:
         response = self.client._make_request('POST', 'setChatTitle', data=data)
         return response.get('ok', False)
 
-    def setChatDescription(self, description: str) -> bool:
+    def set_chat_description(self, description: str) -> bool:
         """Change the description of the chat"""
         data = {
             'chat_id': self.id,
@@ -1107,21 +1103,21 @@ class Chat:
             'POST', 'setChatDescription', data=data)
         return response.get('ok', False)
 
-    def deleteChatPhoto(self) -> bool:
+    def delete_chat_photo(self) -> bool:
         """Delete the chat photo"""
         data = {'chat_id': self.id}
         response = self.client._make_request(
             'POST', 'deleteChatPhoto', data=data)
         return response.get('ok', False)
 
-    def createChatInviteLink(self) -> str:
+    def create_invite_link(self) -> str:
         """Create an invite link for the chat"""
         data = {'chat_id': self.id}
         response = self.client._make_request(
             'POST', 'createChatInviteLink', data=data)
         return response.get('result', {}).get('invite_link')
 
-    def revokeChatInviteLink(self, invite_link: str) -> bool:
+    def revoke_invite_link(self, invite_link: str) -> bool:
         """Revoke an invite link for the chat"""
         data = {
             'chat_id': self.id,
@@ -1131,7 +1127,7 @@ class Chat:
             'POST', 'revokeChatInviteLink', data=data)
         return response.get('ok', False)
 
-    def exportChatInviteLink(self) -> str:
+    def export_invite_link(self) -> str:
         """Generate a new invite link for the chat"""
         data = {'chat_id': self.id}
         response = self.client._make_request(
@@ -1670,6 +1666,8 @@ class Client:
         self._member_leave_handler = None
         self._member_join_handler = None
         self._threads = []
+        self._polling = False
+        self.user = None
 
     def set_state(self,
                   chat_or_user_id: Union[Chat,
@@ -2275,9 +2273,10 @@ class Client:
 
     def _create_thread(self, handler, *args):
         """Helper method to create and start a thread"""
-        thread = threading.Thread(target=handler, args=args, daemon=True)
-        thread.start()
-        self._threads.append(thread)
+        if handler:
+            thread = threading.Thread(target=handler, args=args, daemon=True)
+            thread.start()
+            self._threads.append(thread)
 
     def _handle_message(self, message, update):
         """Handle different types of messages"""
@@ -2316,7 +2315,7 @@ class Client:
                     self._create_thread(handler, *args)
                     return
 
-        if self._message_handler:
+        if hasattr(self, '_message_handler'):
             conds = conditions(
                 self,
                 message.author,
@@ -2335,7 +2334,7 @@ class Client:
         # Handle message and edited message
         message_types = {
             'message': (Message, self._handle_message),
-            'edited_message': (Message, self._message_edit_handler)
+            'edited_message': (Message, lambda m, u: self._create_thread(self._message_edit_handler, m, u) if hasattr(self, '_message_edit_handler') else None)
         }
 
         for update_type, (cls, handler) in message_types.items():
@@ -2346,7 +2345,7 @@ class Client:
                 handler(message, update)
                 return
 
-        if 'callback_query' in update and self._callback_handler:
+        if 'callback_query' in update and hasattr(self, '_callback_handler'):
             obj = CallbackQuery(
                 self, {
                     'ok': True, 'result': update['callback_query']})
@@ -2364,13 +2363,14 @@ class Client:
                     info['last_run'] = current_time
 
     def run(self, debug=False):
-        """Start p-olling for new messages"""
+        """Start polling for new messages"""
         try:
             self.user = self.get_me()
         except BaseException:
             raise BaleTokenNotFoundError("token not found")
 
         self._polling = True
+        self._threads = []
         offset = 0
         past_updates = set()
         source_file = inspect.getfile(self.__class__)
@@ -2391,7 +2391,7 @@ class Client:
                         python = sys.executable
                         os.execl(python, python, *sys.argv)
 
-                updates = self.get_updates(offset=offset, timeout=30)
+                updates = self.get_updates(offset=offset)
                 for update in updates:
                     update_id = update['update_id']
                     if update_id not in past_updates:
@@ -2418,11 +2418,8 @@ class Client:
             print(f"Error checking file modification time: {e}")
             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}
+    def get_updates(self, offset=None) -> List[Dict[str, Any]]:
+        params = {'offset': offset} if offset is not None else {}
         response = self._make_request('GET', 'getUpdates', params=params)
         return response.get('result', [])
 
@@ -2436,4 +2433,4 @@ class Client:
             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}"