LibreTranslate/libretranslate/secret.py

37 lines
951 B
Python
Raw Normal View History

2024-07-16 19:58:20 +00:00
import base64
2023-03-10 03:00:27 +00:00
import random
import string
from libretranslate.storage import get_storage
2023-07-09 10:29:11 +00:00
2023-03-10 03:00:27 +00:00
def generate_secret():
return ''.join(random.choices(string.ascii_uppercase + string.digits, k=7))
def rotate_secrets():
s = get_storage()
secret_1 = s.get_str("secret_1")
s.set_str("secret_0", secret_1)
s.set_str("secret_1", generate_secret())
2023-03-10 03:00:27 +00:00
def secret_match(secret):
s = get_storage()
return secret == s.get_str("secret_0") or secret == s.get_str("secret_1")
def get_current_secret():
return get_storage().get_str("secret_1")
2024-07-16 19:58:20 +00:00
def get_current_secret_b64():
return base64.b64encode(get_current_secret().encode("utf-8")).decode("utf-8")
2023-03-10 04:07:12 +00:00
def setup(args):
if args.api_keys and args.require_api_key_secret:
2023-03-10 03:00:27 +00:00
s = get_storage()
2023-03-11 17:06:36 +00:00
if not s.exists("secret_0"):
s.set_str("secret_0", generate_secret())
if not s.exists("secret_1"):
s.set_str("secret_1", generate_secret())