diff options
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/aio/modules/config.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tools/aio/modules/config.py b/tools/aio/modules/config.py index 962e418..d6e7d87 100644 --- a/tools/aio/modules/config.py +++ b/tools/aio/modules/config.py @@ -1,12 +1,23 @@ import os import typing import uuid +import random +import string from rich.prompt import Prompt from .path import config_file_path def generate_uuid(): return str(uuid.uuid4()) +# generate random characters of digits and alphabets +def generate_random_string(length: int): + characters = string.ascii_letters + string.digits + random_string = ''.join(random.choice(characters) for _ in range(n)) + return random_string + +def generate_random_string_32(): + return generate_random_string(32) + class ConfigVar: def __init__(self, name: str, description: str, default_value_generator: typing.Callable[[], str] | str, /, default_value_for_ask=str | None): """Create a config var. @@ -59,6 +70,12 @@ config_var_list: list = [ "Forgejo SMTP user.", "Please input your Forgejo SMTP user."), ConfigVar("CRUPEST_FORGEJO_MAILER_PASSWD", "Forgejo SMTP password.", "Please input your Forgejo SMTP password."), + ConfigVar("CRUPEST_2FAUTH_APP_KEY", + "2FAuth App Key.", generate_random_string_32), + ConfigVar("CRUPEST_2FAUTH_MAIL_USERNAME", + "2FAuth SMTP user.", "Please input your 2FAuth SMTP user."), + ConfigVar("CRUPEST_2FAUTH_MAIL_PASSWORD", + "2FAuth SMTP password.", "Please input your 2FAuth SMTP password."), ] config_var_name_set = set([config_var.name for config_var in config_var_list]) |