Exemple #1
0
def handler_callback(call):
    if 'black_list' in call.data:
        black_user_id = int(call.data.split()[1])
        db = PostgresDB()
        db.in_black_list(black_user_id, call.message.chat.id)
        db.close()
        bot.send_message(call.message.chat.id,
                         'Пользователь добален в черный список')

    elif 'delete_from_bl' in call.data:
        delete_from_bl_id = int(call.data.split()[1])
        db = PostgresDB()
        db.delete_from_bl(delete_from_bl_id, call.message.chat.id)
        db.close()
        bot.send_message(call.message.chat.id,
                         'Пользователь удален из черного списка')

    else:
        markup = Utils().gen_markup_for_chat(['Выйти из чата'])
        chat_user_id = int(call.data)
        db = PostgresDB()
        login = db.get_user_login(chat_user_id)
        bot.send_message(call.message.chat.id,
                         'Вы в чате с %s' % login,
                         reply_markup=markup)
        bot.answer_callback_query(callback_query_id=call.id,
                                  show_alert=True,
                                  text=config.inChatMsg % (login, login))
        statDictInChat = 'in chat with ' + str(chat_user_id)
        statDict[call.message.chat.id] = statDictInChat
Exemple #2
0
def handler_black_list(message):
    db = PostgresDB()
    if not db.check_user(message.chat.id):
        db.close()
        handler_start(message)
    else:
        statDict[message.chat.id] = 'existUser'
        black_str = db.get_black_list(message.chat.id)
        if black_str != '':
            black_list = black_str.split()
            for i in black_list:
                inline_markup = Utils().gen_inline_markup_for_black_list(i)
                login = db.get_user_login(int(i))
                bot.send_message(message.chat.id,
                                 login,
                                 reply_markup=inline_markup)
                db.close()
        else:
            bot.send_message(message.chat.id, 'Ваш черный список пуст')
            db.close()