MasterBot/main.py at master · 7DSMeliodas/MasterBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from datetime import timedelta
import logging
from telegram import Update
from telegram.ext import Updater, CommandHandler, TypeHandler, CallbackQueryHandler

from callback import Jobs, Misc, Restart, Stats
from const.CONFIG import CONFIG

logging.basicConfig(format='%(asctime)s - %(name)s - %(message)s',
                    level=logging.INFO, filename="logs.log")

logger = logging.getLogger(__name__)


def main():
    updater = Updater(CONFIG.BOTTOKEN)
    dispatcher = updater.dispatcher
    job_q = updater.job_queue
    job_q.run_repeating(
        callback=Jobs.supervisor, interval=timedelta(minutes=5), first=5
    )
    dispatcher.add_handler(
        TypeHandler(type=Update, callback=Misc.block_access), group=0
    )
    dispatcher.add_handler(CommandHandler("start", Misc.start_command), group=1)
    dispatcher.add_handler(CommandHandler("restart", Restart.command), group=1)
    dispatcher.add_handler(
        CallbackQueryHandler(Restart.command, pattern="^restart"), group=1
    )
    dispatcher.add_handler(CommandHandler("get", Misc.get_all), group=1)
    dispatcher.add_handler(CommandHandler("help", Misc.help_command), group=1)
    dispatcher.add_handler(CommandHandler("stats", Stats.command), group=1)
    dispatcher.add_handler(
        CommandHandler("detail_stats", Stats.detail_command), group=1
    )
    dispatcher.add_handler(
        CallbackQueryHandler(Stats.command, pattern="refresh"), group=1
    )

    if CONFIG.PORT_NUM != 0:
        updater.start_webhook(
            listen="127.0.0.1", port=CONFIG.PORT_NUM, url_path=CONFIG.BOTTOKEN
        )
        updater.bot.set_webhook(
            url=f"https://{CONFIG.IP_ADDR}:443/{CONFIG.BOTTOKEN}",
            certificate=open("cert.pem", "rb"),
        )
    else:
        updater.start_polling()
        updater.idle()


if __name__ == "__main__":
    main()