swissobot/main.py at master · denolia/swissobot · 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
# coding=utf-8

import yaml
from telegram import ChatAction
from telegram.ext import Filters
from telegram.ext import MessageHandler
from telegram.ext import Updater, CommandHandler, CallbackQueryHandler

from config import log_conf
from core.diary import diary_handler
from core.money import money_handler, money_callback_handler, money_list_handler, money_edit_handler
from core.task import task, task_list, done_task, task_delete_handler
from core.weather import current_weather
from utils.commands import MONEY_COMMAND, MONEY_EDIT_COMMAND, TASK_DELETE_COMMAND, MONEY_LIST_COMMAND, DIARY_COMMAND

log = log_conf.get_logger(__name__)

help_text = """
Доступные команды:
{money_format}
{money_list_format} - список расходов за дату с id
{money_edit_format} - изменить запись о расходе по id

{diary_format} запись
/task Имя задачи [; категория; дата; ссылка]
/tasklist - показать список всех актуальных задач
{task_delete_format}
/weather - показать погоду


Примеры:
{money_example}
{diary_example}
/task Погладить котика; Дела; 21.01; google.com
""".format(money_format=MONEY_COMMAND.format,
           money_list_format=MONEY_LIST_COMMAND.format,
           money_example=MONEY_COMMAND.example,
           task_delete_format=TASK_DELETE_COMMAND.format,
           money_edit_format=MONEY_EDIT_COMMAND.format,
           diary_format=DIARY_COMMAND.format,
           diary_example=DIARY_COMMAND.example
           )


def start(bot, update):
    bot.sendChatAction(chat_id=update.message.chat_id,
                       action=ChatAction.TYPING)
    bot.sendMessage(chat_id=update.message.chat_id,
                    text="Hi, guys! Welcome to the awesome multifunctional bot!\n" + help_text)


def bot_help(bot, update):
    bot.sendChatAction(chat_id=update.message.chat_id,
                       action=ChatAction.TYPING)
    bot.sendMessage(chat_id=update.message.chat_id,
                    text=help_text)


def unknown(bot, update):
    bot.sendChatAction(chat_id=update.message.chat_id,
                       action=ChatAction.TYPING)
    bot.sendMessage(chat_id=update.message.chat_id,
                    text="Sorry, I didn't understand that command.")


def error(bot, update, error):
    log.warning('Update "%s" caused error "%s"' % (update, error))


if __name__ == '__main__':
    log.info("Starting swissobot...")
    with open("config/config.yaml", 'r') as stream:
        config = yaml.load(stream)
    log.info("Reading token...")
    TOKEN = config.get('token')
    updater = Updater(token=TOKEN)

    log.info("Adding handlers...")
    updater.dispatcher.add_handler(CommandHandler('start', start))
    updater.dispatcher.add_handler(CommandHandler('task', task))
    updater.dispatcher.add_handler(CommandHandler('tasklist', task_list))
    updater.dispatcher.add_handler(CommandHandler(TASK_DELETE_COMMAND.name, task_delete_handler))
    updater.dispatcher.add_handler(CommandHandler('done', done_task))
    updater.dispatcher.add_handler(CommandHandler(DIARY_COMMAND.name, diary_handler))
    updater.dispatcher.add_handler(CommandHandler(MONEY_COMMAND.name, money_handler))
    updater.dispatcher.add_handler(CommandHandler(MONEY_LIST_COMMAND.name, money_list_handler))
    updater.dispatcher.add_handler(CommandHandler(MONEY_EDIT_COMMAND.name, money_edit_handler))
    updater.dispatcher.add_handler(CallbackQueryHandler(money_callback_handler))
    updater.dispatcher.add_handler(CommandHandler('weather', current_weather))
    updater.dispatcher.add_handler(CommandHandler('help', bot_help))
    updater.dispatcher.add_error_handler(error)

    # Note: must be the last added to handler
    updater.dispatcher.add_handler(MessageHandler(Filters.command, unknown))

    # Start the Bot
    updater.start_polling()
    log.info("Listening...")

    # Run the bot until the user presses Ctrl-C or the process receives SIGINT,
    # SIGTERM or SIGABRT
    updater.idle()