diff --git a/db.py b/db.py index 62bfc72..9ed2342 100644 --- a/db.py +++ b/db.py @@ -16,6 +16,7 @@ class BotDB: self.conn = sqlite3.connect(db_file, check_same_thread=False) self.cursor = self.conn.cursor() + # TODO: Deprecated, удалить def get_message_from_db(self, type: str, username): """Функция для запроса к базе данных и получения сообщений для бота. В аргументы передаются: type - тип получаемой обратной связи, строковое значение, сохраненное в БД @@ -35,6 +36,7 @@ class BotDB: except sqlite3.Error as error: print(error) + # TODO: Deprecated. Остался только в voice боте, удалить и оттуда def get_error_message_from_db(self, id: int): """Функция для запроса к базе данных и получения сообщений ошибки. В аргументы передаются: id - идентификатор ошибки diff --git a/main.py b/main.py index 08ee2da..bd9488b 100644 --- a/main.py +++ b/main.py @@ -10,12 +10,14 @@ from datetime import datetime import time from telebot import types from telebot.apihelper import ApiTelegramException +import messages +import traceback -#Настройки +# Настройки config_path = os.path.join(sys.path[0], 'settings.ini') config = configparser.ConfigParser() config.read(config_path) -#TELEGRAM +# TELEGRAM BOT_TOKEN = config.get('Telegram', 'BOT_TOKEN') GROUP_FOR_POST = config.get('Telegram', 'group_for_posts') GROUP_FOR_MESSAGE = config.get('Telegram', 'group_for_message') @@ -23,35 +25,34 @@ MAIN_PUBLIC = config.get('Telegram', 'main_public') GROUP_FOR_LOGS = config.get('Telegram', 'group_for_logs') IMPORTANT_LOGS = config.get('Telegram', 'important_logs') PREVIEW_LINK = config.getboolean('Telegram', 'PREVIEW_LINK') -#SETTINGS +# SETTINGS LOGS = config.getboolean('Settings', 'logs') TEST = config.getboolean('Settings', 'test') - -#Инициализируем бота и базку +# Инициализируем бота и базку bot = telebot.TeleBot(BOT_TOKEN, parse_mode=None) BotDB = BotDB('tg-bot-database') + def telegram_bot(): @bot.message_handler(commands=['start']) def send_welcome(message): - #TODO: Здесь переписать через randint - #TODO: Тексты приветствий вынести в отдельный файл + # TODO: Здесь переписать через randint + # TODO: Тексты приветствий вынести в отдельный файл try: name_stick_hello = list(Path('Stick').rglob('Hello_*')) number_stick_hello = random.randint(1, len(name_stick_hello)) random_stick_hello = open(name_stick_hello[number_stick_hello], 'rb') - #logging + # logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, - from_chat_id=message.chat.id, - message_id=message.message_id) + from_chat_id=message.chat.id, + message_id=message.message_id) bot.send_sticker(message.chat.id, random_stick_hello) sleep(0.3) - except: + except Exception as e: if LOGS: - bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(7)) - + bot.send_message(IMPORTANT_LOGS, f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") try: user_id = message.from_user.id first_name = message.from_user.first_name @@ -59,15 +60,15 @@ def telegram_bot(): is_bot = message.from_user.is_bot username = message.from_user.username language_code = message.from_user.language_code - time_UTC = int(time.time()) - date_added = datetime.fromtimestamp(time_UTC) - date_changed = datetime.fromtimestamp(time_UTC) + time_utc = int(time.time()) + date_added = datetime.fromtimestamp(time_utc) + date_changed = datetime.fromtimestamp(time_utc) markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True) item1 = types.KeyboardButton("📢Предложить свой пост") item2 = types.KeyboardButton("📩Связаться с админами") # TODO: Выпилил, удалить - #item3 = types.KeyboardButton("❌Удалить пост") + # item3 = types.KeyboardButton("❌Удалить пост") if BotDB.user_exists(user_id): is_need_sticker = BotDB.get_info_about_stickers(user_id=message.from_user.id) if is_need_sticker == 0: @@ -77,7 +78,8 @@ def telegram_bot(): else: markup.add(item1, item2) else: - BotDB.add_new_user_in_db(user_id, first_name, full_name, username, is_bot, language_code, date_added, date_changed) + BotDB.add_new_user_in_db(user_id, first_name, full_name, username, is_bot, language_code, date_added, + date_changed) is_need_sticker = BotDB.get_info_about_stickers(user_id=message.from_user.id) if is_need_sticker == 0: item5 = types.KeyboardButton("🤪Хочу стикеры") @@ -85,11 +87,12 @@ def telegram_bot(): markup.add(item1, item2, item5) else: markup.add(item1, item2) - hello_message = BotDB.get_message_from_db('start_message', first_name) - bot.send_message(message.chat.id, hello_message, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) - except: + hello_message = messages.get_message(__get_first_name(message), 'HELLO_MESSAGE') + bot.send_message(message.chat.id, hello_message, parse_mode='html', reply_markup=markup, + disable_web_page_preview=not PREVIEW_LINK) + except Exception as e: if LOGS: - bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(8)) + bot.send_message(IMPORTANT_LOGS, f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") bot.register_next_step_handler(message, go_send_messages) @bot.message_handler(commands=['end_command']) @@ -98,7 +101,7 @@ def telegram_bot(): item1 = types.KeyboardButton("📢Предложить свой пост") item2 = types.KeyboardButton("📩Связаться с админами") # TODO: Скрыл, удалить обработчик - #item3 = types.KeyboardButton("❌Удалить пост") + # item3 = types.KeyboardButton("❌Удалить пост") item5 = types.KeyboardButton("👋🏼Сказать пока!") markup.add(item1, item2, item5) bot.send_message(message.chat.id, @@ -112,17 +115,16 @@ def telegram_bot(): if message.text == '📢Предложить свой пост': try: markup = types.ReplyKeyboardRemove() - first_name = message.from_user.first_name - suggest_news = BotDB.get_message_from_db('suggest_news', first_name) + suggest_news = messages.get_message(__get_first_name(message), 'SUGGEST_NEWS') bot.send_message(message.chat.id, suggest_news, parse_mode='html') sleep(0.3) - first_name = message.from_user.first_name - suggest_news_2 = BotDB.get_message_from_db('suggest_news_2', first_name) - msg = bot.send_message(message.chat.id, suggest_news_2,parse_mode='html', reply_markup=markup) - except: + suggest_news_2 = messages.get_message(__get_first_name(message), 'SUGGEST_NEWS_2') + msg = bot.send_message(message.chat.id, suggest_news_2, parse_mode='html', reply_markup=markup) + except Exception as e: if LOGS: - bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(10)) - #logging + bot.send_message(IMPORTANT_LOGS, + f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") + # logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, from_chat_id=message.chat.id, @@ -130,23 +132,20 @@ def telegram_bot(): bot.register_next_step_handler(msg, resend_message_in_group_for_post) elif message.text == "📩Связаться с админами": - first_name = message.from_user.first_name - connect_with_admin = BotDB.get_message_from_db('connect_with_admin', first_name) + connect_with_admin = messages.get_message(__get_first_name(message), 'CONNECT_WITH_ADMIN') msg = bot.send_message(message.chat.id, connect_with_admin, parse_mode="html") - #logging + # logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, from_chat_id=message.chat.id, message_id=message.message_id) bot.register_next_step_handler(msg, resend_message_in_group_for_message) - elif message.text == "❌Удалить пост": - #TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать - first_name = message.from_user.first_name - del_message = BotDB.get_message_from_db('del_message', first_name) + # TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать + del_message = messages.get_message(__get_first_name(message), 'DEL_MESSAGE') msg = bot.send_message(message.chat.id, del_message, parse_mode="html") - #logging + # logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, from_chat_id=message.chat.id, @@ -159,22 +158,22 @@ 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: + except ApiTelegramException as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(11)) - + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") markup = types.ReplyKeyboardRemove() try: - first_name = message.from_user.first_name - bye_message = BotDB.get_message_from_db('bye_message', first_name) + bye_message = messages.get_message(__get_first_name(message), 'BYE_MESSAGE') bot.send_message(message.chat.id, bye_message, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK) - except: + except Exception as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(6)) + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") if LOGS: - #logging + # logging bot.forward_message(chat_id=GROUP_FOR_LOGS, from_chat_id=message.chat.id, message_id=message.message_id) @@ -183,7 +182,7 @@ def telegram_bot(): item1 = types.KeyboardButton("📢Предложить свой пост") item2 = types.KeyboardButton("📩Связаться с админами") # TODO: Скрыл кнопку, убрать обработчик позднее - #item3 = types.KeyboardButton("❌Удалить пост") + # item3 = types.KeyboardButton("❌Удалить пост") item5 = types.KeyboardButton("👋🏼Сказать пока!") markup.add(item1, item2, item5) try: @@ -191,20 +190,24 @@ def telegram_bot(): bot.forward_message(chat_id=GROUP_FOR_LOGS, from_chat_id=message.chat.id, message_id=message.message_id) - bot.send_message(message.chat.id, text='Хорошо, лови, добавить можно отсюда: https://t.me/addstickers/love_biysk', reply_markup=markup) + bot.send_message(message.chat.id, + text='Хорошо, лови, добавить можно отсюда: https://t.me/addstickers/love_biysk', + reply_markup=markup) bot.register_next_step_handler(message, callback=go_send_messages) - except ApiTelegramException: + except ApiTelegramException as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(12)) + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") else: try: - first_name = message.from_user.first_name - user_error = BotDB.get_message_from_db('user_error', first_name) - bot.send_message(message.chat.id, user_error, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK) - except: + user_error = messages.get_message(__get_first_name(message), 'USER_ERROR') + bot.send_message(message.chat.id, user_error, parse_mode='html', + disable_web_page_preview=not PREVIEW_LINK) + except Exception as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(9)) - #logging + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") + # logging if LOGS: bot.forward_message(chat_id=GROUP_FOR_LOGS, from_chat_id=message.chat.id, @@ -221,7 +224,7 @@ def telegram_bot(): post_text = message.text.lower() if post_text.find('неанон') != -1 or post_text.find('не анон') != -1: bot.send_message( - #TODO: GROUP_FOR_POST + # TODO: GROUP_FOR_POST chat_id=GROUP_FOR_POST, text=f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {message.from_user.first_name} @{message.from_user.username}', reply_markup=markup @@ -240,7 +243,7 @@ def telegram_bot(): text=f'Пост из ТГ:\n{post_text}\n\nАвтор поста: {message.from_user.first_name} @{message.from_user.username}', reply_markup=markup ) - elif message.content_type == 'photo' and message.media_group_id == None: + elif message.content_type == 'photo' and message.media_group_id is None: post_text_for_photo = message.caption.lower() if post_text_for_photo.find('неанон') != -1 or post_text_for_photo.find('не анон') != -1: bot.send_photo( @@ -266,21 +269,20 @@ def telegram_bot(): photo=message.photo[-1].file_id, reply_markup=markup ) - #TODO: Не понятна реализация с альбомами от слова совсем - #elif message.content_type == 'photo' and message.media_group_id != None: + # TODO: Не понятна реализация с альбомами от слова совсем + # elif message.content_type == 'photo' and message.media_group_id != None: # bot.forward_message(chat_id=IMPORTANT_LOGS, from_chat_id=message.chat.id, message_id=message.message_id ) else: pass - except: + except Exception as e: if LOGS: - username = message.from_user.first_name - error_message = str(BotDB.get_error_message_from_db(5)).replace('username', username) - bot.send_message(chat_id=IMPORTANT_LOGS, text=error_message) + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") - username = message.from_user.first_name - success_send_message = BotDB.get_message_from_db('success_send_message', username) + success_send_message = messages.get_message(__get_first_name(message), 'SUCCESS_SEND_MESSAGE') - 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) def resend_message_in_group_for_message(message): @@ -288,27 +290,27 @@ def telegram_bot(): from_chat_id=message.chat.id, message_id=message.message_id ) - username = message.from_user.first_name - question = BotDB.get_message_from_db('question', username) + question = messages.get_message(__get_first_name(message), 'QUESTION') bot.send_message(message.chat.id, question, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK) try: pass - except: + except Exception as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(4)) + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") after_post(message=message) - - #Админка + # Админка @bot.callback_query_handler(func=lambda call: True) def post_for_group(call): if call.data == 'post_post_post' and call.message.content_type == 'text': try: bot.send_message(chat_id=MAIN_PUBLIC, text=call.message.text) bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) - except: + except Exception as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(3)) + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") elif call.data == 'post_post_post' and call.message.content_type == 'photo': try: bot.send_photo( @@ -317,15 +319,20 @@ def telegram_bot(): photo=call.message.photo[-1].file_id, ) bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) - except: + except Exception as e: if LOGS: - bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(2)) + bot.send_message(chat_id=IMPORTANT_LOGS, + text=f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") elif call.data == 'decline': try: bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id) - except: + except Exception as e: if LOGS: - bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(1)) + bot.send_message(IMPORTANT_LOGS, + f"Произошла ошибка: {str(e)}\n\nTraceback:\n{traceback.format_exc()}") + + def __get_first_name(message): + return message.from_user.first_name if __name__ == '__main__': @@ -336,4 +343,4 @@ if __name__ == '__main__': bot.enable_save_next_step_handlers(delay=2) bot.load_next_step_handlers() except (ConnectionError, Exception): - print("Произошла ошибка, перезапуск бота") \ No newline at end of file + print(f"Произошла ошибка: {str(Exception)}\n\nTraceback:\n{traceback.format_exc()}") diff --git a/messages.py b/messages.py new file mode 100644 index 0000000..26c4824 --- /dev/null +++ b/messages.py @@ -0,0 +1,40 @@ + + +def get_message(username: str, type_message: str): + constants = { + 'HELLO_MESSAGE': "Привет, username!👋🏼&Меня зовут Виби, я бот канала 'Влюбленный Бийск'❤🤖" + "&Я был создан для того, чтобы помочь тебе выложить пост в наш канал и если это необходимо, связаться с админами ✍✉" + "&Так же я могу выдать тебе набор стикеров, где я буду главным героем🦸‍♂" + "&Дать возможность высказаться в нашем стендап режиме🎤&Послушать о чем говорит наш город🎧" + "&Предлагай свой пост мне и я обязательно его опубликую😉" + "&Для продолжения взаимодействия воспользуйся меню внизу твоего дисплея⬇" + "&&P.S. Иногда я зависаю по неизвестным причинам, если от меня нет ответа, введи в чат команду /start, это перезапустит сценарий." + "&&Основная группа в ВК: https://vk.com/love_bsk" + "&Основной канал в ТГ: https://t.me/love_bsk", + 'SUGGEST_NEWS': "username, окей, жду от тебя текст поста🙌🏼" + "&В данный момент я работаю в тестовом режиме, поэтому к посту можно прикрепить не более одного фото и никаких аудио или видео👻" + "&&Обещаю, я научусь их обрабатывать, но позже🤝🤖", + 'SUGGEST_NEWS_2': "Обрати внимание, что я умный и смогу из твоего текста понять команды указанные ниже😉" + "&Если хочешь чтобы пост был опубликован анонимно, напиши в любом месте своего поста слово 'анон'." + "&Если хочешь опубликовать пост не анонимно, то напиши 'не анон', 'неанон' или не пиши ничего." + "&&Я обучен только на команды, указанные мной выше👆" + "&‼Проверь, чтобы указание авторства было выполнено так как я попросил." + "&Пост будет опубликован только в группе ТГ📩", + "CONNECT_WITH_ADMIN": "username, напиши свое обращение или предложение✍️" + "&Мы рассмотрим и ответим тебе в ближайшее время☺️❤️", + "DEL_MESSAGE": "username, напиши свое обращение или предложение✍" + "&Мы рассмотрим и ответим тебе в ближайшее время☺❤", + "BYE_MESSAGE": "Если позднее захочешь предложить еще один пост или обратиться к админам с вопросом, то просто пришли в чат команду 👉 /start" + "&&И тебе пока!👋🏼❤️", + "USER_ERROR": "Увы, я не понимаю тебя😐💔 Выбери один из пунктов в нижнем меню, а затем пиши.", + "QUESTION": "Сообщение успешно отправлено❤️ Ответим, как только сможем😉", + "SUCCESS_SEND_MESSAGE": "Пост успешно отправлен❤️ Ожидай одобрения😊", + "MESSAGE_FOR_STANDUP": "Отлично, ты вошел в режим стендапа 📣" + "&Это свободное пространство, в котором может высказаться каждый житель нашего города, и он будет услышан🙌🏼" + "&Для того чтобы высказаться, нажми кнопку: 'Высказаться' и запиши голосовое сообщение, оно выпадет анонимно кому-то другому🗣" + "&Для того чтобы послушать о чем говорит наш город, нажми кнопку: 'Послушать'👂" + "&Ты можешь анонимно пообщаться, поделиться чем-то важным, обратиться напрямую к жителям🤝 Также можешь выступить перед аудиторией (спеть песню, рассказать стихотворение, шутку)🎤" + "&❗️Но пожалуйста не оскорбляй никого, и будь вежлив." + } + message = constants[type_message] + return message.replace('username', username).replace('&', '\n') diff --git a/settings_example.ini b/settings_example.ini index b088ac8..6a953a6 100644 --- a/settings_example.ini +++ b/settings_example.ini @@ -1,9 +1,13 @@ [Telegram] -bot_token = 5719450198:AAEI6hQgwiIcxMLGvVrPdopWqVlwHqR0QoU +bot_token = 000000000:AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA preview_link = false -main_public = @love_biysk -group_for_posts = -793789724 -group_for_message = -736077298 -group_for_logs = -721685792 -important_logs = -625900899 -test_channel = -1001725605158 \ No newline at end of file +main_public = @test +group_for_posts = -00000000 +group_for_message = -00000000 +group_for_logs = -00000000 +important_logs = -00000000 +test_channel = -000000000000 + +[Settings] +logs = true +test = false \ No newline at end of file