mm-telegram 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

mm_telegram/__init__.py

@@ -1 +1,4 @@
-from .simple_message import send_message as send_message
+from .bot import TelegramBot, TelegramHandler
+from .message import send_message
+
+__all__ = ["TelegramBot", "TelegramHandler", "send_message"]

mm_telegram/bot.py

@@ -0,0 +1,81 @@
+import logging
+from typing import Any
+
+from telegram import Update
+from telegram.ext import (
+    Application,
+    ApplicationBuilder,
+    ApplicationHandlerStop,
+    BaseHandler,
+    CallbackContext,
+    CommandHandler,
+    ContextTypes,
+    ExtBot,
+    MessageHandler,
+    filters,
+)
+
+logger = logging.getLogger(__name__)
+
+
+type TelegramHandler = BaseHandler[Any, CallbackContext[ExtBot[None], dict[Any, Any], dict[Any, Any], dict[Any, Any]], Any]
+
+
+async def ping(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    if update.effective_chat is not None:
+        await context.bot.send_message(chat_id=update.effective_chat.id, text="pong")
+
+
+async def unknown(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    if update.effective_chat is not None:
+        await context.bot.send_message(chat_id=update.effective_chat.id, text="Sorry, I didn't understand that command.")
+
+
+async def is_admin(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    admins: list[int] = context.bot_data.get("admins", [])
+
+    if update.effective_user is None or update.message is None:
+        raise ApplicationHandlerStop
+
+    if update.effective_user.id not in admins:
+        logger.warning("is not admin", extra={"telegram_user_id": update.effective_user.id})
+        await update.message.reply_text("Who are you?")
+        raise ApplicationHandlerStop
+
+
+class TelegramBot:
+    app: Application[Any, Any, Any, Any, Any, Any] | None
+
+    def __init__(self, handlers: list[TelegramHandler], bot_data: dict[str, object]) -> None:
+        self.handlers = handlers
+        self.bot_data = bot_data
+        self.app = None
+
+    async def start(self, token: str, admins: list[int]) -> None:
+        if not admins:
+            raise ValueError("No admins provided")
+        logger.debug("Starting telegram bot...")
+        app = ApplicationBuilder().token(token).build()
+        for key, value in self.bot_data.items():
+            app.bot_data[key] = value
+        app.bot_data["admins"] = admins
+
+        for handler in self.handlers:
+            app.add_handler(handler)
+
+        app.add_handler(CommandHandler("ping", ping))
+        app.add_handler(MessageHandler(filters.COMMAND, unknown))
+
+        await app.initialize()
+        await app.start()
+        if app.updater is not None:
+            await app.updater.start_polling()
+            logger.debug("Telegram bot started.")
+
+        self.app = app
+
+    async def shutdown(self) -> None:
+        if self.app is not None:
+            await self.app.shutdown()
+            self.app = None
+            logger.debug("Telegram bot stopped.")

mm_telegram/message.py

@@ -0,0 +1,62 @@
+import asyncio
+
+from mm_std import Result, http_request
+
+
+async def send_message(
+    bot_token: str,
+    chat_id: int,
+    message: str,
+    timeout: float = 5,
+    inter_message_delay_seconds: int = 3,
+) -> Result[list[int]]:
+    """
+    Sends a message to a Telegram chat.
+
+    If the message exceeds the Telegram character limit (4096),
+    it will be split into multiple messages and sent sequentially
+    with a delay between each part.
+
+    Args:
+        bot_token: The Telegram bot token.
+        chat_id: The target chat ID.
+        message: The message text to send.
+        timeout: The HTTP request timeout in seconds. Defaults to 5.
+        inter_message_delay_seconds: The delay in seconds between sending
+            parts of a long message. Defaults to 3.
+
+    Returns:
+        A Result object containing a list of message IDs for the sent messages
+        on success, or an error details on failure. The 'extra' field in the
+        Result contains the raw responses from the Telegram API.
+    """
+    messages = _split_string(message, 4096)
+    responses = []
+    result_message_ids = []
+    while True:
+        text_part = messages.pop(0)
+        params = {"chat_id": chat_id, "text": text_part}
+        res = await http_request(
+            f"https://api.telegram.org/bot{bot_token}/sendMessage", method="post", json=params, timeout=timeout
+        )
+        responses.append(res.to_dict())
+        if res.is_err():
+            return Result.err(res.error or "error sending message", extra={"responses": responses})
+
+        message_id = res.parse_json_body("result.message_id", none_on_error=True)
+        if message_id:
+            result_message_ids.append(message_id)
+        else:
+            # Log the unexpected response for debugging?
+            return Result.err("unknown_response_structure", extra={"responses": responses})
+
+        if len(messages):
+            await asyncio.sleep(inter_message_delay_seconds)
+        else:
+            break
+    return Result.ok(result_message_ids, extra={"responses": responses})
+
+
+def _split_string(text: str, chars_per_string: int) -> list[str]:
+    """Splits a string into a list of strings, each with a maximum length."""
+    return [text[i : i + chars_per_string] for i in range(0, len(text), chars_per_string)]

mm_telegram-0.2.0.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: mm-telegram
+Version: 0.2.0
+Requires-Python: >=3.12
+Requires-Dist: mm-std~=0.4.13
+Requires-Dist: python-telegram-bot~=22.1

mm_telegram-0.2.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+mm_telegram/__init__.py,sha256=MfDt-Y42ew8LaQW34MElvgrpkPcs_RWCW_coQSyFYOE,142
+mm_telegram/bot.py,sha256=ZQeKONCiFkmuinCpgvbfYc1clvd2lB26YR9NAuL8eMs,2614
+mm_telegram/message.py,sha256=nbBWG58PjKU74NYqxwMANBLax0HJqgJ0B6r3FOAeVxg,2285
+mm_telegram/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mm_telegram-0.2.0.dist-info/METADATA,sha256=o0WxxaffFumRugRsp9zfbrXpNy8-fUn0QipGaCupXjw,150
+mm_telegram-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mm_telegram-0.2.0.dist-info/RECORD,,

mm_telegram/simple_message.py

@@ -1,32 +0,0 @@
-import asyncio
-
-from mm_std import Result, http_request
-
-
-async def send_message(bot_token: str, chat_id: int, message: str, long_message_delay: int = 3) -> Result[list[int]]:
-    messages = _split_string(message, 4096)
-    responses = []
-    result = []
-    while True:
-        text = messages.pop(0)
-        params = {"chat_id": chat_id, "text": text}
-        res = await http_request(f"https://api.telegram.org/bot{bot_token}/sendMessage", method="post", json=params)
-        responses.append(res.to_dict())
-        if res.is_err():
-            return Result.err(res.error or "error", extra={"responses": [responses]})
-
-        message_id = res.parse_json_body("result.message_id", none_on_error=True)
-        if message_id:
-            result.append(message_id)
-        else:
-            return Result.err("unknown_response", extra={"responses": responses})
-
-        if len(messages):
-            await asyncio.sleep(long_message_delay)
-        else:
-            break
-    return Result.ok(result, extra={"responses": responses})
-
-
-def _split_string(text: str, chars_per_string: int) -> list[str]:
-    return [text[i : i + chars_per_string] for i in range(0, len(text), chars_per_string)]

mm_telegram-0.1.1.dist-info/METADATA

@@ -1,5 +0,0 @@
-Metadata-Version: 2.4
-Name: mm-telegram
-Version: 0.1.1
-Requires-Python: >=3.12
-Requires-Dist: mm-std~=0.4.8

mm_telegram-0.1.1.dist-info/RECORD

@@ -1,6 +0,0 @@
-mm_telegram/__init__.py,sha256=Nl5I50id12u28Q0H_uHziyd2t_bjySeZAH1etBOVW3A,57
-mm_telegram/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mm_telegram/simple_message.py,sha256=fT0zhmUPqq_w2EvTdOKLC0JhVMh3nZmthApY8dmY_fk,1193
-mm_telegram-0.1.1.dist-info/METADATA,sha256=8Iusxz_r-QCJMu7dN8BNW9VN2EFNUZBxO3G0NwNYnEc,108
-mm_telegram-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_telegram-0.1.1.dist-info/RECORD,,