feat: custom word list
This commit is contained in:
+14
-6
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user