Rubka 4.4.4__py3-none-any.whl → 4.4.8__py3-none-any.whl

rubka/__init__.py

@@ -1,11 +1,10 @@
 from .api import Robot
-from .decorators import on_message
+from .rubino import Bot
 from .exceptions import APIRequestError
-from .keyboards import create_simple_keyboard
 
 __all__ = [
     "Robot",
     "on_message",
     "APIRequestError",
     "create_simple_keyboard",
-]
+]

rubka/adaptorrubka/network/socket.py

@@ -1,4 +1,19 @@
+import sys
+import subprocess
+
+def install_and_import(package_name):
+    try:
+        __import__(package_name)
+    except ModuleNotFoundError:
+        print(f"Module '{package_name}' not found. Installing...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
+    finally:
+        globals()[package_name] = __import__(package_name)
+
+install_and_import("websocket")
+
 from websocket import WebSocketApp
+
 from .helper import Helper
 from json import dumps, loads
 from threading import Thread
@@ -13,4 +28,4 @@ class Socket:
         self.methods = methods
         self.handlers = {}
 
-    ...
+    ...

rubka/api.py

@@ -5,6 +5,10 @@ from .adaptorrubka import Client as Client_get
 from .logger import logger
 from typing import Callable
 from .context import Message,InlineMessage
+from typing import Optional, Union, Literal, Dict, Any
+from pathlib import Path
+import requests
+
 API_URL = "https://botapi.rubika.ir/v3"
 import sys
 import subprocess
@@ -50,23 +54,55 @@ def check_rubka_version():
     latest_version = get_latest_version(package_name)
     if latest_version is None:return
     if installed_version != latest_version:
-        print(f"\n\n⚠️ WARNING: Your installed version of '{package_name}' is outdated!")
-        print(f"Installed version: {installed_version}")
-        print(f"Latest version:    {latest_version}")
-        print(f"Please update it using:\n\npip install --upgrade {package_name}\n")
+        print(f"\n\nWARNING: Your installed version of '{package_name}' is OUTDATED and may cause errors or security risks!")
+        print(f"Installed version : {installed_version}")
+        print(f"Latest available version : {latest_version}")
+        print(f"Please update IMMEDIATELY by running:")
+        print(f"\npip install {package_name}=={latest_version}\n")
+        print(f"Not updating may lead to malfunctions or incompatibility.")
+        print(f"To see new methods : @rubka_library\n\n")
 
 check_rubka_version()
 def show_last_six_words(text):
     text = text.strip()
     return text[-6:]
+import requests
+from pathlib import Path
+from typing import Union, Optional, Dict, Any, Literal
+import tempfile
+import os
 class Robot:
     """
     Main class to interact with Rubika Bot API.
     Initialized with bot token.
     """
 
-    def __init__(self, token: str):
+    def __init__(self, token: str,session_name : str = None,auth : str = None , Key : str = None,platform : str ="web",timeout : int =10):
+        """
+    راه‌اندازی اولیه ربات روبیکا و پیکربندی پارامترهای پایه.
+
+    Parameters:
+        token (str): توکن اصلی ربات برای احراز هویت با Rubika Bot API. این مقدار الزامی است.
+        session_name (str, optional): نام نشست دلخواه برای شناسایی یا جداسازی کلاینت‌ها. پیش‌فرض None.
+        auth (str, optional): مقدار احراز هویت اضافی برای اتصال به کلاینت سفارشی. پیش‌فرض None.
+        Key (str, optional): کلید اضافی برای رمزنگاری یا تأیید. در صورت نیاز استفاده می‌شود. پیش‌فرض None.
+        platform (str, optional): پلتفرم اجرایی مورد نظر (مثل "web" یا "android"). پیش‌فرض "web".
+        timeout (int, optional): مدت‌زمان تایم‌اوت برای درخواست‌های HTTP بر حسب ثانیه. پیش‌فرض 10 ثانیه.
+
+    توضیحات:
+        - این تابع یک شیء Session از requests می‌سازد برای استفاده در تمام درخواست‌ها.
+        - هندلرهای مختلف مانند پیام، دکمه، کوئری اینلاین و ... در این مرحله مقداردهی اولیه می‌شوند.
+        - لیست `self._callback_handlers` برای مدیریت چندین callback مختلف الزامی است.
+
+    مثال:
+        >>> bot = Robot(token="BOT_TOKEN", platform="android", timeout=15)
+    """
         self.token = token
+        self.timeout = timeout
+        self.auth = auth
+        self.session_name = session_name
+        self.Key = Key
+        self.platform = platform
         self._offset_id = None
         self.session = requests.Session()
         self.sessions: Dict[str, Dict[str, Any]] = {}
@@ -82,7 +118,7 @@ class Robot:
     def _post(self, method: str, data: Dict[str, Any]) -> Dict[str, Any]:
         url = f"{API_URL}/{self.token}/{method}"
         try:
-            response = self.session.post(url, json=data, timeout=10)
+            response = self.session.post(url, json=data, timeout=self.timeout)
             response.raise_for_status()
             try:
                 json_resp = response.json()
@@ -236,8 +272,10 @@ class Robot:
         return self._post("sendMessage", payload)
 
     def _get_client(self):
-        return Client_get(show_last_six_words(self.token))
-
+        if self.session_name:
+            return Client_get(self.session_name,self.auth,self.Key,self.platform)
+        else :
+            return Client_get(show_last_six_words(self.token),self.auth,self.Key,self.platform)
     from typing import Union
 
     def check_join(self, channel_guid: str, chat_id: str = None) -> Union[bool, list[str]]:
@@ -332,6 +370,211 @@ class Robot:
     def get_chat(self, chat_id: str) -> Dict[str, Any]:
         """Get chat info."""
         return self._post("getChat", {"chat_id": chat_id})
+    def upload_media_file(self, upload_url: str, name: str, path: Union[str, Path]) -> str:
+        # بررسی اینکه ورودی URL است یا مسیر محلی
+        is_temp_file = False
+        if isinstance(path, str) and path.startswith("http"):
+            response = requests.get(path)
+            if response.status_code != 200:
+                raise Exception(f"Failed to download file from URL ({response.status_code})")
+            temp_file = tempfile.NamedTemporaryFile(delete=False)
+            temp_file.write(response.content)
+            temp_file.close()
+            path = temp_file.name
+            is_temp_file = True
+
+        with open(path, 'rb') as file:
+            files = {
+                'file': (name, file, 'application/octet-stream')
+            }
+            response = requests.post(upload_url, files=files)
+            if response.status_code != 200:
+                raise Exception(f"Upload failed ({response.status_code}): {response.text}")
+            data = response.json()
+
+        if is_temp_file:
+            os.unlink(path)  # حذف فایل موقت
+
+        return data.get('data', {}).get('file_id')
+
+    def get_upload_url(self, media_type: Literal['File', 'Image', 'Voice', 'Music', 'Gif']) -> str:
+        allowed = ['File', 'Image', 'Voice', 'Music', 'Gif']
+        if media_type not in allowed:
+            raise ValueError(f"Invalid media type. Must be one of {allowed}")
+        result = self._post("requestSendFile", {"type": media_type})
+        return result.get("data", {}).get("upload_url")
+    def _send_uploaded_file(
+        self,
+        chat_id: str,
+        file_id: str,
+        text: Optional[str] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        disable_notification: bool = False,
+        reply_to_message_id: Optional[str] = None,
+        chat_keypad_type: Optional[Literal["New", "Removed", "None"]] = "None"
+    ) -> Dict[str, Any]:
+        payload = {
+            "chat_id": chat_id,
+            "file_id": file_id,
+            "text": text,
+            "disable_notification": disable_notification,
+            "chat_keypad_type": chat_keypad_type,
+        }
+        if chat_keypad:
+            payload["chat_keypad"] = chat_keypad
+        if inline_keypad:
+            payload["inline_keypad"] = inline_keypad
+        if reply_to_message_id:
+            payload["reply_to_message_id"] = str(reply_to_message_id)
+
+        return self._post("sendFile", payload)
+
+    def send_document(
+        self,
+        chat_id: str,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        file_name: Optional[str] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        reply_to_message_id: Optional[str] = None,
+        disable_notification: bool = False,
+        chat_keypad_type: Optional[Literal["New", "Removed", "None"]] = "New"
+    ) -> Dict[str, Any]:
+        if path:
+            file_name = file_name or Path(path).name
+            upload_url = self.get_upload_url("File")
+            file_id = self.upload_media_file(upload_url, file_name, path)
+        if not file_id:
+            raise ValueError("Either path or file_id must be provided.")
+        return self._send_uploaded_file(
+            chat_id=chat_id,
+            file_id=file_id,
+            text=text,
+            inline_keypad=inline_keypad,
+            chat_keypad=chat_keypad,
+            reply_to_message_id=reply_to_message_id,
+            disable_notification=disable_notification,
+            chat_keypad_type=chat_keypad_type
+        )
+    def send_music(
+        self,
+        chat_id: str,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        file_name: Optional[str] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        reply_to_message_id: Optional[str] = None,
+        disable_notification: bool = False,
+        chat_keypad_type: Optional[Literal["New", "Removed", "None"]] = "New"
+    ) -> Dict[str, Any]:
+        if path:
+            file_name = file_name or Path(path).name
+            upload_url = self.get_upload_url("Music")
+            file_id = self.upload_media_file(upload_url, file_name, path)
+        if not file_id:
+            raise ValueError("Either path or file_id must be provided.")
+        return self._send_uploaded_file(
+            chat_id=chat_id,
+            file_id=file_id,
+            text=text,
+            inline_keypad=inline_keypad,
+            chat_keypad=chat_keypad,
+            reply_to_message_id=reply_to_message_id,
+            disable_notification=disable_notification,
+            chat_keypad_type=chat_keypad_type
+        )
+    def send_voice(
+        self,
+        chat_id: str,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        file_name: Optional[str] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        reply_to_message_id: Optional[str] = None,
+        disable_notification: bool = False,
+        chat_keypad_type: Optional[Literal["New", "Removed", "None"]] = "New"
+    ) -> Dict[str, Any]:
+        if path:
+            file_name = file_name or Path(path).name
+            upload_url = self.get_upload_url("Voice")
+            file_id = self.upload_media_file(upload_url, file_name, path)
+        if not file_id:
+            raise ValueError("Either path or file_id must be provided.")
+        return self._send_uploaded_file(
+            chat_id=chat_id,
+            file_id=file_id,
+            text=text,
+            inline_keypad=inline_keypad,
+            chat_keypad=chat_keypad,
+            reply_to_message_id=reply_to_message_id,
+            disable_notification=disable_notification,
+            chat_keypad_type=chat_keypad_type
+        )
+    def send_image(
+        self,
+        chat_id: str,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        file_name: Optional[str] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        reply_to_message_id: Optional[str] = None,
+        disable_notification: bool = False,
+        chat_keypad_type: Optional[Literal["New", "Removed", "None"]] = "New"
+    ) -> Dict[str, Any]:
+        if path:
+            file_name = file_name or Path(path).name
+            upload_url = self.get_upload_url("Image")
+            file_id = self.upload_media_file(upload_url, file_name, path)
+        if not file_id:
+            raise ValueError("Either path or file_id must be provided.")
+        return self._send_uploaded_file(
+            chat_id=chat_id,
+            file_id=file_id,
+            text=text,
+            inline_keypad=inline_keypad,
+            chat_keypad=chat_keypad,
+            reply_to_message_id=reply_to_message_id,
+            disable_notification=disable_notification,
+            chat_keypad_type=chat_keypad_type
+        )
+    def send_gif(
+        self,
+        chat_id: str,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        file_name: Optional[str] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        reply_to_message_id: Optional[str] = None,
+        disable_notification: bool = False,
+        chat_keypad_type: Optional[Literal["New", "Removed", "None"]] = "New"
+    ) -> Dict[str, Any]:
+        if path:
+            file_name = file_name or Path(path).name
+            upload_url = self.get_upload_url("Gif")
+            file_id = self.upload_media_file(upload_url, file_name, path)
+        if not file_id:
+            raise ValueError("Either path or file_id must be provided.")
+        return self._send_uploaded_file(
+            chat_id=chat_id,
+            file_id=file_id,
+            text=text,
+            inline_keypad=inline_keypad,
+            chat_keypad=chat_keypad,
+            reply_to_message_id=reply_to_message_id,
+            disable_notification=disable_notification,
+            chat_keypad_type=chat_keypad_type
+        )
 
     def get_updates(
         self,
@@ -381,7 +624,7 @@ class Robot:
         inline_keypad: Dict[str, Any]
     ) -> Dict[str, Any]:
         """Edit inline keypad of a message."""
-        return self._post("editInlineKeypad", {
+        return self._post("editMessageKeypad", {
             "chat_id": chat_id,
             "message_id": message_id,
             "inline_keypad": inline_keypad

rubka/context.py

@@ -184,6 +184,9 @@ class Bot:
         self.username: str = data.get("username")
         self.start_message: str = data.get("start_message")
         self.share_url: str = data.get("share_url")
+from typing import Union
+from pathlib import Path
+
 class Message:
     def __init__(self, bot, chat_id, message_id, sender_id, text=None, raw_data=None):
         self.bot = bot
@@ -227,6 +230,117 @@ class Message:
             "reply_to_message_id": self.message_id,
             **kwargs
         })
+    
+
+    def reply_document(
+        self,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad_type: Optional[str] = "None",
+        disable_notification: bool = False
+    ):
+        return self.bot.send_document(
+            chat_id=self.chat_id,
+            path=path,
+            file_id=file_id,
+            text=text,
+            chat_keypad=chat_keypad,
+            inline_keypad=inline_keypad,
+            chat_keypad_type=chat_keypad_type,
+            disable_notification=disable_notification,
+            reply_to_message_id=self.message_id
+        )
+
+    def reply_image(
+        self,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad_type: Optional[str] = "None",
+        disable_notification: bool = False
+    ):
+        return self.bot.send_image(
+            chat_id=self.chat_id,
+            path=path,
+            file_id=file_id,
+            text=text,
+            chat_keypad=chat_keypad,
+            inline_keypad=inline_keypad,
+            chat_keypad_type=chat_keypad_type,
+            disable_notification=disable_notification,
+            reply_to_message_id=self.message_id
+        )
+
+    def reply_music(
+        self,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad_type: Optional[str] = "None",
+        disable_notification: bool = False
+    ):
+        return self.bot.send_music(
+            chat_id=self.chat_id,
+            path=path,
+            file_id=file_id,
+            text=text,
+            chat_keypad=chat_keypad,
+            inline_keypad=inline_keypad,
+            chat_keypad_type=chat_keypad_type,
+            disable_notification=disable_notification,
+            reply_to_message_id=self.message_id
+        )
+
+    def reply_voice(
+        self,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad_type: Optional[str] = "None",
+        disable_notification: bool = False
+    ):
+        return self.bot.send_voice(
+            chat_id=self.chat_id,
+            path=path,
+            file_id=file_id,
+            text=text,
+            chat_keypad=chat_keypad,
+            inline_keypad=inline_keypad,
+            chat_keypad_type=chat_keypad_type,
+            disable_notification=disable_notification,
+            reply_to_message_id=self.message_id
+        )
+
+    def reply_gif(
+        self,
+        path: Optional[Union[str, Path]] = None,
+        file_id: Optional[str] = None,
+        text: Optional[str] = None,
+        chat_keypad: Optional[Dict[str, Any]] = None,
+        inline_keypad: Optional[Dict[str, Any]] = None,
+        chat_keypad_type: Optional[str] = "None",
+        disable_notification: bool = False
+    ):
+        return self.bot.send_gif(
+            chat_id=self.chat_id,
+            path=path,
+            file_id=file_id,
+            text=text,
+            chat_keypad=chat_keypad,
+            inline_keypad=inline_keypad,
+            chat_keypad_type=chat_keypad_type,
+            disable_notification=disable_notification,
+            reply_to_message_id=self.message_id
+        )
 
     def reply_location(self, latitude: str, longitude: str, **kwargs) -> Dict[str, Any]:
         return self.bot.send_location(