57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from pathlib import Path
|
|
import json
|
|
from typing import Dict, Optional, Any
|
|
from fastapi import Request
|
|
|
|
class I18n:
|
|
def __init__(self, locale_dir: str = "locales"):
|
|
self.locale_dir = Path(locale_dir)
|
|
self.translations: Dict[str, Dict[str, Any]] = {}
|
|
self.load_translations()
|
|
|
|
def load_translations(self):
|
|
if not self.locale_dir.exists():
|
|
self.locale_dir.mkdir()
|
|
return
|
|
|
|
for lang_file in self.locale_dir.glob("*.json"):
|
|
with open(lang_file, 'r', encoding='utf-8') as f:
|
|
self.translations[lang_file.stem.lower()] = json.load(f)
|
|
|
|
def get_language(self, request: Request) -> str:
|
|
accept_language = request.headers.get('accept-language', 'en')
|
|
primary_lang = accept_language.split(',')[0].lower()
|
|
|
|
return primary_lang if primary_lang in self.translations else 'zh-cn'
|
|
|
|
def __call__(self, request: Request) -> 'I18nHelper':
|
|
return I18nHelper(self, request)
|
|
|
|
class I18nHelper:
|
|
def __init__(self, i18n: I18n, request: Request):
|
|
self.i18n = i18n
|
|
self.request = request
|
|
self.lang = i18n.get_language(request)
|
|
|
|
def _get_nested_translation(self, data: Dict[str, Any], key_path: str) -> str:
|
|
keys = key_path.split('.')
|
|
current = data
|
|
|
|
for key in keys:
|
|
if isinstance(current, dict) and key in current:
|
|
current = current[key]
|
|
else:
|
|
return key_path # 如果找不到，返回原键
|
|
|
|
return current if isinstance(current, str) else key_path
|
|
|
|
def __call__(self, arg_key: str, **kwargs) -> str:
|
|
lang_data = self.i18n.translations.get(self.lang, {})
|
|
text = self._get_nested_translation(lang_data, arg_key)
|
|
|
|
if kwargs:
|
|
try:
|
|
return text.format(**kwargs)
|
|
except (KeyError, ValueError):
|
|
return text
|
|
return text |