41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import json
|
||
import os
|
||
import hashlib
|
||
|
||
CACHE_FILE = "product_translation_cache.json"
|
||
MAX_CACHE_ENTRIES = 200_000 # максимум записів у кеші
|
||
|
||
class TranslationCache:
|
||
def __init__(self):
|
||
self.cache = {}
|
||
self.load_cache()
|
||
|
||
def get_hash(self, text: str) -> str:
|
||
return hashlib.md5(text.strip().encode("utf-8")).hexdigest()
|
||
|
||
def load_cache(self):
|
||
if os.path.exists(CACHE_FILE):
|
||
try:
|
||
with open(CACHE_FILE, "r", encoding="utf-8") as f:
|
||
self.cache = json.load(f)
|
||
except Exception as e:
|
||
print(f"[CACHE] Помилка завантаження кешу: {e}")
|
||
|
||
def save_cache(self):
|
||
# якщо занадто великий кеш — залишаємо тільки останні MAX_CACHE_ENTRIES
|
||
if len(self.cache) > MAX_CACHE_ENTRIES:
|
||
# зберігаємо найновіші — останні додані (у Python 3.7+ dict зберігає порядок)
|
||
trimmed = dict(list(self.cache.items())[-MAX_CACHE_ENTRIES:])
|
||
self.cache = trimmed
|
||
|
||
with open(CACHE_FILE, "w", encoding="utf-8") as f:
|
||
json.dump(self.cache, f, ensure_ascii=False, indent=2)
|
||
|
||
def get(self, text: str) -> str | None:
|
||
key = self.get_hash(text)
|
||
return self.cache.get(key)
|
||
|
||
def add(self, text: str, translated: str):
|
||
key = self.get_hash(text)
|
||
self.cache[key] = translated
|