389 lines
20 KiB
Python
389 lines
20 KiB
Python
import configparser
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
from time import sleep
|
||
from db import BotDB
|
||
import telebot
|
||
import random
|
||
from datetime import datetime
|
||
import time
|
||
from telebot import types
|
||
from telebot.apihelper import ApiTelegramException
|
||
|
||
#Настройки
|
||
config_path = os.path.join(sys.path[0], 'settings.ini')
|
||
config = configparser.ConfigParser()
|
||
config.read(config_path)
|
||
#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')
|
||
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
|
||
LOGS = config.getboolean('Settings', 'logs')
|
||
|
||
#Инициализируем бота и базку
|
||
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: Тексты приветствий вынести в отдельный файл
|
||
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
|
||
if LOGS:
|
||
bot.forward_message(chat_id=GROUP_FOR_LOGS,
|
||
from_chat_id=message.chat.id,
|
||
message_id=message.message_id)
|
||
bot.send_sticker(message.chat.id, random_stick_hello)
|
||
sleep(0.3)
|
||
except:
|
||
if LOGS:
|
||
bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(7))
|
||
|
||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||
item1 = types.KeyboardButton("Предложить свой пост")
|
||
item2 = types.KeyboardButton("Связаться с админами")
|
||
item3 = types.KeyboardButton("Удалить пост")
|
||
item4 = types.KeyboardButton("Войти в режим стендапа")
|
||
|
||
markup.add(item1, item2, item3, item4)
|
||
try:
|
||
user_id = message.from_user.id
|
||
first_name = message.from_user.first_name
|
||
full_name = message.from_user.full_name
|
||
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)
|
||
if BotDB.user_exists(user_id):
|
||
pass
|
||
else:
|
||
BotDB.add_new_user_in_db(user_id, first_name, full_name, username, is_bot, language_code, date_added, date_changed)
|
||
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:
|
||
if LOGS:
|
||
bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(8))
|
||
bot.register_next_step_handler(message, go_send_messages)
|
||
|
||
@bot.message_handler(commands=['end_command'])
|
||
def after_post(message):
|
||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||
item1 = types.KeyboardButton("Предложить свой пост")
|
||
item2 = types.KeyboardButton("Связаться с админами")
|
||
item3 = types.KeyboardButton("Удалить пост")
|
||
item4 = types.KeyboardButton("Войти в режим стендапа")
|
||
item5 = types.KeyboardButton("Сказать пока!")
|
||
markup.add(item1, item2, item3, item4, item5)
|
||
bot.send_message(message.chat.id,
|
||
"Выбери нужную кнопку внизу экрана".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)
|
||
|
||
def go_send_messages(message):
|
||
global msg
|
||
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)
|
||
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:
|
||
if LOGS:
|
||
bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(10))
|
||
#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_post)
|
||
|
||
elif message.text == "Связаться с админами":
|
||
first_name = message.from_user.first_name
|
||
connect_with_admin = BotDB.get_message_from_db('connect_with_admin', first_name)
|
||
msg = bot.send_message(message.chat.id, connect_with_admin, parse_mode="html")
|
||
#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 == "Войти в режим стендапа":
|
||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||
item1 = types.KeyboardButton("Высказаться")
|
||
item2 = types.KeyboardButton("Послушать")
|
||
item3 = types.KeyboardButton("Вернуться в меню")
|
||
markup.add(item1, item2, item3)
|
||
first_name = message.from_user.first_name
|
||
message_for_standup = BotDB.get_message_from_db('message_for_standup', first_name)
|
||
msg = bot.send_message(message.chat.id, message_for_standup, parse_mode='html', reply_markup=markup, disable_web_page_preview=not PREVIEW_LINK)
|
||
#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, standup)
|
||
|
||
|
||
elif message.text == "Удалить пост":
|
||
#TODO: требует автоматизации. На входе говорим пришли мне пост, на выходе получаем идентификатор поста, удаляем из ТГ. Насчет удаления из ВК надо подумать
|
||
first_name = message.from_user.first_name
|
||
del_message = BotDB.get_message_from_db('del_message', first_name)
|
||
msg = bot.send_message(message.chat.id, del_message, parse_mode="html")
|
||
#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 == "Сказать пока!":
|
||
try:
|
||
name_stick_bye = list(Path('Stick').rglob('Universal_*'))
|
||
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:
|
||
if LOGS:
|
||
bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(11))
|
||
|
||
markup = types.ReplyKeyboardRemove()
|
||
try:
|
||
first_name = message.from_user.first_name
|
||
bye_message = BotDB.get_message_from_db('bye_message', first_name)
|
||
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(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(6))
|
||
|
||
if LOGS:
|
||
#logging
|
||
bot.forward_message(chat_id=GROUP_FOR_LOGS,
|
||
from_chat_id=message.chat.id,
|
||
message_id=message.message_id)
|
||
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:
|
||
if LOGS:
|
||
bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(9))
|
||
#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(message, callback=go_send_messages)
|
||
|
||
def standup(message):
|
||
# Клавиатуру добавляем
|
||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||
item1 = types.KeyboardButton("Высказаться")
|
||
item2 = types.KeyboardButton("Послушать")
|
||
item3 = types.KeyboardButton("Вернуться в меню")
|
||
markup.add(item1, item2, item3)
|
||
try:
|
||
if message.text == 'Высказаться' or message.text == 'Высказаться еще':
|
||
markup = types.ReplyKeyboardRemove()
|
||
msg = bot.send_message(chat_id=message.chat.id, text='Пришли мне свое голосовое сообщение', reply_markup=markup)
|
||
bot.register_next_step_handler(msg, save_voice_message)
|
||
elif message.text == 'Послушать':
|
||
check_audio = BotDB.check_listen_audio(user_id=message.from_user.id)
|
||
list_audio = list(check_audio)
|
||
if list_audio == []:
|
||
msg = bot.send_message(message.chat.id, 'Прости, ты прослушал все аудио. Возвращайся позже', reply_markup=markup)
|
||
bot.register_next_step_handler(msg, standup)
|
||
else:
|
||
number_element = random.randint(0, len(list_audio) - 1)
|
||
audio_for_user = check_audio[number_element]
|
||
path = Path(f'voice_users/{audio_for_user}.ogg')
|
||
voice = open(path, 'rb')
|
||
#Маркируем сообщение как прослушанное
|
||
BotDB.mark_listened_audio(audio_for_user, user_id=message.from_user.id)
|
||
|
||
msg = bot.send_voice(message.chat.id, voice=voice, reply_markup=markup)
|
||
bot.register_next_step_handler(msg, standup)
|
||
elif message.text == 'Вернуться в меню':
|
||
after_post(message=message)
|
||
except:
|
||
if LOGS:
|
||
bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(4))
|
||
|
||
def save_voice_message(message):
|
||
if message.content_type == 'voice':
|
||
file_name = ''
|
||
file_id = 1
|
||
#Проверяем что запись о файле есть в базе данных
|
||
is_having_audio_from_user = BotDB.get_last_user_audio_record(user_id=message.from_user.id)
|
||
if is_having_audio_from_user is False:
|
||
#Если нет, то генерируем имя файла
|
||
file_name = f'message_from_{message.from_user.id}_number_{file_id}'
|
||
else:
|
||
#Иначе берем последнюю запись из БД, добавляем к ней 1, и создаем новую запись
|
||
file_name = BotDB.get_path_for_audio_record(user_id=message.from_user.id)
|
||
file_id = BotDB.get_id_for_audio_record(message.from_user.id) + 1
|
||
path = Path(f'voice_users/{file_name}.ogg')
|
||
if path.exists():
|
||
file_name = f'message_from_{message.from_user.id}_number_{file_id}'
|
||
else:
|
||
pass
|
||
#Собираем инфо о сообщении
|
||
author_id = message.from_user.id
|
||
time_UTC = int(time.time())
|
||
date_added = datetime.fromtimestamp(time_UTC)
|
||
#Сохраняем в базку
|
||
BotDB.add_audio_record(file_name, author_id, date_added, 0, file_id)
|
||
#Сохраняем файл на сервер
|
||
file_info = bot.get_file(message.voice.file_id)
|
||
downloaded_file = bot.download_file(file_info.file_path)
|
||
with open(f'voice_users/{file_name}.ogg', 'wb') as new_file:
|
||
new_file.write(downloaded_file)
|
||
#инициализируем кнопки
|
||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||
item1 = types.KeyboardButton("Вернуться в меню")
|
||
markup.add(item1)
|
||
bot.send_message(chat_id=message.chat.id, text='Окей, сохранил!', reply_markup=markup)
|
||
#menu_standup(message=message)
|
||
bot.register_next_step_handler(message, standup)
|
||
else:
|
||
#TODO: Если пришлют фото, он не работает
|
||
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
||
item1 = types.KeyboardButton("Вернуться в меню")
|
||
markup.add(item1)
|
||
msg = bot.send_message(chat_id=message.chat.id, text='Я тебя не понимаю, запиши голосовое', reply_markup=markup)
|
||
bot.register_next_step_handler(msg, standup)
|
||
|
||
def resend_message_in_group_for_post(message):
|
||
markup = types.InlineKeyboardMarkup(row_width=1)
|
||
item1 = types.InlineKeyboardButton("Опубликовать", callback_data='post_post_post')
|
||
item2 = types.InlineKeyboardButton("Отклонить", callback_data='decline')
|
||
markup.add(item1, item2)
|
||
try:
|
||
if message.content_type == 'text':
|
||
post_text = message.text.lower()
|
||
if post_text.find('неанон') != -1 or post_text.find('не анон') != -1:
|
||
bot.send_message(
|
||
#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
|
||
)
|
||
elif post_text.find('анон') != -1:
|
||
bot.send_message(
|
||
# TODO: GROUP_FOR_POST
|
||
chat_id=GROUP_FOR_POST,
|
||
text=f'Пост из ТГ:\n{message.text}\n\nПост опубликован анонимно',
|
||
reply_markup=markup
|
||
)
|
||
else:
|
||
bot.send_message(
|
||
# 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
|
||
)
|
||
elif message.content_type == 'photo' and message.media_group_id == None:
|
||
post_text_for_photo = message.caption.lower()
|
||
bot.send_photo(
|
||
# TODO: GROUP_FOR_POST
|
||
chat_id=GROUP_FOR_POST,
|
||
caption=f'Пост из ТГ:\n{post_text_for_photo}\n\nАвтор поста: {message.from_user.first_name} @{message.from_user.username}',
|
||
photo=message.photo[-1].file_id,
|
||
reply_markup=markup
|
||
)
|
||
#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:
|
||
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)
|
||
|
||
username = message.from_user.first_name
|
||
success_send_message = BotDB.get_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)
|
||
|
||
def resend_message_in_group_for_message(message):
|
||
try:
|
||
bot.forward_message(chat_id=GROUP_FOR_MESSAGE,
|
||
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)
|
||
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=BotDB.get_error_message_from_db(4))
|
||
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:
|
||
if LOGS:
|
||
bot.send_message(chat_id=IMPORTANT_LOGS, text=BotDB.get_error_message_from_db(3))
|
||
elif call.data == 'post_post_post' and call.message.content_type == 'photo':
|
||
try:
|
||
bot.send_photo(
|
||
chat_id=MAIN_PUBLIC,
|
||
caption=call.message.caption,
|
||
photo=call.message.photo[-1].file_id,
|
||
)
|
||
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=BotDB.get_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:
|
||
bot.send_message(IMPORTANT_LOGS, BotDB.get_error_message_from_db(1))
|
||
|
||
|
||
if __name__ == '__main__':
|
||
telegram_bot()
|
||
try:
|
||
bot.polling(none_stop=True)
|
||
bot.enable_save_next_step_handlers(delay=2)
|
||
bot.load_next_step_handlers()
|
||
except ConnectionError as e:
|
||
if LOGS:
|
||
bot.send_message(IMPORTANT_LOGS, "Ошибка соединения, потерял связь")
|
||
except Exception as r:
|
||
if LOGS:
|
||
bot.send_message(IMPORTANT_LOGS, "Произошло что-то непредвиденное, хелп")
|
||
finally:
|
||
if LOGS:
|
||
bot.send_message(IMPORTANT_LOGS, 'Я упал, помогите') |