перешёл на aiogram, добавил коменты

.gitignore

@@ -28,7 +28,7 @@ share/python-wheels/
 .installed.cfg
 *.egg
 MANIFEST
-
+.idea/
 # PyInstaller
 #  Usually these files are written by a python script from a template
 #  before PyInstaller builds the exe, so as to inject date/other infos into it.

Bot.py

@@ -1,24 +1,25 @@
+from aiogram.utils import executor
 from loguru import logger
-import telebot
+from aiogram import Bot, Dispatcher, types
-from telebot import types
 from functions import get_audio_messages_func
 
-bot = telebot.TeleBot("5941118321:AAG0g0keLrlnuH_9U9X6ehpFFAdOX38qeXI")
+bot = Bot(token="5941118321:AAG0g0keLrlnuH_9U9X6ehpFFAdOX38qeXI")
+dp = Dispatcher(bot)
 logger.add("logs/logging_log.log", level="INFO")
 
 
-@bot.message_handler(commands=['start'])
+@dp.message_handler(commands=['start'])
 def start(message: types.Message):
     name = message.chat.first_name if message.chat.first_name else 'No_name'
     logger.info(f"Группа {name} (ID: {message.chat.id}) запустил бота")
     bot.send_message(message.chat.id, 'Привет! Отправляй голосовое, я расшифрую!')
 
 
-@bot.message_handler(content_types=['voice', 'video_note', 'video'])
+@dp.message_handler(content_types=['voice', 'video_note', 'video'])
-def get_audio_messages(message: types.Message):
+async def get_audio_messages(message: types.Message):
-    get_audio_messages_func.get_audio_messages_func(message, logger, bot)
+    await get_audio_messages_func.get_audio_messages_func(message, logger, bot)
 
 
 def start_work_bot():
     logger.info("Бот запустился")
-    bot.polling(none_stop=True, timeout=123)
+    executor.start_polling(dp, skip_updates=False)

functions/convert.py

@@ -12,27 +12,39 @@ class Converter:
         self.message = message
 
     def audio_to_text(self) -> str:
-        import mutagen
+        try:
-        f = mutagen.File(self.path_to_file)
+            try:
-        if float(f.info.length) >= 30.0:
+                import mutagen
-            self.service = "google"
+                f = mutagen.File(self.path_to_file)
-            return self.google()
+                if float(f.info.length) >= 30.0:
-        else:
+                    self.service = "google"
-            self.service = "yandex"
+                    return self.google()
-            return self.yandex()
+                else:
+                    self.service = "yandex"
+                    return self.yandex()
+            except:
+                return self.google()
+        except sr.RequestError:
+            return "Толи файл слишком большой, толи с ботом что-то случилось"
 
     def google(self) -> str:
         subprocess.run(['ffmpeg', '-v', 'quiet', '-i', self.path_to_file, self.path_to_file.replace(".ogg", ".wav")])
-        os.remove(self.path_to_file)
+        try:
-        wav_file = self.path_to_file.replace(".ogg", ".wav")
+            os.remove(self.path_to_file)
-        r = sr.Recognizer()
+            wav_file = self.path_to_file.replace(".ogg", ".wav")
-        with sr.AudioFile(wav_file) as source:
+            r = sr.Recognizer()
-            audio = r.record(source)
+            with sr.AudioFile(wav_file) as source:
-            r.adjust_for_ambient_noise(source)
+                audio = r.record(source)
-        response = "google\n\n" + r.recognize_google(audio, language=self.language)
+                r.adjust_for_ambient_noise(source)
-        loging(self.message, "google", wav_file, response)
+            try:
-        os.remove(wav_file)
+                response = "google\n\n" + r.recognize_google(audio, language=self.language)
-        return response
+                loging(self.message, "google", wav_file, response)
+                os.remove(wav_file)
+                return response
+            except sr.UnknownValueError:
+                return "google\n\n Слова не распознаны, вы что вообще мне впихнуть пытаетесь"
+        except FileNotFoundError:
+            return "Толи файл слишком большой, толи с ботом что-то случилось"
 
     def yandex(self) -> str:
         from speechkit import ShortAudioRecognition, Session

