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

Merge: Add functionality to check for new commits and trigger reboot

Browse Source 
Merging pull request #111 which adds functionality to check for new commits and automatically trigger a reboot in the Telegram bot.
This commit is contained in:
bucolucas
2024-08-20 14:31:31 -05:00
committed by GitHub
parent 43202e3b6e 42e86d5fb3
commit ceae996bb5
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
telegram_helper.py
+9 -5
View File
@@ -3,6 +3,7 @@ import logging
import sys
import asyncio
import time
import git
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes, CallbackQueryHandler
@@ -10,7 +11,8 @@ class TelegramHelper:
def __init__(self, bot):
self.bot = bot
self.telegram_bot_token = os.getenv('TELEGRAM_BOT_TOKEN')
self.repo = os.getenv("GITHUB_REPOSITORY")
self.repo_path = os.getenv("GITHUB_REPOSITORY", ".")
self.repo = git.Repo(self.repo_path)
self.start_time = time.time()
async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
@@ -76,12 +78,13 @@ class TelegramHelper:
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...")
if update:
await update.message.reply_text("Rebooting the bot...")
logging.info("Received reboot command. Exiting process...")
reboot_file_path = "./.doreboot"
if not os.path.exists(reboot_file_path):
with open(reboot_file_path, 'w') as f:
f.write(str(update.effective_chat.id))
f.write(str(update.effective_chat.id) if update else "")
sys.exit(0)
async def check_doreboot_file(self, application: Application):
@@ -89,7 +92,8 @@ class TelegramHelper:
if os.path.exists(reboot_file_path):
with open(reboot_file_path, 'r') as f:
chat_id = f.read().strip()
await application.bot.send_message(chat_id=chat_id, text="The application has finished initializing.")
if chat_id:
await application.bot.send_message(chat_id=chat_id, text="The application has finished initializing.")
os.remove(reboot_file_path)
async def check_for_new_commits(self):
@@ -124,4 +128,4 @@ class TelegramHelper:
# Start the commit checking task
asyncio.get_event_loop().create_task(self.check_for_new_commits())
application.run_polling()
application.run_polling()