Rubka 1.3.0__tar.gz → 1.5.0__tar.gz

{rubka-1.3.0 → rubka-1.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Rubka
-Version: 1.3.0
+Version: 1.5.0
 Summary: A Python library for interacting with Rubika Bot API.
 Home-page: https://github.com/Mahdy-Ahmadi/Rubka
 Download-URL: https://github.com/Mahdy-Ahmadi/Rubka/archive/refs/tags/v0.1.0.tar.gz

{rubka-1.3.0 → rubka-1.5.0}/Rubka.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Rubka
-Version: 1.3.0
+Version: 1.5.0
 Summary: A Python library for interacting with Rubika Bot API.
 Home-page: https://github.com/Mahdy-Ahmadi/Rubka
 Download-URL: https://github.com/Mahdy-Ahmadi/Rubka/archive/refs/tags/v0.1.0.tar.gz

{rubka-1.3.0 → rubka-1.5.0}/Rubka.egg-info/SOURCES.txt

@@ -8,8 +8,11 @@ Rubka.egg-info/top_level.txt
 rubka/__init__.py
 rubka/api.py
 rubka/config.py
+rubka/context.py
 rubka/decorators.py
 rubka/exceptions.py
+rubka/jobs.py
 rubka/keyboards.py
+rubka/keypad.py
 rubka/logger.py
 rubka/utils.py

{rubka-1.3.0 → rubka-1.5.0}/rubka/api.py

@@ -2,7 +2,8 @@ import requests
 from typing import List, Optional, Dict, Any, Literal
 from .exceptions import APIRequestError
 from .logger import logger
-
+from typing import Callable
+from .context import MessageContext
 API_URL = "https://botapi.rubika.ir/v3"
 
 class Robot:
@@ -13,7 +14,9 @@ class Robot:
 
     def __init__(self, token: str):
         self.token = token
+        self._offset_id = None
         self.session = requests.Session()
+        self.sessions: Dict[str, Dict[str, Any]] = {}
         logger.info(f"Initialized RubikaBot with token: {token[:8]}***")
 
     def _post(self, method: str, data: Dict[str, Any]) -> Dict[str, Any]:
@@ -31,7 +34,63 @@ class Robot:
     def get_me(self) -> Dict[str, Any]:
         """Get info about the bot itself."""
         return self._post("getMe", {})
+    def on_message(self, filters: Optional[Callable[[MessageContext], bool]] = None, commands: Optional[List[str]] = None):
+        def decorator(func: Callable[[Any, MessageContext], None]):
+            self._message_handler = {
+                "func": func,
+                "filters": filters,
+                "commands": commands
+            }
+            return func
+        return decorator
+
+
+
+    def _process_update(self, update: Dict[str, Any]):
+        update_data = update.get('update', {})
+        if update_data.get('type') == 'NewMessage':
+            msg = update_data.get('new_message', {})
+            chat_id = update_data.get('chat_id')
+            message_id = msg.get('message_id')
+            sender_id = msg.get('sender_id')
+            text = msg.get('text')
+
+            if self._message_handler:
+                handler = self._message_handler
+                context = MessageContext(self, chat_id, message_id, sender_id, text)
+
+
+                if handler["commands"]:
+                    if not context.text or not context.text.startswith("/"):
+                        return
+                    parts = context.text.split()
+                    cmd = parts[0][1:]
+                    if cmd not in handler["commands"]:
+                        return
+                    context.args = parts[1:]
+
+                if handler["filters"]:
+                    if not handler["filters"](context):
+                        return
+
+                handler["func"](self, context)
+
+
+
 
+    def run(self):
+        print("Bot started running...")
+        while True:
+            try:
+                updates = self.get_updates(offset_id=self._offset_id, limit=10)
+                if updates and updates.get('data'):
+                    for update in updates['data']:
+                        self._process_update(update)
+                        self._offset_id = update.get('update_id', self._offset_id)
+            except Exception as e:
+                print(f"Error in run loop: {e}")
+                logger.error(f"API request failed: {e}")
+                raise APIRequestError(f"API request failed: {e}") from e
     def send_message(
         self,
         chat_id: str,

rubka-1.5.0/rubka/context.py

@@ -0,0 +1,99 @@
+from typing import Any, Dict, List
+
+
+class Message:
+    def __init__(self, bot, chat_id, message_id, sender_id, text):
+        self.bot = bot
+        self.chat_id = chat_id
+        self.message_id = message_id
+        self.sender_id = sender_id
+        self.text = text
+        self.args = []
+    @property
+    def session(self):
+        if self.chat_id not in self.bot.sessions:
+            self.bot.sessions[self.chat_id] = {}
+        return self.bot.sessions[self.chat_id]
+    def reply(self, text: str, **kwargs):
+        return self.bot.send_message(
+            self.chat_id,
+            text,
+            reply_to_message_id=self.message_id,
+            **kwargs
+        )
+
+    def reply_poll(self, question: str, options: List[str], **kwargs) -> Dict[str, Any]:
+        return self.bot._post("sendPoll", {
+            "chat_id": self.chat_id,
+            "question": question,
+            "options": options,
+            "reply_to_message_id": self.message_id,
+            **kwargs
+        })
+
+    def reply_location(self, latitude: str, longitude: str, **kwargs) -> Dict[str, Any]:
+        return self.bot.send_location(
+            chat_id=self.chat_id,
+            latitude=latitude,
+            longitude=longitude,
+            reply_to_message_id=self.message_id,
+            **kwargs
+        )
+
+    def reply_contact(self, first_name: str, last_name: str, phone_number: str, **kwargs) -> Dict[str, Any]:
+        return self.bot.send_contact(
+            chat_id=self.chat_id,
+            first_name=first_name,
+            last_name=last_name,
+            phone_number=phone_number,
+            reply_to_message_id=self.message_id,
+            **kwargs
+        )
+
+    def reply_keypad(self, text: str, keypad: Dict[str, Any], **kwargs) -> Dict[str, Any]:
+        return self.bot.send_message(
+            chat_id=self.chat_id,
+            text=text,
+            chat_keypad_type="New",
+            chat_keypad=keypad,
+            reply_to_message_id=self.message_id,
+            **kwargs
+        )
+
+    def reply_inline(self, text: str, inline_keypad: Dict[str, Any], **kwargs) -> Dict[str, Any]:
+        return self.bot.send_message(
+            chat_id=self.chat_id,
+            text=text,
+            inline_keypad=inline_keypad,
+            reply_to_message_id=self.message_id,
+            **kwargs
+        )
+
+    def reply_sticker(self, sticker_id: str, **kwargs) -> Dict[str, Any]:
+        return self.bot._post("sendSticker", {
+            "chat_id": self.chat_id,
+            "sticker_id": sticker_id,
+            "reply_to_message_id": self.message_id,
+            **kwargs
+        })
+
+    def reply_file(self, file_id: str, **kwargs) -> Dict[str, Any]:
+        return self.bot._post("sendFile", {
+            "chat_id": self.chat_id,
+            "file_id": file_id,
+            "reply_to_message_id": self.message_id,
+            **kwargs
+        })
+
+    def edit(self, new_text: str) -> Dict[str, Any]:
+        return self.bot.edit_message_text(
+            chat_id=self.chat_id,
+            message_id=self.message_id,
+            text=new_text
+        )
+
+    def delete(self) -> Dict[str, Any]:
+        return self.bot.delete_message(
+            chat_id=self.chat_id,
+            message_id=self.message_id
+        )

rubka-1.5.0/rubka/jobs.py

@@ -0,0 +1,15 @@
+import threading
+import time
+from typing import Callable
+
+class Job:
+    def __init__(self, delay: int, callback: Callable):
+        self.delay = delay
+        self.callback = callback
+        thread = threading.Thread(target=self.run)
+        thread.daemon = True
+        thread.start()
+
+    def run(self):
+        time.sleep(self.delay)
+        self.callback()

rubka-1.5.0/rubka/keypad.py

@@ -0,0 +1,15 @@
+from typing import Dict
+
+class InlineBuilder:
+    def __init__(self):
+        self.rows = []
+
+    def row(self, *buttons: Dict[str, str]):
+        self.rows.append({"buttons": list(buttons)})
+        return self
+
+    def button(self, id: str, text: str, type: str = "Simple") -> Dict[str, str]:
+        return {"id": id, "type": type, "button_text": text}
+
+    def build(self):
+        return {"rows": self.rows}

{rubka-1.3.0 → rubka-1.5.0}/setup.py

@@ -8,7 +8,7 @@ except FileNotFoundError:
 
 setup(
     name='Rubka',
-    version='1.3.0',
+    version='1.5.0',
     description='A Python library for interacting with Rubika Bot API.',
     long_description=long_description,
     long_description_content_type='text/markdown',