summaryrefslogtreecommitdiff
path: root/main.py
blob: 10c05a73986d85729d8f5be6d4b8246489e7b300 (plain)
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
55
56
57
58
59
60
61
62
63
64
65
66
# This file is part of python-cinema-club-bot
# contributed in 2024 by Mikhail Kirillov (~w96k) <w96k@runbox.com>

# To the extent possible under law, the author(s) have dedicated all copyright
# and related and neighboring rights to this software to the public domain
# worldwide. This software is distributed without any warranty.

# You should have received a copy of the CC0 Public Domain Dedication along
# with this software. If not, see:
# <http://creativecommons.org/publicdomain/zero/1.0/>

from telegram import Update
from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes, \
    MessageHandler, filters
from dotenv import load_dotenv
import os
import logging

from strings import INVALID_COMMAND
from persistence import Persistence
import commands


load_dotenv()

logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)


async def unknown(update: Update, context: ContextTypes.DEFAULT_TYPE):
    await context.bot.send_message(
        chat_id=update.effective_chat.id,
        text=INVALID_COMMAND
    )

async def error_handler(update: object, context: ContextTypes.DEFAULT_TYPE) -> None:
    logging.error("Exception:", exc_info=context.error)

    await context.bot.send_message(
        chat_id=update.effective_chat.id,
        text=context.error.message
    )

if __name__ == "__main__":
    app = ApplicationBuilder().token(
        os.environ.get("TELEGRAM_TOKEN")
    ).persistence(Persistence).build()

    # Meta commands
    app.add_handler(CommandHandler("about", commands.about))

    # Movie commands
    app.add_handler(CommandHandler("set", commands.set_users))
    app.add_handler(CommandHandler("add", commands.add_users))
    app.add_handler(CommandHandler("list", commands.list_users))
    app.add_handler(CommandHandler("remove", commands.remove_users))
    app.add_handler(CommandHandler("chooser", commands.chooser_user))

    app.add_handler(MessageHandler(filters.COMMAND, unknown))
    app.add_error_handler(error_handler)


    app.run_polling()