gonets/SRV/utils/i18n.py

59 lines
No EOL
2.8 KiB
Python

from fastapi import Request, Depends
from babel.support import Translations
import logging
logger = logging.getLogger(__name__)
def get_translations(domain: str = "api_v1") -> callable:
"""
Создаёт зависимость для загрузки переводов на основе заголовка Accept-Language
или параметра запроса 'locale'. По умолчанию используется русский язык.
Args:
domain (str): Домен переводов (например, 'api_v1', 'services'). По умолчанию 'api_v1'.
Returns:
callable: Функция, возвращающая объект Translations для указанной локали и домена.
"""
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
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)