Cleanup types

smsbot/telegram.py

@@ -12,9 +12,7 @@ from telegram.ext import (
 
 from smsbot.utils import get_smsbot_version
 
-REQUEST_TIME = Summary(
-    "telegram_request_processing_seconds", "Time spent processing request"
-)
+REQUEST_TIME = Summary("telegram_request_processing_seconds", "Time spent processing request")
 COMMAND_COUNT = Counter("telegram_command_count", "Total number of commands processed")
 
 
@@ -35,21 +33,18 @@ class TelegramSmsBot:
 
     async def callback(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """Handle the update"""
-        if update.effective_user.id in self.owners:
-            self.logger.info(
-                f"{update.effective_user.username} sent {update.message.text}"
-            )
-            COMMAND_COUNT.inc()
-        else:
-            self.logger.debug(f"Ignoring message from user {update.effective_user.id}")
-            raise ApplicationHandlerStop
+        if update.effective_user and update.message:
+            if update.effective_user.id in self.owners:
+                self.logger.info(f"{update.effective_user.username} sent {update.message.text}")
+                COMMAND_COUNT.inc()
+            else:
+                self.logger.debug(f"Ignoring message from user {update.effective_user.username}")
+                raise ApplicationHandlerStop
 
     async def send_message(self, chat_id: int, text: str):
         """Send a message to a specific chat"""
         self.logger.info(f"Sending message to chat {chat_id}: {text}")
-        await self.app.bot.send_message(
-            chat_id=chat_id, text=text, parse_mode="MarkdownV2"
-        )
+        await self.app.bot.send_message(chat_id=chat_id, text=text, parse_mode="MarkdownV2")
 
     async def send_subscribers(self, text: str):
         """Send a message to all subscribers"""
@@ -64,48 +59,41 @@ class TelegramSmsBot:
             await self.send_message(owner, text)
 
     @REQUEST_TIME.time()
-    async def handler_help(self, update, context):
+    async def handler_help(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """Send a message when the command /help is issued."""
-        self.logger.info("/help command received in chat: %s", update.message.chat)
+        if update.message:
+            self.logger.info("/help command received in chat: %s", update.message.chat)
 
-        commands = []
-        for command in self.app.handlers[0]:
-            if isinstance(command, CommandHandler):
-                commands.extend(["/{0}".format(cmd) for cmd in command.commands])
+            commands = []
+            for command in self.app.handlers[0]:
+                if isinstance(command, CommandHandler):
+                    commands.extend(["/{0}".format(cmd) for cmd in command.commands])
 
-        await update.message.reply_markdown(
-            "Smsbot v{0}\n\n{1}".format(get_smsbot_version(), "\n".join(commands))
-        )
-        COMMAND_COUNT.inc()
+            await update.message.reply_markdown("Smsbot v{0}\n\n{1}".format(get_smsbot_version(), "\n".join(commands)))
+            COMMAND_COUNT.inc()
 
     @REQUEST_TIME.time()
-    async def handler_subscribe(
-        self, update: Update, context: ContextTypes.DEFAULT_TYPE
-    ):
+    async def handler_subscribe(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """Handle subscription requests"""
-        user_id = update.effective_user.id
-        if user_id not in self.subscribers:
-            self.subscribers.append(user_id)
-            self.logger.info(f"User {user_id} subscribed.")
-            self.logger.info(f"Current subscribers: {self.subscribers}")
-            await update.message.reply_markdown(
-                "You have successfully subscribed to updates."
-            )
-        else:
-            self.logger.info(f"User {user_id} is already subscribed.")
+        if update.effective_user and update.message:
+            user_id = update.effective_user.id
+            if user_id not in self.subscribers:
+                self.subscribers.append(user_id)
+                self.logger.info(f"User {user_id} subscribed.")
+                self.logger.info(f"Current subscribers: {self.subscribers}")
+                await update.message.reply_markdown("You have successfully subscribed to updates.")
+            else:
+                self.logger.info(f"User {user_id} is already subscribed.")
 
     @REQUEST_TIME.time()
-    async def handler_unsubscribe(
-        self, update: Update, context: ContextTypes.DEFAULT_TYPE
-    ):
+    async def handler_unsubscribe(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """Handle unsubscription requests"""
-        user_id = update.effective_user.id
-        if user_id in self.subscribers:
-            self.subscribers.remove(user_id)
-            self.logger.info(f"User {user_id} unsubscribed.")
-            self.logger.info(f"Current subscribers: {self.subscribers}")
-            await update.message.reply_markdown(
-                "You have successfully unsubscribed from updates."
-            )
-        else:
-            self.logger.info(f"User {user_id} is not subscribed.")
+        if update.effective_user and update.message:
+            user_id = update.effective_user.id
+            if user_id in self.subscribers:
+                self.subscribers.remove(user_id)
+                self.logger.info(f"User {user_id} unsubscribed.")
+                self.logger.info(f"Current subscribers: {self.subscribers}")
+                await update.message.reply_markdown("You have successfully unsubscribed from updates.")
+            else:
+                self.logger.info(f"User {user_id} is not subscribed.")