PyRubikaBotAPI 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

{pyrubikabotapi-1.0.0.dist-info → pyrubikabotapi-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyRubikaBotAPI
-Version: 1.0.0
+Version: 1.0.1
 Summary: A Python library with a simple and familiar interface for working with the official Rubika bot API
 Home-page: https://github.com/alireza-sadeghian/PyRubikaBotAPI
 Author: Alireza Sadeghian
@@ -9,14 +9,14 @@ License: MIT
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENCE
 Requires-Dist: requests
 
 # PyRubikaBotAPI
 
-یک کتابخانه برای ساخت ربات‌های روبیکا با پایتون
+کتابخانه قدرتمند و آشنا برای ساخت ربات های تخصصی روبیکا
 
 ## نصب
 ```bash
@@ -38,17 +38,8 @@ def start(message):
 bot.polling()
 ```
 
-## مثال پیشرفته تر
-```py
-@bot.message_handler(content_types=['file'])
-def handle(msg):
-    file_id = msg.file.id
-    file_url = bot.get_file(file_id)
-    file = bot.download_file(file_url)
-    with open('file.format', 'wb') as f
-        f.write(file)
-    # استفاده از فایل ارسالی
-```
+برای استفاده دقیقتر حتما مستندات رو مطالعه بفرمایید و یا برای دریافت سورس کد ها 
+و مثال ها به کانال روبیکای ما سر بزنید
 
 ## مستندات:
 

pyrubikabotapi-1.0.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+rubibot/__init__.py,sha256=brh_qfPanBsFaTHIjtcX9GWJ7DgF3GujRy4scn4dqww,38036
+rubibot/exceptions.py,sha256=68MnHRgJWi4asxeTfH2rrHeRvBI49LR8KkKxYEAYcxg,250
+rubibot/metadata.py,sha256=eNUuFnRVQ_Ve8dh-VGMfcX0wxR0d06DiceE-sj1g_oQ,1413
+rubibot/parse.py,sha256=5F-jq0l1mc0q8SHl3iby8dv9N2r4WTf-FzlPesrdzHQ,4837
+rubibot/rubika_api.py,sha256=SgdvirBEXuocjKsFsvtdhXwK5JO-8t4jORZrGZPDiOY,9343
+rubibot/types.py,sha256=mnLTbYaD727YkMDcRWssIa326TlpAL4hc5JbVHb7uko,10483
+rubibot/updates.py,sha256=eXUkppIBhN-LDo0GS8VzkLi7t8U6v0kbb-Squy1BkZA,12366
+pyrubikabotapi-1.0.1.dist-info/LICENCE,sha256=mds18XAwr8xSo4pIhwMfB9spNkvrm5KKmLl_KIKQky4,1095
+pyrubikabotapi-1.0.1.dist-info/METADATA,sha256=Df-N0guE_VdzbGPKTtQZHpYxpz4WN8b-wccUHfUuS4o,1362
+pyrubikabotapi-1.0.1.dist-info/WHEEL,sha256=BNRMDyzLkkcmlv0J8ppDQkk2VED33SesJDynr9ED1gc,91
+pyrubikabotapi-1.0.1.dist-info/top_level.txt,sha256=TAhaSYH8WLX-dab28vY3_Wif-mrUw1pmS_RJOFaiogE,8
+pyrubikabotapi-1.0.1.dist-info/RECORD,,

rubibot/__init__.py

@@ -47,8 +47,8 @@ class RubiBot:
             commands: Optional[List[str]]=None,
             content_types: Optional[List[str]]=None,
             func=None,
-            update_types: Optional[List[str]]=None
-    ):
+            update_types: Optional[List[str]]=None,
+        ):
         """
         It is responsible for managing messages received from the user. 
         It handles all types of messages such as text, video, Voice, Poll, etc.
@@ -134,7 +134,7 @@ class RubiBot:
                         "commands": commands,
                         "content_types": content_types,
                         "func": func,
-                        "update_types": update_types
+                        "update_types": update_types,
                     }
                 })
             return handler
@@ -151,12 +151,15 @@ class RubiBot:
         return decorator
     
     def register_message_handler(self, handler: Callable, **filters):
+        for filter in ['commands', 'content_types', 'func', 'update_types']:
+            filters.setdefault(filter, None)
+
         self._message_handlers.append({
             'handler': handler,
             'filters': filters
         })
 
-    def register_next_step_message_handler_by_chat_id(self, chat_id: str, handler: callable):
+    def register_next_step_message_handler_by_chat_id(self, chat_id: str, handler: Callable):
         """ To register a handler for the next message in a specific chat """
 
         if chat_id in self._next_step_message_handlers_by_chat_id:
@@ -214,8 +217,16 @@ class RubiBot:
             res = requests.post(f"{self.BASE_URL}/getUpdates", json=params, timeout=15)
         except Exception as e:
             raise Exception("Error: {}".format(e))
-        data = res.json()
-        print(data)
+        data: dict = res.json()
+        if data.get('status') != 'OK':
+            if data["status"] == "INVALID_INPUT":
+                raise exceptions.RubiBotInputError(
+                    "invalid inputs {}".format(data.get('dev_message', ''))
+                    )
+            if data["status"] == "INVALID_ACCESS":
+                raise exceptions.RubiBotAccessError("not Access")
+            raise exceptions.RubiBotError(data["status"])
+
         if not res or res.status_code !=200:
             raise Exception("Error in receiving updates: {}".format(data.get("status")))
         
@@ -228,13 +239,13 @@ class RubiBot:
     def __load_offset(self) -> dict:
         if not os.path.exists(self.OFFSET_FILE):
             with open(self.OFFSET_FILE, 'w', encoding="utf-8") as f:
-                json.dump({}, f, ensure_ascii=False, indent=4)
+                json.dump({self.token: None}, f, ensure_ascii=False, indent=4)
 
         with open(self.OFFSET_FILE, "r", encoding="utf-8") as f:
             return json.load(f)
     
-    def __get_offset(self) -> str:
-        return self.__load_offset()[self.token]
+    def __get_offset(self) -> Optional[str]:
+        return self.__load_offset().get(self.token)
 
     def __save_offset(self, offset_id):
         next_offset_ids = self.__load_offset()
@@ -286,10 +297,10 @@ class RubiBot:
                 self.polling(*args, **kwargs)
             except requests.exceptions.JSONDecodeError:
                 print('Json Decode Error')
-                continue
+                time.sleep(1)
             except Exception as e:
                 print(f"{e.__class__.__name__}: {e}")
-                continue
+                time.sleep(1)
 
 
     def send_message(
@@ -350,6 +361,10 @@ class RubiBot:
         if parse_mode:
             parser = parse_mode()
             text, metadata_list = parser(text)
+
+        if not metadata_list:
+            metadata_list = []    
+
         text = self._process_metadata(text, *metadata_list)
 
         return rubika_api.send_message(self.token, chat_id, text, chat_keypad, inline_keypad,
@@ -807,6 +822,7 @@ class RubiBot:
         if filters["content_types"]:
             if message.get_content_type() not in filters["content_types"]:
                 return False
+    
         
         if filters["func"]:
             return filters["func"](message)
@@ -850,22 +866,25 @@ class RubiBot:
 
             if isinstance(upd, rubibot.updates.Update):
                 message = upd.to_message()
-                message.token = self.token
-
-                if message.type == "NewMessage":
-                    if message.chat_id in self._next_step_message_handlers_by_chat_id:
-                        self._next_step_message_handlers_by_chat_id.pop(message.chat_id)(message)
-                        return
-                    elif message.sender_id in self._next_step_message_handlers_by_sender_id:
-                        self._next_step_message_handlers_by_sender_id.pop(message.sender_id)(message)
-                        return
-
-                for handler in self._message_handlers:
-                    if self._test_message_handler(handler, message):
-                        handler["handler"](message)
-                        break
+                if isinstance(message, rubibot.updates.InlineMessage):
+                    upd = rubibot.updates.InlineUpdate(upd.dict)
+                else:   
+                    message.token = self.token
+    
+                    if message.type == "NewMessage":
+                        if message.chat_id in self._next_step_message_handlers_by_chat_id:
+                            self._next_step_message_handlers_by_chat_id.pop(message.chat_id)(message)
+                            return
+                        elif message.sender_id in self._next_step_message_handlers_by_sender_id:
+                            self._next_step_message_handlers_by_sender_id.pop(message.sender_id)(message)
+                            return
+    
+                    for handler in self._message_handlers:
+                        if self._test_message_handler(handler, message):
+                            handler["handler"](message)
+                            break
 
-            elif isinstance(upd, rubibot.updates.InlineUpdate):
+            if isinstance(upd, rubibot.updates.InlineUpdate):
                 message = upd.to_inline_message()
                 message.token = self.token
                 for handler in self._inline_message_handlers:
@@ -875,5 +894,52 @@ class RubiBot:
             else: raise exceptions.RubiBotValueError("Invalid type for update")
 
 
+
+class GroupBot(RubiBot):
+    def __init__(self, token, set_token_env=True):
+        super().__init__(token, set_token_env)
+    
+    def command_handler(self, command: str):
+        def decorator(handler):
+            self.register_message_handler(handler, commands=[command])
+            return handler
+        return decorator
+    
+    def in_text_handler(self, text: str):
+        def decorator(handler):
+            self.register_message_handler(
+                handler,
+                func= lambda m: text in m.text
+            )
+            return handler
+        return decorator
+    
+    def text_equal_handler(self, text: str):
+        def decorator(handler):
+            self.register_message_handler(
+                handler,
+                func= lambda m: m.text==text
+            )
+            return handler
+        return decorator
+    
+    def text_startswith_handler(self, text: str):
+        def decorator(handler):
+            self.register_message_handler(
+                handler,
+                func= lambda m: m.text.startswith(text)
+            )
+            return handler
+        return decorator
+    
+    def text_endswith_handler(self, text: str):
+        def decorator(handler):
+            self.register_message_handler(
+                handler,
+                func= lambda m: m.text.endswith(text)
+            )
+            return handler
+        return decorator
+
 # soon...               
 # This is not the end of our work... 😎

rubibot/exceptions.py

@@ -1,14 +1,14 @@
 class RubiBotError(Exception):
     pass
 
-class RubiBotSyntaxError(Exception):
+class RubiBotSyntaxError(RubiBotError):
     pass
 
-class RubiBotValueError(Exception):
+class RubiBotValueError(RubiBotError):
     pass
 
-class RubiBotInputError(Exception):
+class RubiBotInputError(RubiBotError):
     pass
 
-class RubiBotAccessError(Exception):
+class RubiBotAccessError(RubiBotError):
     pass

rubibot/rubika_api.py

@@ -1,4 +1,3 @@
-import time
 import requests
 from rubibot import exceptions, types, updates
 

rubibot/types.py

@@ -281,7 +281,7 @@ class KeypadTextboxButton(BaseKeypadButton):
             dic['button_textbox']['default_value'] = self.default_value
         if self.title:
             dic['button_textbox']['title'] = self.title
-        print(dic)
+
         return dic
 
 class KeypadRow:
@@ -293,16 +293,12 @@ class KeypadRow:
             self.btns.append(btn.to_dict())
         return self
     
-    def remove(self, btn_or_btn_id):
-        if isinstance(btn_or_btn_id, str):
-            for btn in self.btns:
-                if btn.id == btn_or_btn_id:
-                    buttton = self.btns.pop(btn)
-                    break
-            return buttton
-        if isinstance(btn_or_btn_id, KeypadSimpleButton):
-            return self.btns.pop(btn_or_btn_id)
-        
+    def remove(self, btn_id):
+        if isinstance(btn_id, str):
+            for i, btn in enumerate(self.btns):
+                if btn.get('id') == btn_id:
+                    return self.btns.pop(i)
+        return None
 
     def to_dict(self):
         return {"buttons": self.btns}
@@ -319,9 +315,6 @@ class BaseKeypad:
         for row in rows:
             self.rows.append(row.to_dict())
         return self
-        
-    def remove(self, row):
-        return self.rows.pop(row)
     
     def clear(self):
         self.rows = []

rubibot/updates.py

@@ -1,10 +1,11 @@
+from typing import Optional
 from rubibot import types
 from rubibot import rubika_api
 import json, os
 
 
 class BaseUpdate:
-    def __init__(self, type: str, chat_id: str, update_time: str):
+    def __init__(self, type: str, chat_id: str, update_time: Optional[str]):
         self.type = type
         self.chat_id = chat_id
         self.update_time = update_time
@@ -18,7 +19,15 @@ class BaseUpdate:
             token = os.getenv('RUBIKA_BOT_API_TOKEN')
         return rubika_api.get_chat(token, self.chat_id)
 
-
+    @property
+    def chat_type(self):
+        if self.chat_id.startswith('b'):
+            return 'Private'
+        if self.chat_id.startswith('g'):
+            return 'Group'
+        if self.chat_id.startswith('c'):
+            return 'Channel'
+        return 'NotFound'
 
 class Message(BaseUpdate):
     def __init__(
@@ -78,17 +87,18 @@ class Message(BaseUpdate):
         return f"Message: from {self.chat.get_full_name()}"
 
 
-class InlineMessage:
+class InlineMessage(BaseUpdate):
     def __init__(self, chat_id: str, sender_id: str, message_id: str, text: str, aux_data: types.AuxData,
                  file: types.File, location: types.Location):
-        self.chat_id = chat_id
+        super().__init__('InlineMessage', chat_id, None)
         self.sender_id = sender_id
         self.message_id = message_id
         self.text = text
         self.aux = aux_data
         self.file = file
         self.location = location
-    
+        self.token = None
+
     def has_file(self):
         return self.file is not None
 
@@ -107,6 +117,7 @@ class InlineMessage:
         if self.has_location(): return "location"
 
 
+
 class RemovedMessage(BaseUpdate):
     def __init__(self, type, chat_id, update_time, removed_message_id):
         super().__init__(type, chat_id, update_time)
@@ -221,7 +232,7 @@ class Update:
                 update_time=update_time,
                 removed_message_id=update.get('removed_message_id')
             )
-        elif type_ == "NewMessage":
+        elif type_ == "NewMessage" or type_ == "UpdatedMessage":
 
             new_message: dict = update.get("new_message")
             message_id = new_message.get("message_id")

pyrubikabotapi-1.0.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-rubibot/__init__.py,sha256=dKqsh0sUKsfnykrycQLGnNH-Ba-su2tNRjQDDnFbpl0,35712
-rubibot/exceptions.py,sha256=dtKzNmDm1n3QfiRV6pdz3X3eHsrlLH9oa-0eHtfNaAc,238
-rubibot/metadata.py,sha256=eNUuFnRVQ_Ve8dh-VGMfcX0wxR0d06DiceE-sj1g_oQ,1413
-rubibot/parse.py,sha256=5F-jq0l1mc0q8SHl3iby8dv9N2r4WTf-FzlPesrdzHQ,4837
-rubibot/rubika_api.py,sha256=qAiyqynGvFp6ZItGpb_p08_yNw5FbFZENl0Vwb24IcQ,9356
-rubibot/types.py,sha256=mE-U7pOAbWOEaVKqEuwSZp6a0Ffnl2ce1T07Vru7nRs,10731
-rubibot/updates.py,sha256=n444wsom56FK9_dWpjaS4XTBuTPnKWcj7Vre5rYSvo0,11955
-pyrubikabotapi-1.0.0.dist-info/LICENCE,sha256=mds18XAwr8xSo4pIhwMfB9spNkvrm5KKmLl_KIKQky4,1095
-pyrubikabotapi-1.0.0.dist-info/METADATA,sha256=mQViVjdp9Idk7Axlcbo0nRC-7p6Uaw6xTcF37ZlmCSQ,1458
-pyrubikabotapi-1.0.0.dist-info/WHEEL,sha256=BNRMDyzLkkcmlv0J8ppDQkk2VED33SesJDynr9ED1gc,91
-pyrubikabotapi-1.0.0.dist-info/top_level.txt,sha256=TAhaSYH8WLX-dab28vY3_Wif-mrUw1pmS_RJOFaiogE,8
-pyrubikabotapi-1.0.0.dist-info/RECORD,,