Add functionality to check and send message if semaphore file exists on initialization

This commit is contained in:
2024-08-20 13:24:45 -05:00
parent 7783d8fc22
commit 7807a6eac7
+13
View File
@@ -8,6 +8,9 @@ class TelegramHelper:
def __init__(self, bot): def __init__(self, bot):
self.bot = bot self.bot = bot
self.telegram_bot_token = os.getenv('TELEGRAM_BOT_TOKEN') self.telegram_bot_token = os.getenv('TELEGRAM_BOT_TOKEN')
# Call the method to check for reboot file and send message
self._check_and_send_reboot_message()
async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
await self.bot.start() await self.bot.start()
@@ -93,3 +96,13 @@ class TelegramHelper:
logging.info("Bot is running...") logging.info("Bot is running...")
application.run_polling() application.run_polling()
def _check_and_send_reboot_message(self):
reboot_file_path = "./.doreboot"
if os.path.exists(reboot_file_path):
with open(reboot_file_path, 'r') as f:
chat_id = f.read().strip()
if chat_id:
message = "The application has been successfully initialized after a reboot."
application = Application.builder().token(self.telegram_bot_token).build()
application.bot.send_message(chat_id=chat_id, text=message)