diff --git a/main.py b/main.py index b354ab6..90b543a 100644 --- a/main.py +++ b/main.py @@ -28,7 +28,7 @@ LOGS = config.getboolean('Settings', 'logs') #Инициализируем бота bot = telebot.TeleBot(BOT_TOKEN, parse_mode=None) -def select_message_from_database(type: str, username): +def select_message_from_db(type: str, username): """Функция для запроса к базе данных и получения сообщений для бота. В аргументы передаются: type - тип получаемой обратной связи, строковое значение, сохраненное в БД username - имя пользователя @@ -50,6 +50,26 @@ def select_message_from_database(type: str, username): if conn: conn.close() + +def error_message_from_db(id: int): + """Функция для запроса к базе данных и получения сообщений ошибки. В аргументы передаются: + id - идентификатор ошибки + """ + # Подключаемся к базе + conn = sqlite3.connect('tg-bot-database', check_same_thread=False) + try: + cursor = conn.cursor() + cursor.execute(f"SELECT * FROM error_messages WHERE id=?", (id,)) + response_from_database = str(cursor.fetchone()[1]) + return response_from_database + except sqlite3.Error as error: + if LOGS: + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f'Ошибка при работе с SQLite, {error}') + finally: + if conn: + conn.close() + def telegram_bot(): @bot.message_handler(commands=['start']) def send_welcome(message): @@ -66,9 +86,9 @@ def telegram_bot(): message_id=message.message_id) bot.send_sticker(message.chat.id, random_stick_hello) sleep(0.3) - except : + except: if LOGS: - bot.send_message(IMPORTANT_LOGS, 'Отправка приветственных стикеров лажает') + bot.send_message(IMPORTANT_LOGS, error_message_from_db(7)) markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True) item1 = types.KeyboardButton("Предложить свой пост") @@ -78,11 +98,11 @@ def telegram_bot(): markup.add(item1, item2, item3) try: username = message.from_user.first_name - hello_message = select_message_from_database('start_message', username) + hello_message = select_message_from_db('start_message', username) bot.send_message(message.chat.id, hello_message, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) except: if LOGS: - bot.send_message(IMPORTANT_LOGS, 'Что-то не так с отправкой первого сообщения') + bot.send_message(IMPORTANT_LOGS, error_message_from_db(8)) bot.register_next_step_handler(message, go_send_messages) @bot.message_handler(commands=['end_command']) @@ -94,7 +114,7 @@ def telegram_bot(): item4 = types.KeyboardButton("Сказать пока!") markup.add(item1, item2, item3, item4) bot.send_message(message.chat.id, - "Выбери нужную кнопку ниже".format( + "Выбери нужную кнопку внизу экрана".format( message.from_user, bot.get_me()), parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) bot.register_next_step_handler(message, go_send_messages) @@ -105,15 +125,15 @@ def telegram_bot(): try: markup = types.ReplyKeyboardRemove() username = message.from_user.first_name - suggest_news = select_message_from_database('suggest_news', username) + suggest_news = select_message_from_db('suggest_news', username) bot.send_message(message.chat.id, suggest_news, parse_mode='html') sleep(0.3) username = message.from_user.first_name - suggest_news_2 = select_message_from_database('suggest_news_2', username) + suggest_news_2 = select_message_from_db('suggest_news_2', username) msg = bot.send_message(message.chat.id, suggest_news_2,parse_mode='html', reply_markup=markup) except: if LOGS: - bot.send_message(IMPORTANT_LOGS, 'Не сработала предложка') + bot.send_message(IMPORTANT_LOGS, error_message_from_db(10)) #logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, @@ -123,7 +143,7 @@ def telegram_bot(): elif message.text == "Связаться с админами": username = message.from_user.first_name - connect_with_admin = select_message_from_database('connect_with_admin', username) + connect_with_admin = select_message_from_db('connect_with_admin', username) msg = bot.send_message(message.chat.id, connect_with_admin, parse_mode="html") #logging if LOGS: @@ -135,7 +155,7 @@ def telegram_bot(): elif message.text == "Удалить пост": #TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать username = message.from_user.first_name - del_message = select_message_from_database('del_message', username) + del_message = select_message_from_db('del_message', username) msg = bot.send_message(message.chat.id, del_message, parse_mode="html") #logging if LOGS: @@ -150,19 +170,19 @@ def telegram_bot(): number_stick_bye = random.randint(1, len(name_stick_bye)) random_stick_bye = open(name_stick_bye[number_stick_bye], 'rb') bot.send_sticker(message.chat.id, random_stick_bye) - except ApiTelegramException.with_traceback('tb') as r: + except ApiTelegramException: if LOGS: - bot.send_message(IMPORTANT_LOGS, r) + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(11)) markup = types.ReplyKeyboardRemove() try: username = message.from_user.first_name - bye_message = select_message_from_database('bye_message', username) + bye_message = select_message_from_db('bye_message', username) bot.send_message(message.chat.id, bye_message, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) except: if LOGS: - bot.send_message(IMPORTANT_LOGS, 'Что-то не так с прощанием') + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(6)) if LOGS: #logging @@ -173,11 +193,11 @@ def telegram_bot(): else: try: username = message.from_user.first_name - user_error = select_message_from_database('user_error', username) - bot.send_message(message.chat.id,user_error, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK) + user_error = select_message_from_db('user_error', username) + bot.send_message(message.chat.id, user_error, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK) except: if LOGS: - bot.send_message(IMPORTANT_LOGS, 'Что-то не так с пользовательским вводом') + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(9)) #logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, @@ -231,10 +251,12 @@ def telegram_bot(): pass except: if LOGS: - bot.send_message(IMPORTANT_LOGS, f'ALARM, у пользователя не отправилось сообщение\n\nАвтор: {message.from_user.first_name}\nПост можно посмотреть в логах') + username = message.from_user.first_name + error_message = str(error_message_from_db(5)).replace('username', username) + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message) username = message.from_user.first_name - success_send_message = select_message_from_database('success_send_message', username) + success_send_message = select_message_from_db('success_send_message', username) bot.send_message(message.chat.id, success_send_message, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK) after_post(message=message) @@ -246,11 +268,11 @@ def telegram_bot(): message_id=message.message_id ) username = message.from_user.first_name - question = select_message_from_database('question', username) + question = select_message_from_db('question', username) bot.send_message(message.chat.id, question, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK) except: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text='Что-то не так с отправкой обратной связи') + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(4)) after_post(message=message) @@ -263,7 +285,7 @@ def telegram_bot(): bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) except: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text='Не удалось выложить/удалить текстовый пост, что-то пошло не так') + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(3)) elif call.data == 'post_post_post' and call.message.content_type == 'photo': try: bot.send_photo( @@ -274,13 +296,13 @@ def telegram_bot(): bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) except: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text='Не удалось выложить/удалить пост с фотографией, что-то пошло не так') + bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(2)) elif call.data == 'decline': try: bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) except: if LOGS: - logs(IMPORTANT_LOGS, 'Не удалось отклонить пост') + logs(IMPORTANT_LOGS, error_message_from_db(1)) def logs(chat_name, text_error): bot.send_message(chat_id=chat_name, text=text_error)