self-hosted/LibreTranslate
1
0
Fork 0
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2025-11-29 22:08:02 +02:00
Code Issues Releases Activity

Add fingerprinting mechanism

Browse Source
This commit is contained in:
Piero Toffanin
2025-04-18 12:21:16 -04:00
parent da0890d60f
commit f2268fe4d9
8 changed files with 65 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
libretranslate/storage.py
View File

@@ -1,4 +1,5 @@
import redis
import time
storage = None
def get_storage():
@@ -18,7 +19,7 @@ class Storage:
def get_int(self, key):
raise Exception("not implemented")
def set_str(self, key, value):
def set_str(self, key, value, ex=None):
raise Exception("not implemented")
def get_str(self, key):
raise Exception("not implemented")
@@ -56,11 +57,22 @@ class MemoryStorage(Storage):
def get_int(self, key):
return int(self.store.get(key, 0))
def set_str(self, key, value):
self.store[key] = value
def set_str(self, key, value, ex=None):
self.store[key] = {
'value': value,
'ex': time.time() + ex
}
def get_str(self, key):
return str(self.store.get(key, ""))
d = self.store.get(key, {'value': '', 'ex': None})
if d['ex'] is None:
return d['value']
else:
if d['ex'] <= time.time():
del self.store[key]
return ''
else:
return d['value']
def set_hash_int(self, ns, key, value):
if ns not in self.store:
@@ -123,8 +135,8 @@ class RedisStorage(Storage):
else:
return v
def set_str(self, key, value):
self.conn.set(key, value)
def set_str(self, key, value, ex=None):
self.conn.set(key, value, ex=ex)
def get_str(self, key):
v = self.conn.get(key)