Refactor project structure and add new features

app/infrastructure/storage/cache.py

@@ -0,0 +1,128 @@
+"""
+Модуль cache.py содержит функции для кэширования данных.
+"""
+
+import time
+from typing import Dict, Any, Optional, Callable
+from functools import wraps
+import logging
+
+logger = logging.getLogger('app.cache')
+
+
+class SimpleCache:
+    """
+    Простой кэш на основе словаря с поддержкой TTL (Time To Live).
+    
+    Attributes:
+        cache (Dict): Словарь для хранения кэшированных данных.
+        ttl (int): Время жизни кэша в секундах.
+    """
+    
+    def __init__(self, ttl: int = 300):
+        """
+        Инициализация кэша.
+        
+        Args:
+            ttl: Время жизни кэша в секундах (по умолчанию 5 минут).
+        """
+        self.cache: Dict[str, Dict[str, Any]] = {}
+        self.ttl = ttl
+    
+    def get(self, key: str) -> Optional[Any]:
+        """
+        Получение значения из кэша.
+        
+        Args:
+            key: Ключ для получения значения.
+            
+        Returns:
+            Кэшированное значение или None, если ключ не найден или срок действия истек.
+        """
+        if key in self.cache:
+            item = self.cache[key]
+            if time.time() - item["timestamp"] < self.ttl:
+                logger.debug(f"Кэш hit для ключа: {key}")
+                return item["value"]
+            else:
+                # Удаление просроченного элемента
+                del self.cache[key]
+                logger.debug(f"Кэш expired для ключа: {key}")
+        
+        logger.debug(f"Кэш miss для ключа: {key}")
+        return None
+    
+    def set(self, key: str, value: Any) -> None:
+        """
+        Установка значения в кэш.
+        
+        Args:
+            key: Ключ для хранения значения.
+            value: Значение для кэширования.
+        """
+        self.cache[key] = {
+            "value": value,
+            "timestamp": time.time()
+        }
+        logger.debug(f"Значение кэшировано для ключа: {key}")
+    
+    def clear(self) -> None:
+        """
+        Очистка кэша.
+        """
+        self.cache.clear()
+        logger.debug("Кэш очищен")
+    
+    def delete(self, key: str) -> bool:
+        """
+        Удаление значения из кэша.
+        
+        Args:
+            key: Ключ для удаления.
+            
+        Returns:
+            True, если ключ был удален, иначе False.
+        """
+        if key in self.cache:
+            del self.cache[key]
+            logger.debug(f"Значение удалено из кэша для ключа: {key}")
+            return True
+        return False
+
+
+# Глобальные экземпляры кэша
+model_cache = SimpleCache(ttl=3600)  # Кэш для метаданных модели (1 час)
+config_cache = SimpleCache(ttl=300)   # Кэш для конфигурации (5 минут)
+
+
+def cache_result(cache_instance: SimpleCache, key_prefix: str = ""):
+    """
+    Декоратор для кэширования результатов функции.
+    
+    Args:
+        cache_instance: Экземпляр кэша.
+        key_prefix: Префикс для ключа кэша.
+        
+    Returns:
+        Декорированная функция.
+    """
+    def decorator(func: Callable) -> Callable:
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            # Генерация ключа кэша на основе имени функции и аргументов
+            cache_key = f"{key_prefix}{func.__name__}_{str(args)}_{str(kwargs)}"
+            
+            # Попытка получить результат из кэша
+            cached_result = cache_instance.get(cache_key)
+            if cached_result is not None:
+                return cached_result
+            
+            # Если результат не в кэше, вызываем функцию
+            result = func(*args, **kwargs)
+            
+            # Сохраняем результат в кэш
+            cache_instance.set(cache_key, result)
+            
+            return result
+        return wrapper
+    return decorator

app/infrastructure/storage/file_manager.py