functions/get_audio_messages_func.py

@@ -1,8 +1,17 @@
+import os
 from functions.convert import Converter
-from telebot import types
+from aiogram import types
 
 
-def get_audio_messages_func(message: types.Message, logger, bot):
+async def get_audio_messages_func(message: types.Message, logger, bot) -> None:
+    """
+    Функция выкачивает файл, потом конвертирует его в текст, потом логирует
+
+    :param message: types.Message - объект сообщения
+    :param logger: - объект логера
+    :param bot: - объект бота
+    :return: None
+    """
     # достаю file_id из разнообразных полей
     if message.content_type in ['voice']:
         file_id = message.voice.file_id
@@ -10,17 +19,16 @@ def get_audio_messages_func(message: types.Message, logger, bot):
         file_id = message.video.file_id
     else:
         file_id = message.video_note.file_id
-    # создаю file_name
+    file_name = "config/" + str(message.message_id) + '.ogg'  # создаю file_name
-    file_name = "config/" + str(message.message_id) + '.ogg'
+    name = message.chat.first_name if message.chat.first_name else 'No_name'  # достаю имя человека
-    # достаю имя человека
+    with open(file_name, 'wb') as new_file:  # открываю файл для записи
-    name = message.chat.first_name if message.chat.first_name else 'No_name'
+        file = await bot.get_file(file_id)  # ссылка для скачивая файла
-    # записываю в файл
+        file_dow = await bot.download_file(file.file_path)  # скачанный файл
-    with open(file_name, 'wb') as new_file:
+        new_file.write(file_dow.getvalue())  # записываю
-        new_file.write(bot.download_file(bot.get_file(file_id).file_path))
+    converter = Converter(file_name, message)  # запуск конвертора
-    # запуск конвертора
+    message_text = converter.audio_to_text()  # получаю текст из конвертора
-    converter = Converter(file_name, message)
+    logger.info(f"Чат {name} (ID: {message.chat.id}) обработал файл {file_name}, "
-    message_text = converter.audio_to_text()
+                f"сервисом {converter.service}")  # записываю всё в logger
-    # записываю всё в logger
+    await bot.send_message(message.chat.id, message_text, reply_to_message_id=message.message_id)  # отправляю сообщение
-    logger.info(f"Чат {name} (ID: {message.chat.id}) обработал файл {file_name}, сервисом {converter.service}")
+    os.removedirs("config")  # удаляю папку с файлами
-    # отправляю сообщение
+    os.mkdir("config")  # создаю папку обратно
-    bot.send_message(message.chat.id, message_text, reply_to_message_id=message.message_id)

functions/loging.py

@@ -1,2 +1,25 @@
-def loging(message, service, file_path, text):
+import os
-    print(message)
+import shutil
+from aiogram import types
+
+
+def loging(message: types.Message, service: str, file_path: str, text: str) -> None:
+    """
+    Функция логирует каждое сообщение
+
+    :param message: types.Message - объект сообщения
+    :param service: str - название сервиса для генерации
+    :param file_path: str - путь до файла
+    :param text: str - текст который был разобран
+    :return: None
+    """
+    if not os.path.isdir(f"logs/{message.chat.id}"):  # если папки нет
+        os.makedirs(f"logs/{message.chat.id}")  # создаю папку
+    file_name = file_path.split("/")[-1].split(".")  # достаю имя файла
+    try:  # защита
+        os.makedirs(f"logs/{message.chat.id}/{file_name[0]}")  # создание папки
+    except FileExistsError:
+        pass
+    shutil.copy(file_path, f"logs/{message.chat.id}/{file_name[0]}/audio.{file_name[1]}")  # копирование файла
+    with open(f'logs/{message.chat.id}/{file_name[0]}/{service}-text.txt', 'w+') as the_file:  # открываю файл в запись
+        the_file.write(text)  # записываю