mm-telegram 0.1.2__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-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-0.1.2.dist-info/METADATA

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

mm_telegram-0.1.2.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=nbBWG58PjKU74NYqxwMANBLax0HJqgJ0B6r3FOAeVxg,2285
-mm_telegram-0.1.2.dist-info/METADATA,sha256=SX9nfEVX-V_8fdzTahfrY_RuDRMpNR0yq_BXugWXDd0,109
-mm_telegram-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mm_telegram-0.1.2.dist-info/RECORD,,