feat: custom word list

This commit is contained in:
2026-06-15 15:09:27 +02:00
parent f91384173b
commit 85a1c47c8c
6 changed files with 31 additions and 15 deletions
+14 -6
View File
@@ -20,6 +20,7 @@ class Parameters:
lang: str = "english"
cache_file: str = str(pathlib.Path(tempfile.gettempdir()) / ".cool-domain-cache")
tld_list_uri: str = "https://publicsuffix.org/list/public_suffix_list.dat"
custom_words: str = "words.txt"
@classmethod
def from_namespace(cls, args: argparse.Namespace) -> "Parameters":
@@ -98,12 +99,6 @@ def parse_parameters(args: typing.Sequence[str]) -> Parameters:
default=default_values.count,
help_txt="number of domains to generate",
)
parser.add_argument(
"--lang",
default=default_values.lang,
help="word list lang (default: english)",
choices=LANGS.keys(),
)
__add_arg_int(
parser,
"--words",
@@ -111,6 +106,19 @@ def parse_parameters(args: typing.Sequence[str]) -> Parameters:
default=default_values.words,
help_txt="number of words per domain",
)
parser.add_argument(
"--lang",
default=default_values.lang,
help="word list lang (default: english)",
choices=[*list(LANGS.keys()), "custom"],
)
__add_arg_str(
parser,
"--custom-words",
env_var="CUSTOM_WORDS",
default=default_values.custom_words,
help_txt="custom word list location",
)
__add_arg_bool(
parser,
"--dig",