@@ -0,0 +1,118 @@
+"""
+Модуль file_manager.py содержит классы для централизованного управления временными файлами.
+Предоставляет унифицированный интерфейс для создания, отслеживания и очистки временных файлов.
+"""
+
+import os
+import uuid
+import tempfile
+import contextlib
+from typing import List, Tuple, Optional, Generator
+import logging
+
+logger = logging.getLogger('app.file_manager')
+
+
+class TempFileManager:
+    """
+    Класс для централизованного управления временными файлами.
+    
+    Предоставляет методы для создания временных файлов и их последующей очистки.
+    Использует контекстные менеджеры для автоматической очистки ресурсов.
+    """
+    
+    def __init__(self):
+        """
+        Инициализация менеджера временных файлов.
+        """
+        self.temp_files = []
+        self.temp_dirs = []
+    
+    def create_temp_file(self, suffix: str = ".wav") -> Tuple[str, str]:
+        """
+        Создает временный файл с уникальным именем.
+        
+        Args:
+            suffix: Расширение временного файла.
+            
+        Returns:
+            Кортеж (путь к файлу, путь к временной директории).
+        """
+        temp_dir = tempfile.mkdtemp()
+        temp_file = os.path.join(temp_dir, f"{uuid.uuid4()}{suffix}")
+        
+        self.temp_files.append(temp_file)
+        self.temp_dirs.append(temp_dir)
+        
+        logger.debug(f"Создан временный файл: {temp_file}")
+        return temp_file, temp_dir
+    
+    def cleanup_temp_files(self, file_paths: Optional[List[str]] = None) -> None:
+        """
+        Очищает временные файлы и директории.
+        
+        Args:
+            file_paths: Список путей к файлам для очистки. Если None, очищает все отслеживаемые файлы.
+        """
+        paths_to_clean = file_paths if file_paths is not None else self.temp_files
+        
+        for path in paths_to_clean:
+            try:
+                if os.path.exists(path):
+                    os.remove(path)
+                    logger.debug(f"Удален временный файл: {path}")
+                    
+                    # Попытка удалить директорию, если она пуста
+                    temp_dir = os.path.dirname(path)
+                    if os.path.exists(temp_dir) and not os.listdir(temp_dir):
+                        os.rmdir(temp_dir)
+                        logger.debug(f"Удалена временная директория: {temp_dir}")
+                        
+                        # Удаление из списка отслеживаемых директорий
+                        if temp_dir in self.temp_dirs:
+                            self.temp_dirs.remove(temp_dir)
+            except Exception as e:
+                logger.warning(f"Не удалось очистить временный файл {path}: {e}")
+        
+        # Удаление файлов из списка отслеживаемых
+        if file_paths is None:
+            self.temp_files.clear()
+        else:
+            for path in file_paths:
+                if path in self.temp_files:
+                    self.temp_files.remove(path)
+    
+    @contextlib.contextmanager
+    def temp_file(self, suffix: str = ".wav") -> Generator[str, None, None]:
+        """
+        Контекстный менеджер для создания и автоматической очистки временного файла.
+        
+        Args:
+            suffix: Расширение временного файла.
+            
+        Yields:
+            Путь к временному файлу.
+        """
+        temp_file, _ = self.create_temp_file(suffix)
+        try:
+            yield temp_file
+        finally:
+            self.cleanup_temp_files([temp_file])
+    
+    def cleanup_all(self) -> None:
+        """
+        Очищает все отслеживаемые временные файлы и директории.
+        """
+        self.cleanup_temp_files()
+        for temp_dir in self.temp_dirs:
+            try:
+                if os.path.exists(temp_dir) and not os.listdir(temp_dir):
+                    os.rmdir(temp_dir)
+                    logger.debug(f"Удалена временная директория: {temp_dir}")
+            except Exception as e:
+                logger.warning(f"Не удалось очистить временную директорию {temp_dir}: {e}")
+        self.temp_dirs.clear()
+
+
+# Глобальный экземпляр менеджера временных файлов
+temp_file_manager = TempFileManager()