bot.py

import logging

from telegram.ext import Updater, CommandHandler, CallbackQueryHandler, Filters, MessageHandler, RegexHandler

import config
import keyboards
import get_date
import parse

import search_city

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

def start_greeting(bot, update, chat_data):
    chat_data.clear()
    update.message.reply_text('Привет! Я помогу тебе быстро подобрать отличный отель с помощью сервиса booking.com.',
                              reply_markup=keyboards.REMOVE_KB)
    update.message.reply_text('Если ты готов - просто напиши мне любое сообщение')

    chat_data['message_type'] = 'start_description'


def text_message_handler(bot, update, chat_data):
    if chat_data['message_type'] == 'start_description':
        bot.send_message(update.message.chat_id,
                         text='Рад твоему сообщению! \n'
                              'Как я уже говорил, я использую сервис booking.com, как наиболее надежный и удобный сервис. \n'
                              'Все бронирования ты будешь делать на booking.com самостоятельно, а я отберу для тебя отели согласно твоим личным предпочтениям. \n'
                              'Если ты не останавливаешься в хостелах – ты больше не увидишь их в своем списке отелей, но, если захочешь, я буду показывать тебе только хостелы. \n'
                              'Я не буду просить дополнительную комиссию, все цены – согласно ценам на booking.com, обязательно проверь! \n'
                              'Чтобы подобрать лучшие варианты, мне нужно задать тебе несколько вопросов. Давай попробуем?)',
                         reply_markup=keyboards.YES_NO_KB)
        return

    elif chat_data['message_type'] == 'amount_adults':
        try:
            if int(update.message.text) > 30 or int(update.message.text) < 1:
                bot.send_message(update.message.chat_id,
                                 text='Количество указано некорректно! Ввведи количество взрослых еще раз')
                return
        except ValueError:
            bot.send_message(update.message.chat_id,
                             text='Количество указано некорректно! Ввведи количество взрослых еще раз')
            return

        chat_data['group_adults'] = 'group_adults=' + update.message.text

        bot.send_message(update.message.chat_id,
                         text='Укажите количество номеров (не больше, чем количество взрослых!)')
        chat_data['message_type'] = 'no_rooms'

    elif chat_data['message_type'] == 'no_rooms':
        try:
            if int(update.message.text) > 30 or int(update.message.text) < 1:
                bot.send_message(update.message.chat_id,
                                 text='Количество номеров указано некорректно! Ввведи число еще раз')
                return
        except ValueError:
            bot.send_message(update.message.chat_id,
                             text='Количество номеров указано некорректно! Ввведи число еще раз')
            return

        chat_data['no_rooms'] = 'no_rooms=' + update.message.text

        bot.send_message(update.message.chat_id,
                         text='Будут ли с тобой дети? Укажи количество (от 1 до 10) или 0, если их не будет')
        chat_data['message_type'] = 'amount_children'

    elif chat_data['message_type'] == 'amount_children':
        try:
            if int(update.message.text) > 10 or int(update.message.text) < 0:
                bot.send_message(update.message.chat_id,
                                 text='Количество указано некорректно! Ввведи количество детей еще раз')
                return
        except ValueError:
            bot.send_message(update.message.chat_id,
                             text='Количество указано некорректно! Ввведи количество детей еще раз')
            return

        chat_data['group_children'] = 'group_children=' + update.message.text
        add_params(bot, update, chat_data)

    elif chat_data['message_type'] == 'go_travel':
        bot.send_message(update.message.chat_id,
                         text='Ищем город в базе данных и проверяем правильность введенной даты...')

        res = parse.str_parse(update.message.text)

        if res[0] == 'error':
            bot.send_message(update.message.chat_id,
                             text='Не можем распознать твой запрос. Попробуй еще раз. Например, Рим 18 марта')
            return

        chat_data['city'] = search_city.search_city(res[0])

        if chat_data['city'] == 'error':
            bot.send_message(update.message.chat_id,
                             text='Город не найден или ошибка в написании запроса. Возможно, вы ошиблись в названии. '
                                  'Попробуй ввести его еще раз;)')
            return

        chat_data['city'] = 'city=' + chat_data['city']

        check_in = get_date.parse_date(res[1])
        if check_in[0] == 'error':
            bot.send_message(update.message.chat_id,
                            text='Не смогли распознать дату или ошибка в написании запроса. Возможно, вы ошиблись в '
                                 'написании запроса. Попробуйте ввести его еще раз;)')
            return

        chat_data['date_in'] = check_in[2] + '-' + check_in[1] + '-' + check_in[0]

        bot.send_message(update.message.chat_id,
                         text='Отлично! А на сколько ночей ты планируешь остановиться? Напиши количество, например, 7')

        chat_data['message_type'] = 'night_amount'

    elif chat_data['message_type'] == 'night_amount':
        travel_dates = parse.night_amount(chat_data['date_in'], update.message.text)
        chat_data['date_in'] = 'checkin_monthday=' + str(travel_dates[0]) + '&checkin_month=' + str(travel_dates[1]) + \
                               '&checkin_year=' + str(travel_dates[2])
        chat_data['date_out'] = 'checkout_monthday=' + str(travel_dates[3]) + '&checkout_month=' + str(travel_dates[4]) + \
                                '&checkout_year=' + str(travel_dates[5])

        ref = 'www.booking.com/searchresults.ru.html?' + chat_data['city'] + '&' + chat_data['date_in'] + '&' + \
              chat_data['date_out'] + '&aid=1433748&' + chat_data['no_rooms'] + '&' + chat_data['group_adults'] + '&' +\
              chat_data['group_children'] + '&nflt=' + chat_data['hotel'] + chat_data['quality'] + chat_data['stars'] + \
              chat_data['breakfast'] + chat_data['wifi'] + chat_data['parking'] + chat_data['pool'] + \
              chat_data['disabled'] + chat_data['pets'] + chat_data['reception']

        bot.send_message(update.message.chat_id,
                         text='Готово! Пройди по ссылке с подходящими вариантами: \n' + ref)
        chat_data['message_type'] = 'go_travel'


