def delete_address(bot, update):
    chat_id = update.message.chat_id
    coin_address = update.message.text
    if coin_address == 'cancel':
        bot.send_message(chat_id=update.message.chat_id,
                         text='Cancel',
                         reply_markup=main_markup)
        return MAIN_MENU
    else:
        try:
            split_string = coin_address.split(
            )  # Separation of the received line
            coin = split_string[0]
            address = split_string[2]
            # Delete from DB
            db_worker = SQLighter(database_name)
            answer = db_worker.delete_row('users_address', chat_id, coin,
                                          address)
            db_worker.close()
            update.message.reply_text(answer, reply_markup=main_markup)
            return MAIN_MENU
        except:
            update.message.reply_text(
                'Invalid input format. The string format should be:\n`coin - address`',
                parse_mode='MARKDOWN',
                reply_markup=cancel_markup)
            return DELETE_ADDRESS
Exemplo n.º 2
0
Arquivo: bot.py Projeto: viktorvm/bot
def message_handle(message):
    try:
        db_worker = SQLighter(config.database_name)

        #убрать после того, как имена всех существующих обновятся
        x = db_worker.select_single(message.chat.id)
        if x:
            username = '******' + message.chat.username.encode('utf-8') + '"' if message.chat.username else 'null'
            if username:
                db_worker.upd_col('chat_username', username, message.chat.id)
            first_name = '"' + message.chat.first_name.encode('utf-8') + '"' if message.chat.first_name else 'null'
            if first_name:
                db_worker.upd_col('chat_first_name', first_name, message.chat.id)
        #---------------------------------------------------------

        if message.text == u'Да. То, что нужно!':
            username = '******' + message.chat.username.encode('utf-8') + '"' if message.chat.username else 'null'
            first_name = '"' + message.chat.first_name.encode('utf-8') + '"' if message.chat.first_name else 'null'
            db_worker.add_row(message.chat.id, username, first_name)
            bot.send_message(message.chat.id, 'Больше тебе приседать не придется:) Я буду ежедневно напоминать тебе '
                                              'об отчете каждый час, начиная с 20:00. '
                                              'До связи!', reply_markup=utils.hide_markup())
        elif message.text == u'+':
            db_worker.upd_col('done', 1, message.chat.id)
            bot.send_message(message.chat.id, 'Отлично! встретимся завтра:)')
        elif message.text == u'Спроси в другой раз':
            bot.send_message(message.chat.id, 'Окей, напомню через час.')
        elif message.text == u'Забудь про меня':
            db_worker.delete_row(message.chat.id)
            bot.send_message(message.chat.id, 'Хорошо, больше не буду тебе надоедать. '
                                              'Если передумаешь подай команду /start')
        elif message.text == u'Беру на себя всю ответственность':
            bot.send_message(message.chat.id, 'Смотри, так до приседаний недалеко))')
        else:
            bot.send_message(message.chat.id, 'Извини, такой команды я не знаю. '
                                          'Я пока примитивный глуповатый бот:( Но я учусь')
    except Exception as e:
        logging.error(e)