import atexit import random import string from libretranslate.storage import get_storage 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()) 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") def setup(args): if args.api_keys and args.require_api_key_secret: s = get_storage() s.set_str("secret_0", generate_secret()) s.set_str("secret_1", generate_secret())