def type_of_hotel(bot, update, chat_data):
    query = update.callback_query

    bot.edit_message_text(text='Отлично! Где ты предпочитаешь останавливаться? Выбери один из возможных вариантов ниже:',
                          reply_markup=keyboards.TYPE_HOTEL_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def quality_of_hotel(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'th1':
        chat_data['hotel'] = 'ht_id%253D204%253Bht_id%253D218%253Bht_id%253D215%253Bht_id%253D206%253Bht_id%253D208%253B'
    elif query.data == 'th2':
        chat_data['hotel'] = 'ht_id%253D221%253Bht_id%253D223%253Bht_id%253D201%253Bht_id%253D216%253Bht_id%253D220%253B'
    elif query.data == 'th3':
        chat_data['hotel'] = 'ht_id%253D203%253B'
    elif query.data == 'th0':
        chat_data['hotel'] = ''

    bot.edit_message_text(text='Какой рейтинг должен быть у варианта размещения? Выбери один из возможных вариантов ниже:',
                          reply_markup=keyboards.QUALITY_HOTEL_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def num_stars(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'qh60':
        chat_data['quality'] = 'review_score%253D60%253B'
    elif query.data == 'qh70':
        chat_data['quality'] = 'review_score%253D70%253B'
    elif query.data == 'qh80':
        chat_data['quality'] = 'review_score%253D80%253B'
    elif query.data == 'qh90':
        chat_data['quality'] = 'review_score%253D90%253B'
    elif query.data == 'qh0':
        chat_data['quality'] = ''

    if chat_data['hotel'] == 'ht_id%253D204%253Bht_id%253D218%253Bht_id%253D215%253Bht_id%253D206%253Bht_id%253D208%253B':
        bot.edit_message_text(text='А сколько звезд? Выбери один из возможных вариантов ниже:',
                          reply_markup=keyboards.NUM_STARS_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)
    else:
        chat_data['stars'] = ''
        bot.edit_message_text(text='Укажите количество взрослых(от 1 до 30)',
                              chat_id=query.message.chat_id,
                              message_id=query.message.message_id)
        chat_data['message_type'] = 'amount_adults'


def amount_adults(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'ns1':
        chat_data['stars'] = 'class%253D1%253B'
    elif query.data == 'ns2':
        chat_data['stars'] = 'class%253D2%253B'
    elif query.data == 'ns3':
        chat_data['stars'] = 'class%253D3%253B'
    elif query.data == 'ns4':
        chat_data['stars'] = 'class%253D4%253B'
    elif query.data == 'ns5':
        chat_data['stars'] = 'class%253D5%253B'
    elif query.data == 'ns0':
        chat_data['stars'] = ''

    bot.edit_message_text(text='Укажите количество взрослых(от 1 до 30)',
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)
    chat_data['message_type'] = 'amount_adults'


#def amount_children(bot, update, chat_data):
#    query = update.callback_query

#    bot.edit_message_text(text='Укажите количество детей(от 1 до 10)',
#                          chat_id=query.message.chat_id,
#                          message_id=query.message.message_id)
#    chat_data['message_type'] = 'amount_children'


def add_params(bot, update, chat_data):
    bot.send_message(text='Хочешь ввести дополнительные параметры? Например, наличие бесплатного wi-fi, парковки или бассейна.',
                     reply_markup=keyboards.ADD_PARAMETERS_KB,
                     chat_id=update.message.chat_id,
                     message_id=update.message.message_id)


def reception(bot, update, chat_data):
    query = update.callback_query

    bot.edit_message_text(text='Важно ли, чтобы стойка регистрации работала круглосуточно?',
                          reply_markup=keyboards.RECEPTION_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def add_breakfast(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_reception':
        chat_data['reception'] = 'hr_24%3D8%3B&lsf=hr_24'
    elif query.data == 'no_reception':
        chat_data['reception'] = ''

    bot.edit_message_text(text='Завтрак должен быть включен в проживание?',
                          reply_markup=keyboards.BREAKFAST_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def add_wifi(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_breakfast':
        chat_data['breakfast'] = 'mealplan%3D1%3B'
    elif query.data == 'no_breakfast':
        chat_data['breakfast'] = ''

    bot.edit_message_text(text='Важно ли наличие бесплатного wi-fi?',
                          reply_markup=keyboards.WIFI_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def add_parking(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_wifi':
        chat_data['wifi'] = 'hotelfacility%3D107%3B'
    elif query.data == 'no_wifi':
        chat_data['wifi'] = ''

    bot.edit_message_text(text='Нужна ли парковка?',
                          reply_markup=keyboards.PARKING_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def add_pool(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_parking':
        chat_data['parking'] = 'hotelfacility%3D2%3B'
    elif query.data == 'no_parking':
        chat_data['parking'] = ''

    bot.edit_message_text(text='Нужен ли бассейн?',
                          reply_markup=keyboards.POOL_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def add_disabled(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_pool':
        chat_data['pool'] = 'hotelfacility%3D301%3B'
    elif query.data == 'no_pool':
        chat_data['pool'] = ''

    bot.edit_message_text(text='Необходимы ли удобства для гостей с ограниченными физическими возможностями?',
                          reply_markup=keyboards.DISABLED_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def add_pets(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_disabled':
        chat_data['disabled'] = 'hotelfacility=25%3B'
    elif query.data == 'no_disabled':
        chat_data['disabled'] = ''

    bot.edit_message_text(text='Допускается ли размещение домашних животных?',
                          reply_markup=keyboards.PETS_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def save_add_params(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'yes_pets':
        chat_data['pets'] = 'hotelfacility%3D4%3B'
    elif query.data == 'no_pets':
        chat_data['pets'] = ''

    bot.edit_message_text(text='Отлично! Теперь я смогу подобрать для тебя хорошие варианты проживания. Попробуем?',
                          reply_markup=keyboards.GO_TRAVEL_KB,
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)


def go_travel(bot, update, chat_data):
    query = update.callback_query

    if query.data == 'go_travel_no_add_params':
        chat_data['breakfast'] = ''
        chat_data['wifi'] = ''
        chat_data['parking'] = ''
        chat_data['pool'] = ''
        chat_data['disabled'] = ''
        chat_data['pets'] = ''
        chat_data['reception'] = ''

    bot.edit_message_text(text='Отлично! Куда и когда ты собираешься в следующий раз? Напиши мне город и дату.\n'
                          'Например: Рим 18 марта, \n'
                          'Москва 12.06.2018, \n',
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)
    chat_data['message_type'] = 'go_travel'


def back_to_main(bot, update):
    query = update.callback_query

    bot.edit_message_text(text='Хорошо! Обращайся в любое время!',
                          chat_id=query.message.chat_id,
                          message_id=query.message.message_id)
    bot.send_message(query.message.chat_id,
                     text='Нажми на кнопку снизу, как только снова захочешь начать общение с ботом',
                     reply_markup=keyboards.START_KB)


def help(bot, update):
    update.message.reply_text('Наш бот поможет найти лучшие места для отдыха при помощи сервиса booking.com\n'
                              'Просто выберите нужные вам параметры, и вы получите удобную ссылку на подходящие вам варианты')
    update.message.reply_text('Наберите /start, чтобы начать')


def error(bot, update, error):
    logging.warning('Update {} caused error {}'.format(update, error))


if __name__ == '__main__':
    updater = Updater(config.TOKEN)
    dispatcher = updater.dispatcher

    dispatcher.add_handler(CommandHandler('start', start_greeting, pass_chat_data=True))
    dispatcher.add_handler(CommandHandler('help', help))
    dispatcher.add_error_handler(error)

    dispatcher.add_handler(MessageHandler(Filters.text, text_message_handler, pass_chat_data=True))

    dispatcher.add_handler(CallbackQueryHandler(type_of_hotel, pattern='type_of_hotel', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(back_to_main, pattern='back_to_main'))
    dispatcher.add_handler(CallbackQueryHandler(quality_of_hotel, pattern='(th1)|(th2)|(th3)|(th0)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(num_stars, pattern='(qh60)|(qh70)|(qh80)|(qh90)|(qh0)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(amount_adults, pattern='(ns1)|(ns2)|(ns3)|(ns4)|(ns5)|(ns0)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_params, pattern='add_params', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(reception, pattern='yes_add_params', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_breakfast, pattern='(yes_reception)|(no_reception)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_wifi, pattern='(yes_breakfast)|(no_breakfast)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_parking, pattern='(yes_wifi)|(no_wifi)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_pool, pattern='(yes_parking)|(no_parking)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_disabled, pattern='(yes_pool)|(no_pool)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(add_pets, pattern='(yes_disabled)|(no_disabled)', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(save_add_params, pattern='yes_pets', pass_chat_data=True))
    dispatcher.add_handler(CallbackQueryHandler(go_travel, pattern='(go_travel)|(go_travel_no_add_params)', pass_chat_data=True))

    updater.start_polling()
    updater.idle()