errors in db

This commit is contained in:
KerradKerridi
2022-08-28 10:44:45 +03:00
parent 6840bbf932
commit bf99afa89b

72
main.py
View File

@@ -28,7 +28,7 @@ LOGS = config.getboolean('Settings', 'logs')
#Инициализируем бота #Инициализируем бота
bot = telebot.TeleBot(BOT_TOKEN, parse_mode=None) 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 - тип получаемой обратной связи, строковое значение, сохраненное в БД type - тип получаемой обратной связи, строковое значение, сохраненное в БД
username - имя пользователя username - имя пользователя
@@ -50,6 +50,26 @@ def select_message_from_database(type: str, username):
if conn: if conn:
conn.close() 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(): def telegram_bot():
@bot.message_handler(commands=['start']) @bot.message_handler(commands=['start'])
def send_welcome(message): def send_welcome(message):
@@ -66,9 +86,9 @@ def telegram_bot():
message_id=message.message_id) message_id=message.message_id)
bot.send_sticker(message.chat.id, random_stick_hello) bot.send_sticker(message.chat.id, random_stick_hello)
sleep(0.3) sleep(0.3)
except : except:
if LOGS: 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) markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
item1 = types.KeyboardButton("Предложить свой пост") item1 = types.KeyboardButton("Предложить свой пост")
@@ -78,11 +98,11 @@ def telegram_bot():
markup.add(item1, item2, item3) markup.add(item1, item2, item3)
try: try:
username = message.from_user.first_name 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) bot.send_message(message.chat.id, hello_message, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK)
except: except:
if LOGS: 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.register_next_step_handler(message, go_send_messages)
@bot.message_handler(commands=['end_command']) @bot.message_handler(commands=['end_command'])
@@ -94,7 +114,7 @@ def telegram_bot():
item4 = types.KeyboardButton("Сказать пока!") item4 = types.KeyboardButton("Сказать пока!")
markup.add(item1, item2, item3, item4) markup.add(item1, item2, item3, item4)
bot.send_message(message.chat.id, bot.send_message(message.chat.id,
"Выбери нужную кнопку ниже".format( "Выбери нужную кнопку внизу экрана".format(
message.from_user, bot.get_me()), message.from_user, bot.get_me()),
parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK)
bot.register_next_step_handler(message, go_send_messages) bot.register_next_step_handler(message, go_send_messages)
@@ -105,15 +125,15 @@ def telegram_bot():
try: try:
markup = types.ReplyKeyboardRemove() markup = types.ReplyKeyboardRemove()
username = message.from_user.first_name 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') bot.send_message(message.chat.id, suggest_news, parse_mode='html')
sleep(0.3) sleep(0.3)
username = message.from_user.first_name 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) msg = bot.send_message(message.chat.id, suggest_news_2,parse_mode='html', reply_markup=markup)
except: except:
if LOGS: if LOGS:
bot.send_message(IMPORTANT_LOGS, 'Не сработала предложка') bot.send_message(IMPORTANT_LOGS, error_message_from_db(10))
#logging #logging
if LOGS: if LOGS:
bot.forward_message(chat_id=GROUP_FOR_LOGS, bot.forward_message(chat_id=GROUP_FOR_LOGS,
@@ -123,7 +143,7 @@ def telegram_bot():
elif message.text == "Связаться с админами": elif message.text == "Связаться с админами":
username = message.from_user.first_name 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") msg = bot.send_message(message.chat.id, connect_with_admin, parse_mode="html")
#logging #logging
if LOGS: if LOGS:
@@ -135,7 +155,7 @@ def telegram_bot():
elif message.text == "Удалить пост": elif message.text == "Удалить пост":
#TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать #TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать
username = message.from_user.first_name 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") msg = bot.send_message(message.chat.id, del_message, parse_mode="html")
#logging #logging
if LOGS: if LOGS:
@@ -150,19 +170,19 @@ def telegram_bot():
number_stick_bye = random.randint(1, len(name_stick_bye)) number_stick_bye = random.randint(1, len(name_stick_bye))
random_stick_bye = open(name_stick_bye[number_stick_bye], 'rb') random_stick_bye = open(name_stick_bye[number_stick_bye], 'rb')
bot.send_sticker(message.chat.id, random_stick_bye) bot.send_sticker(message.chat.id, random_stick_bye)
except ApiTelegramException.with_traceback('tb') as r: except ApiTelegramException:
if LOGS: if LOGS:
bot.send_message(IMPORTANT_LOGS, r) bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(11))
markup = types.ReplyKeyboardRemove() markup = types.ReplyKeyboardRemove()
try: try:
username = message.from_user.first_name 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, bot.send_message(message.chat.id, bye_message,
parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK)
except: except:
if LOGS: if LOGS:
bot.send_message(IMPORTANT_LOGS, 'Что-то не так с прощанием') bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(6))
if LOGS: if LOGS:
#logging #logging
@@ -173,11 +193,11 @@ def telegram_bot():
else: else:
try: try:
username = message.from_user.first_name username = message.from_user.first_name
user_error = select_message_from_database('user_error', username) 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) bot.send_message(message.chat.id, user_error, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK)
except: except:
if LOGS: if LOGS:
bot.send_message(IMPORTANT_LOGS, 'Что-то не так с пользовательским вводом') bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message_from_db(9))
#logging #logging
if LOGS: if LOGS:
bot.forward_message(chat_id=GROUP_FOR_LOGS, bot.forward_message(chat_id=GROUP_FOR_LOGS,
@@ -231,10 +251,12 @@ def telegram_bot():
pass pass
except: except:
if LOGS: 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 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) bot.send_message(message.chat.id, success_send_message, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK)
after_post(message=message) after_post(message=message)
@@ -246,11 +268,11 @@ def telegram_bot():
message_id=message.message_id message_id=message.message_id
) )
username = message.from_user.first_name 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) bot.send_message(message.chat.id, question, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK)
except: except:
if LOGS: 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) 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) bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
except: except:
if LOGS: 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': elif call.data == 'post_post_post' and call.message.content_type == 'photo':
try: try:
bot.send_photo( bot.send_photo(
@@ -274,13 +296,13 @@ def telegram_bot():
bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
except: except:
if LOGS: 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': elif call.data == 'decline':
try: try:
bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
except: except:
if LOGS: if LOGS:
logs(IMPORTANT_LOGS, 'Не удалось отклонить пост') logs(IMPORTANT_LOGS, error_message_from_db(1))
def logs(chat_name, text_error): def logs(chat_name, text_error):
bot.send_message(chat_id=chat_name, text=text_error) bot.send_message(chat_id=chat_name, text=text_error)