Merge remote-tracking branch 'origin/voice-1' into merge-voice-1

helper_bot/handlers/private/private_handlers.py

@@ -18,8 +18,9 @@ from helper_bot.middlewares.blacklist_middleware import BlacklistMiddleware
 from helper_bot.utils import messages
 from helper_bot.utils.base_dependency_factory import get_global_instance
 from helper_bot.utils.helper_func import get_first_name, get_text_message, send_text_message, send_photo_message, \
-    send_media_group_message_to_private_chat, prepare_media_group_from_middlewares, check_username_and_full_name, \
-    send_video_message, send_video_note_message, send_audio_message, send_voice_message, add_in_db_media
+    send_media_group_message_to_private_chat, prepare_media_group_from_middlewares, send_video_message, \
+    send_video_note_message, send_audio_message, send_voice_message, add_in_db_media, \
+    update_user_info, check_user_emoji
 from logs.custom_logger import logger
 
 private_router = Router()
@@ -42,6 +43,35 @@ BotDB = bdf.get_db()
 # Expose sleep for tests (tests patch helper_bot.handlers.private.private_handlers.sleep)
 sleep = asyncio.sleep
 
+@private_router.message(
+    ChatTypeFilter(chat_type=["private"]),
+    Command("emoji")
+)
+async def handle_emoji_message(message: types.Message, state: FSMContext):
+    await message.forward(chat_id=GROUP_FOR_LOGS)
+    user_emoji = check_user_emoji(message)
+    await state.set_state("START")
+    if user_emoji is not None:
+        await message.answer(f'Твоя эмодзя - {user_emoji}', parse_mode='HTML')
+
+
+@private_router.message(
+    ChatTypeFilter(chat_type=["private"]),
+    Command("restart")
+)
+async def handle_restart_message(message: types.Message, state: FSMContext):
+    try:
+        markup = get_reply_keyboard(BotDB, message.from_user.id)
+        await message.forward(chat_id=GROUP_FOR_LOGS)
+        await state.set_state("START")
+        await update_user_info('love', message)
+        check_user_emoji(message)
+        await message.answer('Я перезапущен!', reply_markup=markup, parse_mode='HTML')
+    except Exception as e:
+        logger.error(f"Произошла ошибка handle_restart_message. Ошибка:{str(e)}")
+        await message.bot.send_message(chat_id=IMPORTANT_LOGS,
+                                       text=f"Произошла ошибка handle_restart_message: {str(e)}\n\nTraceback:\n{traceback.format_exc()}")
+
 
 @private_router.message(
     ChatTypeFilter(chat_type=["private"]),
@@ -60,6 +90,8 @@ async def handle_start_message(message: types.Message, state: FSMContext):
         is_bot = message.from_user.is_bot
         language_code = message.from_user.language_code
         user_id = message.from_user.id
+        await update_user_info('love', message)
+        check_user_emoji(message)
         
         # Проверяем наличие username для логирования
         if not username:

helper_bot/utils/helper_func.py

@@ -1,13 +1,20 @@
 import html
 import os
+import random
 from datetime import datetime, timedelta
+from time import sleep
 
+import emoji
 from aiogram import types
 from aiogram.types import InputMediaPhoto, FSInputFile, InputMediaVideo, InputMediaAudio
 
 from helper_bot.utils.base_dependency_factory import BaseDependencyFactory
 from logs.custom_logger import logger
 
+GROUP_FOR_LOGS = bdf.settings['Telegram']['group_for_logs']
+
+emoji_list = list(emoji.EMOJI_DATA.keys())
+
 
 def safe_html_escape(text: str) -> str:
     """
@@ -458,3 +465,52 @@ def unban_notifier(self):
 
     # Отправка сообщения в канал
     self.bot.send_message(self.GROUP_FOR_MESSAGE, message)
+
+
+async def update_user_info(source: str, message: types.Message):
+    # Собираем данные
+    full_name = message.from_user.full_name
+    username = message.from_user.username
+    first_name = get_first_name(message)
+    is_bot = message.from_user.is_bot
+    language_code = message.from_user.language_code
+    user_id = message.from_user.id
+    current_date = datetime.now()
+    date = current_date.strftime("%Y-%m-%d %H:%M:%S")
+    # Выбираем эмодзю, пробегаемся циклом и смотрим что в базе такого еще не было
+    user_emoji = get_random_emoji()
+
+    if not BotDB.user_exists(user_id):
+        BotDB.add_new_user_in_db(user_id, first_name, full_name, username, is_bot, language_code, user_emoji, date,
+                                 date)
+    else:
+        is_need_update = check_username_and_full_name(user_id, username, full_name)
+        if is_need_update:
+            BotDB.update_username_and_full_name(user_id, username, full_name)
+            if source != 'voice':
+                await message.answer(
+                    f"Давно не виделись! Вижу что ты изменился;) Теперь буду звать тебя: {full_name}")
+            await message.bot.send_message(chat_id=GROUP_FOR_LOGS,
+                                           text=f'Для пользователя: {user_id} обновлены данные в БД.\nНовое имя: {full_name}\nНовый ник:{username}. Новый эмодзи:{user_emoji}')
+        sleep(1)
+    BotDB.update_date_for_user(date, user_id)
+
+
+def check_user_emoji(message: types.Message):
+    user_id = message.from_user.id
+    user_emoji = BotDB.check_emoji_for_user(user_id=user_id)
+    if user_emoji is None:
+        user_emoji = get_random_emoji()
+        BotDB.update_emoji_for_user(user_id=user_id, emoji=user_emoji)
+    return user_emoji
+
+
+def get_random_emoji():
+    attempts = 0
+    while attempts < 100:
+        user_emoji = random.choice(emoji_list)
+        if not BotDB.check_emoji(user_emoji):
+            return user_emoji
+        attempts += 1
+    logger.error("Не удалось найти уникальный эмодзи после нескольких попыток.")
+    return "Эмоджи не определен"

helper_bot/utils/messages.py

@@ -26,7 +26,7 @@ def get_message(username: str, type_message: str):
                               "&Мы рассмотрим и ответим тебе в ближайшее время☺️❤️",
         "DEL_MESSAGE": "username, напиши свое обращение или предложение✍"
                        "&Мы рассмотрим и ответим тебе в ближайшее время☺❤",
-        "BYE_MESSAGE": "Если позднее захочешь предложить еще один пост или обратиться к админам с вопросом, то просто пришли в чат команду 👉 /start"
+        "BYE_MESSAGE": "Если позднее захочешь предложить еще один пост или обратиться к админам с вопросом, то просто пришли в чат команду 👉 /restart"
                        "&&И тебе пока!👋🏼❤️",
         "USER_ERROR": "Увы, я не понимаю тебя😐💔 Выбери один из пунктов в нижнем меню, а затем пиши.",
         "QUESTION": "Сообщение успешно отправлено❤️ Ответим, как только сможем😉",