Example #1
0
def add_trigger(message):
    try:
        if message.reply_to_message:
            if len(
                    message.text.lower()[13:]
            ) >= 3:  # and is_good_name_for_trigger(message.text.lower()):
                db = DataBase()

                if not db.is_trigger(message.text.lower()[13:],
                                     message.chat.id):
                    # добавить в бд
                    if message.reply_to_message.sticker:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.sticker.file_id,
                            "sticker", message.from_user.id,
                            message.from_user.username, message.chat.id,
                            message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.photo:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.photo[0].file_id, "photo",
                            message.from_user.id, message.from_user.username,
                            message.chat.id, message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.video:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.video.file_id, "video",
                            message.from_user.id, message.from_user.username,
                            message.chat.id, message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.voice:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.voice.file_id, "voice",
                            message.from_user.id, message.from_user.username,
                            message.chat.id, message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.audio:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.audio.file_id, "audio",
                            message.from_user.id, message.from_user.username,
                            message.chat.id, message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.document:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.document.file_id,
                            "document", message.from_user.id,
                            message.from_user.username, message.chat.id,
                            message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.video_note:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.video_note.file_id,
                            "video_note", message.from_user.id,
                            message.from_user.username, message.chat.id,
                            message.date
                        ]
                        db.add_trigger(data)
                    elif message.reply_to_message.text:
                        data = [
                            message.text.lower()[13:],
                            message.reply_to_message.text, "text",
                            message.from_user.id, message.from_user.username,
                            message.chat.id, message.date
                        ]
                        db.add_trigger(data)
                    bot.send_message(
                        message.chat.id,
                        "Триггер '" + message.text[13:] + "' добавлен.")

                else:
                    bot.send_message(
                        message.chat.id, "Триггер '" + message.text[13:] +
                        "' уже занесен в базу.")

                db.close()
            else:
                bot.send_message(message.chat.id,
                                 "Неккоректное имя триггера менее 3 символов")
        else:
            bot.send_message(message.chat.id, "Нет реплейнутого сообщения.")
    except:
        db.close()
        print("don't add_trigger.  ~~~" +
              str(time.strftime("%d.%m.%y %H:%M:%S", time.localtime())) +
              "\n\n" + traceback.format_exc() + "\n\n")