messages in database

main.py

@@ -1,6 +1,7 @@
 import configparser
 import os
 import sys
+import sqlite3
 from pathlib import Path
 from time import sleep
 
@@ -27,6 +28,27 @@ LOGS = config.getboolean('Settings', 'logs')
 #Инициализируем бота
 bot = telebot.TeleBot(BOT_TOKEN, parse_mode=None)
 
+def select_message_from_database(type: str, username):
+    """Функция для запроса к базе данных и получения сообщений для бота. В аргументы передаются:
+    type - тип получаемой обратной связи, строковое значение, сохраненное в БД
+    username - имя пользователя
+    """
+    # Подключаемся к базе
+    conn = sqlite3.connect('tg-bot-database', check_same_thread=False)
+    try:
+        cursor = conn.cursor()
+        cursor.execute(f"SELECT * FROM messages WHERE type=?", (type,))
+        #Забираем данные из таблицы, преобразуем в строку, заменяем поле username на имя пользователя,
+        #и вместо амберсанда подставляем перенос строки
+        response_from_database = str(cursor.fetchone()[1]).replace('username', username).replace('&', '\n')
+        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'])
@@ -55,16 +77,9 @@ def telegram_bot():
 
         markup.add(item1, item2, item3)
         try:
-            bot.send_message(message.chat.id,
-                             "Привет, {0.first_name}!\nЯ бот группы 'Влюбленный Бийск', "
-                             "создан для того, "
-                             "чтобы помочь тебе выложить пост в группу,"
-                             "или связаться с админами.\n"
-                             "Основная группа в ВК: https://vk.com/love_bsk\n"
-                             "Основная группа в ТГ: https://t.me/love_biysk\n"
-                             "<i>Have a nice time</i>".format(
-                                 message.from_user, bot.get_me()),
-                             parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK)
+            username = message.from_user.first_name
+            hello_message = select_message_from_database('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, 'Что-то не так с отправкой первого сообщения')
@@ -79,7 +94,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)
@@ -89,23 +104,13 @@ def telegram_bot():
         if message.text == 'Предложить свой пост':
             try:
                 markup = types.ReplyKeyboardRemove()
