admin/cyclop
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #100 from bucolucas/add-reboot-command

Browse Source 
Add /reboot command to exit with status code 355
This commit is contained in:
bucolucas
2024-08-20 12:38:10 -05:00
committed by GitHub
parent 7d651bbbc0 96530e50fb
commit 1f51b2ebbf
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
telegram_helper.py
+7
View File
@@ -1,5 +1,6 @@
import os import os
import logging import logging
import sys
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes, CallbackQueryHandler from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes, CallbackQueryHandler
@@ -70,6 +71,11 @@ class TelegramHelper:
result = await self.bot.abort_processing(user_id) result = await self.bot.abort_processing(user_id)
await query.edit_message_text(text=result) await query.edit_message_text(text=result)
async def reboot(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
await update.message.reply_text("Rebooting the bot...")
logging.info("Received reboot command. Exiting with status code 355.")
sys.exit(355)
def run(self): def run(self):
application = Application.builder().token(self.telegram_bot_token).build() application = Application.builder().token(self.telegram_bot_token).build()
@@ -77,6 +83,7 @@ class TelegramHelper:
application.add_handler(CommandHandler("clear", self.clear)) application.add_handler(CommandHandler("clear", self.clear))
application.add_handler(CommandHandler("switch", self.switch)) application.add_handler(CommandHandler("switch", self.switch))
application.add_handler(CommandHandler("status", self.status)) application.add_handler(CommandHandler("status", self.status))
application.add_handler(CommandHandler("reboot", self.reboot))
application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message)) application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message))
application.add_handler(CallbackQueryHandler(self.abort_processing, pattern='^abort$')) application.add_handler(CallbackQueryHandler(self.abort_processing, pattern='^abort$'))