LibreTranslate/libretranslate/manage.py

61 lines
1.9 KiB
Python
Raw Normal View History

import argparse
import os
from libretranslate.api_keys import Database
from libretranslate.default_values import DEFAULT_ARGUMENTS as DEFARGS
2021-05-18 03:41:02 +00:00
def manage():
2021-05-18 03:41:02 +00:00
parser = argparse.ArgumentParser(description="LibreTranslate Manage Tools")
subparsers = parser.add_subparsers(
help="", dest="command", required=True, title="Command List"
)
keys_parser = subparsers.add_parser("keys", help="Manage API keys database")
keys_parser.add_argument(
"--api-keys-db-path",
default=DEFARGS['API_KEYS_DB_PATH'],
type=str,
help="Use a specific path inside the container for the local database",
)
2021-05-18 03:41:02 +00:00
keys_subparser = keys_parser.add_subparsers(
help="", dest="sub_command", title="Command List"
)
keys_add_parser = keys_subparser.add_parser("add", help="Add API keys to database")
keys_add_parser.add_argument(
2023-04-03 15:06:46 +00:00
"req_limit", type=int, help="Request Limits (per minute)"
2021-05-18 03:41:02 +00:00
)
keys_add_parser.add_argument(
"--key", type=str, default="auto", required=False, help="API Key"
)
keys_remove_parser = keys_subparser.add_parser(
"remove", help="Remove API keys to database"
)
keys_remove_parser.add_argument("key", type=str, help="API Key")
args = parser.parse_args()
2021-05-18 03:41:02 +00:00
if args.command == "keys":
if not os.path.exists(args.api_keys_db_path):
print("No such database: %s" % args.api_keys_db_path)
exit(1)
db = Database(args.api_keys_db_path)
if args.sub_command is None:
# Print keys
keys = db.all()
if not keys:
print("There are no API keys")
else:
for item in keys:
2023-07-09 10:29:11 +00:00
print("{}: {}".format(*item))
2021-05-18 03:41:02 +00:00
elif args.sub_command == "add":
print(db.add(args.req_limit, args.key)[0])
2021-05-18 03:41:02 +00:00
elif args.sub_command == "remove":
print(db.remove(args.key))
else:
parser.print_help()
2021-05-18 03:41:02 +00:00
exit(1)