-                bot.send_message(message.chat.id,
-                                 "{0.first_name}, окей, пришли мне текст поста\n"
-                                 "В данный момент я работаю в тестовом режиме\n"
-                                 "поэтому к посту можно прикрепить не более одного фото\n"
-                                 "и никаких аудио или видео:(\nОбещаю, я научусь их обрабатывать, но позже".format(message.from_user),
-                                 parse_mode='html'
-                                 )
-                sleep(0.6)
-                msg = bot.send_message(message.chat.id,
-                                 "Если хочешь чтобы пост был опубликован анонимно напиши в любом месте "
-                                 "своего поста слово 'анон'.\nЕсли хочешь опубликовать пост не анонимно, "
-                                 "то напиши 'не анон', 'неанон' или не пиши ничего\n"
-                                 "Обрати внимание, что я обучен только на эти команды, и проверь\nчто ты написал так как я тебя попросил\n"
-                                 "Пост будет опубликован только в группе ТГ\n"
-                                 "<i>Have a nice time</i>".format(message.from_user),
-                                 parse_mode='html', reply_markup=markup
-                                 )
+                username = message.from_user.first_name
+                suggest_news = select_message_from_database('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)
+                msg = bot.send_message(message.chat.id, suggest_news_2,parse_mode='html', reply_markup=markup)
             except:
                 if LOGS:
                     bot.send_message(IMPORTANT_LOGS, 'Не сработала предложка')
@@ -117,9 +122,9 @@ def telegram_bot():
             bot.register_next_step_handler(msg, resend_message_in_group_for_post)
 
         elif message.text == "Связаться с админами":
-            msg = bot.send_message(message.chat.id, "{0.first_name}, напиши свое обращение,"
-                                          "Мы постараемся ответить в ближайшее время:)".format(message.from_user),
-                         parse_mode="html")
+            username = message.from_user.first_name
+            connect_with_admin = select_message_from_database('connect_with_admin', username)
+            msg = bot.send_message(message.chat.id, connect_with_admin, parse_mode="html")
             #logging
             if LOGS:
                 bot.forward_message(chat_id=GROUP_FOR_LOGS,
@@ -129,9 +134,9 @@ def telegram_bot():
 
         elif message.text == "Удалить пост":
             #TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать
-            msg = bot.send_message(message.chat.id, "{0.first_name}, перешли твой пост из основного канала,"
-                                              "\nИмей ввиду, что можно удалить только тот пост в котором ты автор".format(
-                message.from_user), parse_mode="html")
+            username = message.from_user.first_name
+            del_message = select_message_from_database('del_message', username)
+            msg = bot.send_message(message.chat.id, del_message, parse_mode="html")
             #logging
             if LOGS:
                 bot.forward_message(chat_id=GROUP_FOR_LOGS,
@@ -151,13 +156,9 @@ def telegram_bot():
 
             markup = types.ReplyKeyboardRemove()
             try:
-                bot.send_message(message.chat.id,
-                                 "И тебе пока, "
-                                 "если позднее захочешь предложить еще один пост, "
-                                 "или обратиться к нам с вопросом, "
-                                 "то просто пришли в чат команду /start\n"
-                                 "<i>Have a nice time</i>".format(
-                                     message.from_user, bot.get_me()),
+                username = message.from_user.first_name
+                bye_message = select_message_from_database('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:
@@ -170,11 +171,13 @@ def telegram_bot():
                                     message_id=message.message_id)
 
         else:
-            bot.send_message(message.chat.id,
-                             "Я тебя не понимаю. Повтори команду.\n"
-                             "<i>Have a nice time</i>".format(
-                                 message.from_user, bot.get_me()),
-                             parse_mode='html', disable_web_page_preview=not PREVIEW_LINK)
+            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)
+            except:
+                if LOGS:
+                    bot.send_message(IMPORTANT_LOGS, 'Что-то не так с пользовательским вводом')
             #logging
             if LOGS:
                 bot.forward_message(chat_id=GROUP_FOR_LOGS,
@@ -230,11 +233,10 @@ def telegram_bot():
             if LOGS:
                 bot.send_message(IMPORTANT_LOGS, f'ALARM, у пользователя не отправилось сообщение\n\nАвтор: {message.from_user.first_name}\nПост можно посмотреть в логах')
 
-        bot.send_message(message.chat.id,
-                         "Пост успешно отправлен. Ожидай одобрения\n"
-                         "<i>Have a nice time</i>".format(message.from_user),
-                         parse_mode='html', disable_web_page_preview=not PREVIEW_LINK
-                         )
+        username = message.from_user.first_name
+        success_send_message = select_message_from_database('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)
 
     def resend_message_in_group_for_message(message):
@@ -243,11 +245,9 @@ def telegram_bot():
                                 from_chat_id=message.chat.id,
                                 message_id=message.message_id
                                 )
-            bot.send_message(message.chat.id,
-                             "Вопрос успешно отправлен. Ответим как только сможем\n"
-                             "<i>Have a nice time</i>".format(message.from_user),
-                             parse_mode='html', disable_web_page_preview=not PREVIEW_LINK
-                             )
+            username = message.from_user.first_name
+            question = select_message_from_database('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='Что-то не так с отправкой обратной связи')
@@ -280,8 +280,10 @@ 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='Не удалось отклонить пост')
+                    logs(IMPORTANT_LOGS, 'Не удалось отклонить пост')
 
+def logs(chat_name, text_error):
+    bot.send_message(chat_id=chat_name, text=text_error)
 
 if __name__ == '__main__':
     telegram_bot()
@@ -291,13 +293,10 @@ if __name__ == '__main__':
         bot.load_next_step_handlers()
     except ConnectionError as e:
         if LOGS:
-            print('Ошибка соединения: ', e)
             bot.send_message(IMPORTANT_LOGS, "Ошибка соединения, потерял связь")
     except Exception as r:
         if LOGS:
-            print("Непредвиденная ошибка: ", r)
             bot.send_message(IMPORTANT_LOGS, "Произошло что-то непредвиденное, хелп")
     finally:
         if LOGS:
-            print("Здесь всё закончилось")
             bot.send_message(IMPORTANT_LOGS, 'Я упал, помогите')