add new func in voice bot
This commit is contained in:
Andrey
68
voice_bot/handlers/callback_handler.py
Normal file
68
voice_bot/handlers/callback_handler.py
Normal file
@@ -0,0 +1,68 @@
|
||||
import time
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
from aiogram import Router, F
|
||||
from aiogram.types import CallbackQuery
|
||||
|
||||
from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
|
||||
|
||||
callback_router = Router()
|
||||
|
||||
bdf = BaseDependencyFactory()
|
||||
|
||||
GROUP_FOR_LOGS = bdf.settings['Telegram']['group_for_logs']
|
||||
GROUP_FOR_POST = bdf.settings['Telegram']['group_for_posts']
|
||||
IMPORTANT_LOGS = bdf.settings['Telegram']['important_logs']
|
||||
PREVIEW_LINK = bdf.settings['Telegram']['preview_link']
|
||||
LOGS = bdf.settings['Settings']['logs']
|
||||
TEST = bdf.settings['Settings']['test']
|
||||
|
||||
BotDB = bdf.get_db()
|
||||
|
||||
|
||||
@callback_router.callback_query(
|
||||
F.data == "save"
|
||||
)
|
||||
async def save_voice_message(call: CallbackQuery):
|
||||
file_name = ''
|
||||
file_id = 1
|
||||
user_id = BotDB.get_user_id_by_message_id_for_voice_bot(call.message.message_id)
|
||||
# Проверяем что запись о файле есть в базе данных
|
||||
is_having_audio_from_user = BotDB.get_last_user_audio_record(user_id=user_id)
|
||||
if is_having_audio_from_user is False:
|
||||
# Если нет, то генерируем имя файла
|
||||
file_name = f'message_from_{user_id}_number_{file_id}'
|
||||
else:
|
||||
# Иначе берем последнюю запись из БД, добавляем к ней 1, и создаем новую запись
|
||||
file_name = BotDB.get_path_for_audio_record(user_id=user_id)
|
||||
file_id = BotDB.get_id_for_audio_record(user_id) + 1
|
||||
path = Path(f'voice_users/{file_name}.ogg')
|
||||
if path.exists():
|
||||
file_name = f'message_from_{user_id}_number_{file_id}'
|
||||
else:
|
||||
pass
|
||||
# Собираем инфо о сообщении
|
||||
time_UTC = int(time.time())
|
||||
date_added = datetime.fromtimestamp(time_UTC)
|
||||
|
||||
# Сохраняем в базку
|
||||
BotDB.add_audio_record(file_name, user_id, date_added, 0, file_id)
|
||||
|
||||
file_info = await call.message.bot.get_file(file_id=call.message.voice.file_id)
|
||||
downloaded_file = await call.message.bot.download_file(file_path=file_info.file_path)
|
||||
with open(f'voice_users/{file_name}.ogg', 'wb') as new_file:
|
||||
new_file.write(downloaded_file.read())
|
||||
|
||||
await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
|
||||
await call.answer(text='Сохранено!', cache_time=3)
|
||||
|
||||
|
||||
@callback_router.callback_query(
|
||||
F.data == "delete"
|
||||
)
|
||||
async def delete_voice_message(call: CallbackQuery):
|
||||
# Получаем из базы автора + отправляем сообщение + удаляем сообщение из предложки
|
||||
|
||||
await call.bot.delete_message(chat_id=GROUP_FOR_POST, message_id=call.message.message_id)
|
||||
await call.answer(text='Удалено!', cache_time=3)
|
||||
@@ -1,6 +1,5 @@
|
||||
import random
|
||||
import time
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
from aiogram import Router, types, F
|
||||
@@ -11,9 +10,9 @@ from aiogram.types import FSInputFile
|
||||
from helper_bot.filters.main import ChatTypeFilter
|
||||
from helper_bot.middlewares.blacklist_middleware import BlacklistMiddleware
|
||||
from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
|
||||
from helper_bot.utils.helper_func import update_user_info, check_user_emoji
|
||||
from helper_bot.utils.helper_func import update_user_info, check_user_emoji, send_voice_message
|
||||
from logs.custom_logger import logger
|
||||
from voice_bot.keyboards.keyboards import get_main_keyboard
|
||||
from voice_bot.keyboards.keyboards import get_main_keyboard, get_reply_keyboard_for_voice
|
||||
from voice_bot.utils.helper_func import last_message
|
||||
|
||||
voice_router = Router()
|
||||
@@ -21,6 +20,7 @@ voice_router = Router()
|
||||
bdf = BaseDependencyFactory()
|
||||
|
||||
GROUP_FOR_LOGS = bdf.settings['Telegram']['group_for_logs']
|
||||
GROUP_FOR_POST = bdf.settings['Telegram']['group_for_posts']
|
||||
IMPORTANT_LOGS = bdf.settings['Telegram']['important_logs']
|
||||
PREVIEW_LINK = bdf.settings['Telegram']['preview_link']
|
||||
LOGS = bdf.settings['Settings']['logs']
|
||||
@@ -116,6 +116,22 @@ async def start(message: types.Message, state: FSMContext):
|
||||
disable_web_page_preview=not PREVIEW_LINK)
|
||||
|
||||
|
||||
@voice_router.message(
|
||||
ChatTypeFilter(chat_type=["private"]),
|
||||
Command("refresh")
|
||||
)
|
||||
async def refresh_listen_function(message: types.Message, state: FSMContext):
|
||||
await message.forward(chat_id=GROUP_FOR_LOGS)
|
||||
await update_user_info('voice', message)
|
||||
markup = get_main_keyboard()
|
||||
check_user_emoji(message.from_user.id)
|
||||
BotDB.delete_listen_count_for_user(message.from_user.id)
|
||||
await message.answer(
|
||||
text='Прослушивания очищены. Можешь начать слушать заново🤗', disable_web_page_preview=not PREVIEW_LINK,
|
||||
markup=markup)
|
||||
await state.set_state('START')
|
||||
|
||||
|
||||
@voice_router.message(
|
||||
StateFilter("START"),
|
||||
ChatTypeFilter(chat_type=["private"]),
|
||||
@@ -125,47 +141,33 @@ async def standup_write(message: types.Message, state: FSMContext):
|
||||
await message.forward(chat_id=GROUP_FOR_LOGS)
|
||||
markup = types.ReplyKeyboardRemove()
|
||||
await message.answer(text='Хорошо, теперь пришли мне свое голосовое сообщение', reply_markup=markup)
|
||||
message_with_date = last_message()
|
||||
await message.answer(text=message_with_date, parse_mode="html")
|
||||
try:
|
||||
message_with_date = last_message()
|
||||
await message.answer(text=message_with_date, parse_mode="html")
|
||||
except Exception as e:
|
||||
logger.error(f'Не удалось получить дату последнего сообщения - {e}')
|
||||
await state.set_state('STANDUP_WRITE')
|
||||
|
||||
|
||||
@voice_router.message(
|
||||
StateFilter("STANDUP_WRITE"),
|
||||
ChatTypeFilter(chat_type=["private"])
|
||||
ChatTypeFilter(chat_type=["private"]),
|
||||
)
|
||||
async def save_voice_message(message: types.Message, state: FSMContext):
|
||||
async def suggest_voice(message: types.Message, state: FSMContext):
|
||||
logger.info(
|
||||
f"Вызов функции suggest_voice. Пользователь: {message.from_user.id} Имя автора сообщения: {message.from_user.full_name}")
|
||||
await message.forward(chat_id=GROUP_FOR_LOGS)
|
||||
markup = get_main_keyboard()
|
||||
if message.content_type == 'voice':
|
||||
await message.forward(chat_id=GROUP_FOR_LOGS)
|
||||
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)
|
||||
markup_for_voice = get_reply_keyboard_for_voice()
|
||||
# Отправляем аудио в приватный канал
|
||||
sent_message = await send_voice_message(GROUP_FOR_POST, message,
|
||||
message.voice.file_id, markup_for_voice)
|
||||
|
||||
# Сохраняем в базку
|
||||
BotDB.add_audio_record(file_name, author_id, date_added, 0, file_id)
|
||||
# Сохраняем в базу инфо о посте
|
||||
BotDB.set_user_id_and_message_id_for_voice_bot(sent_message.message_id, message.from_user.id)
|
||||
|
||||
file_info = await message.bot.get_file(file_id=message.voice.file_id)
|
||||
downloaded_file = await message.bot.download_file(file_path=file_info.file_path)
|
||||
with open(f'voice_users/{file_name}.ogg', 'wb') as new_file:
|
||||
new_file.write(downloaded_file.read())
|
||||
# Отправляем юзеру ответ и возвращаем его в меню
|
||||
await message.answer(text='Окей, сохранил!👌', reply_markup=markup)
|
||||
await state.set_state('START')
|
||||
else:
|
||||
@@ -180,23 +182,33 @@ async def save_voice_message(message: types.Message, state: FSMContext):
|
||||
ChatTypeFilter(chat_type=["private"]),
|
||||
F.text == '🎧Послушать'
|
||||
)
|
||||
async def standup_listen_audio(message: types.Message, state: FSMContext):
|
||||
async def standup_listen_audio(message: types.Message):
|
||||
check_audio = BotDB.check_listen_audio(user_id=message.from_user.id)
|
||||
list_audio = list(check_audio)
|
||||
markup = get_main_keyboard()
|
||||
if not list_audio:
|
||||
await message.answer(text='Прости, ты прослушал все аудио😔. Возвращайся позже, возможно наша база пополнится',
|
||||
reply_markup=markup)
|
||||
message_with_date = last_message()
|
||||
await message.answer(text=message_with_date, parse_mode="html")
|
||||
try:
|
||||
message_with_date = last_message()
|
||||
await message.answer(text=message_with_date, parse_mode="html")
|
||||
except Exception as e:
|
||||
logger.error(f'Не удалось получить последнюю дату {e}')
|
||||
else:
|
||||
# Получаем эмоджи пользователя
|
||||
number_element = random.randint(0, len(list_audio) - 1)
|
||||
audio_for_user = check_audio[number_element]
|
||||
# Получаем автора записи + эмодзи по нему
|
||||
user_id = BotDB.get_user_id_by_file_name(audio_for_user)
|
||||
user_emoji = BotDB.check_emoji_for_user(user_id)
|
||||
|
||||
path = Path(f'voice_users/{audio_for_user}.ogg')
|
||||
# voice = open(path, 'rb')
|
||||
voice = FSInputFile(path)
|
||||
# Маркируем сообщение как прослушанное
|
||||
BotDB.mark_listened_audio(audio_for_user, user_id=message.from_user.id)
|
||||
await message.bot.send_voice(message.chat.id, voice=voice, reply_markup=markup)
|
||||
await message.answer(text=f'Осталось непрослушанных: <b>{len(check_audio)}</b>', reply_markup=markup)
|
||||
await state.set_state('START')
|
||||
if user_emoji:
|
||||
await message.bot.send_voice(chat_id=message.chat.id, voice=voice, caption=user_emoji, reply_markup=markup)
|
||||
else:
|
||||
await message.bot.send_voice(chat_id=message.chat.id, voice=voice, reply_markup=markup)
|
||||
await message.answer(text=f'Осталось непрослушанных: <b>{len(check_audio) - 1}</b>', reply_markup=markup)
|
||||
|
||||
Reference in New Issue
Block a user