chromium.py

import logging
import os
import pickle
from telegram.ext import CommandHandler, Job, run_async
from telegram import ChatAction
from config import Config
from selenium.webdriver.common.by import By
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from os import execl
from sys import executable
from bot import updater, dp, browser, restricted

from bot.meet import meet
from bot.zoom import zoom

if Config.SCHEDULE == True:
    from bot.meet_schedule import mJobQueue, timeTable
    from bot.zoom_schedule import zJobQueue

userId = Config.USERID

@restricted
@run_async
def exit(update, context):
    context.bot.send_message(chat_id=userId, text="Restarting bot, Please wait!")
    browser.quit()
    execl(executable, executable, "chromium.py")

@restricted
@run_async
def status(update, context):
    browser.save_screenshot("ss.png")
    context.bot.send_chat_action(chat_id=userId, action=ChatAction.UPLOAD_PHOTO)
    context.bot.send_photo(chat_id=userId, photo=open('ss.png', 'rb'), timeout = 120)
    os.remove('ss.png')

@restricted
@run_async
def help(update, context):
    context.bot.send_message(chat_id=userId, text="""/meet <link> -   Bunk a Google Meet meeting
/zoom <userID> <password>   - Bunk a zoom meeting
/exit -   Exit the meeting and restart
/status -   Send screenshot
/help -   Send this help message""")

def main():
    j = updater.job_queue

    dp.add_handler(CommandHandler("zoom", zoom))
    dp.add_handler(CommandHandler("meet", meet))

    if Config.SCHEDULE == True:
        mJobQueue()
        zJobQueue()
        dp.add_handler(CommandHandler("timetable", timeTable))

    dp.add_handler(CommandHandler("exit", exit))
    dp.add_handler(CommandHandler("status", status))
    dp.add_handler(CommandHandler("help", help))
    logging.info("Bot started")

    updater.start_polling()

if __name__ == '__main__':
    main()