Переключение языка по аргументу Accept-Language в запросе

SRV/utils/i18n.py

@@ -1,36 +1,59 @@
-from fastapi import Request
+from fastapi import Request, Depends
 from babel.support import Translations
 import logging
 
 logger = logging.getLogger(__name__)
 
 
-def get_translations(request: Request) -> Translations:
+def get_translations(domain: str = "api_v1") -> callable:
     """
-    Загружает переводы для текущего запроса на основе заголовка Accept-Language
+    Создаёт зависимость для загрузки переводов на основе заголовка Accept-Language
     или параметра запроса 'locale'. По умолчанию используется русский язык.
 
     Args:
-        request (Request): Объект запроса FastAPI.
+        domain (str): Домен переводов (например, 'api_v1', 'services'). По умолчанию 'api_v1'.
 
     Returns:
-        Translations: Объект переводов для текущей локали.
+        callable: Функция, возвращающая объект Translations для указанной локали и домена.
     """
-    # Проверяем параметр запроса 'locale', если он есть
-    locale = request.query_params.get("locale")
-    if not locale:
-        # Если параметр отсутствует, используем заголовок Accept-Language
-        locale = request.headers.get("Accept-Language", "ru").split(",")[0]
 
-    try:
-        translations = Translations.load("locales", [locale], domain="messages")
-        logger.debug(f"Загружены переводы для локали: {locale}")
-        return translations
-    except Exception as e:
-        logger.error(f"Не удалось загрузить переводы для локали {locale}: {str(e)}")
-        # Возвращаем переводы по умолчанию (русский)
-        return Translations.load("locales", ["ru"], domain="messages")
+    def translations_dependency(request: Request) -> Translations:
+        # Проверяем параметр запроса 'locale', если он есть
+        locale = request.query_params.get("locale")
+        if not locale:
+            # Если параметр отсутствует, используем заголовок Accept-Language
+            locale = request.headers.get("Accept-Language", "ru").split(",")[0].split("-")[0]
+
+        try:
+            translations = Translations.load("locales", [locale], domain)
+            if not translations._catalog:
+                logger.warning(f"Translations for locale {locale} and domain {domain} not found, falling back to ru")
+                translations = Translations.load("locales", ["ru"], domain)
+            logger.debug(f"Loaded translations for locale {locale} and domain {domain}")
+            return translations
+        except Exception as e:
+            logger.error(f"Failed to load translations for locale {locale} and domain {domain}: {str(e)}")
+            # Fallback на русский
+            return Translations.load("locales", ["ru"], domain)
+
+    return translations_dependency
 
 
-# Определяем функцию `_` для удобства использования в коде
-_ = lambda s: get_translations(Request).gettext(s)
+def _(msgid: str, domain: str = "api_v1", request: Request = None) -> str:
+    """
+    Переводит строку msgid для указанного домена и текущей локали.
+
+    Args:
+        msgid (str): Строка для перевода.
+        domain (str): Домен переводов (например, 'api_v1', 'services'). По умолчанию 'api_v1'.
+        request (Request, optional): Объект запроса FastAPI для получения локали.
+
+    Returns:
+        str: Переведенная строка.
+    """
+    if request is None:
+        # Язык, который будет отображаться по умолчанию
+        translations = Translations.load("locales", ["ru"], domain)
+    else:
+        translations = get_translations(domain)(request)
+    return translations.gettext(msgid)