Compare commits
5
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
15a937e39c | ||
|
|
d1deaad03f | ||
|
|
4df2bab348 | ||
|
|
d5b9018ef2 | ||
|
|
b387fec2b3 |
@@ -23,14 +23,15 @@ uvx --from git+https://git.klemek.fr/klemek/cool-domain cool-domain
|
||||
## Usage
|
||||
|
||||
```txt
|
||||
usage: cool-domain [-h] [--count DOMAIN_COUNT] [--lang {english,french}] [--dig | --no-dig] [--debug | --no-debug] [--quiet | --no-quiet]
|
||||
[--cache | --no-cache] [--cache-file CACHE_FILE] [--tld-list-uri TLD_LIST_URI]
|
||||
usage: cool-domain [-h] [--count DOMAIN_COUNT] [--lang {english,french}] [--words WORD_COUNT] [--dig | --no-dig] [--debug | --no-debug]
|
||||
[--quiet | --no-quiet] [--cache | --no-cache] [--cache-file CACHE_FILE] [--tld-list-uri TLD_LIST_URI]
|
||||
|
||||
options:
|
||||
-h, --help show this help message and exit
|
||||
--count DOMAIN_COUNT number of domains to generate (default: 1)
|
||||
--lang {english,french}
|
||||
word list lang (default: english)
|
||||
--words WORD_COUNT number of words per domain (default: 1)
|
||||
--dig, --no-dig check availability with dig (default: true)
|
||||
--debug, --no-debug
|
||||
--quiet, --no-quiet
|
||||
|
||||
@@ -19,7 +19,7 @@ def main() -> int:
|
||||
TLDList(cache=cache, remote_uri=params.tld_list_uri) as tld_list,
|
||||
LANGS[params.lang] as word_list,
|
||||
):
|
||||
factory = Factory(tld_list=tld_list, word_list=word_list)
|
||||
factory = Factory(tld_list=tld_list, word_list=word_list, words=params.words)
|
||||
logging.getLogger().info(
|
||||
"Searching %s domain%s", params.count, "s" if params.count > 1 else ""
|
||||
)
|
||||
|
||||
+10
-6
@@ -14,12 +14,13 @@ class FactoryError(Exception):
|
||||
|
||||
|
||||
class Factory:
|
||||
__slots__ = ["__found", "__tld_list", "__word_list", "logger", "logger"]
|
||||
__slots__ = ["__found", "__tld_list", "__word_list", "__words", "logger"]
|
||||
|
||||
def __init__(self, tld_list: TLDList, word_list: WordList) -> None:
|
||||
def __init__(self, tld_list: TLDList, word_list: WordList, words: int) -> None:
|
||||
self.logger: logging.Logger = logging.getLogger(self.__class__.__name__)
|
||||
self.__tld_list = tld_list
|
||||
self.__word_list = word_list
|
||||
self.__words = words
|
||||
self.__found: set[str] = set()
|
||||
|
||||
def __iter__(self) -> typing.Self:
|
||||
@@ -41,10 +42,10 @@ class Factory:
|
||||
msg = "dig is not installed"
|
||||
raise FactoryError(msg)
|
||||
try:
|
||||
output = subprocess.check_output([dig_bin, domain, "+short"]) # noqa: S603
|
||||
output = subprocess.check_output([dig_bin, domain, "NS", "+short"]) # noqa: S603
|
||||
return len(output) == 0
|
||||
except subprocess.CalledProcessError:
|
||||
return True
|
||||
return False
|
||||
|
||||
def random_domain(self) -> str:
|
||||
k = 10000
|
||||
@@ -65,12 +66,15 @@ class Factory:
|
||||
return domain
|
||||
|
||||
def __random_domain(self) -> str | None:
|
||||
word = random.choice(self.__word_list.words) # noqa: S311
|
||||
word = self.__random_word()
|
||||
for tld in self.__tld_list_shuffled:
|
||||
tld_escaped = tld.replace(".", "")
|
||||
if word.endswith(tld_escaped) and len(word) > len(tld_escaped):
|
||||
domain = word[: -len(tld_escaped)] + "." + tld
|
||||
if domain not in self.__found:
|
||||
if "-." not in domain and domain not in self.__found:
|
||||
self.__found.add(domain)
|
||||
return domain
|
||||
return None
|
||||
|
||||
def __random_word(self) -> str:
|
||||
return "-".join(random.sample(self.__word_list.words, self.__words))
|
||||
|
||||
@@ -15,6 +15,7 @@ class Parameters:
|
||||
quiet: bool = False
|
||||
cache: bool = True
|
||||
count: int = 1
|
||||
words: int = 1
|
||||
dig: bool = True
|
||||
lang: str = "english"
|
||||
cache_file: str = str(pathlib.Path(tempfile.gettempdir()) / ".cool-domain-cache")
|
||||
@@ -104,6 +105,13 @@ def parse_parameters(args: typing.Sequence[str]) -> Parameters:
|
||||
help="word list lang (default: english)",
|
||||
choices=LANGS.keys(),
|
||||
)
|
||||
__add_arg_int(
|
||||
parser,
|
||||
"--words",
|
||||
env_var="WORD_COUNT",
|
||||
default=default_values.words,
|
||||
help_txt="number of words per domain",
|
||||
)
|
||||
__add_arg_bool(
|
||||
parser,
|
||||
"--dig",
|
||||
|
||||
@@ -22,7 +22,10 @@ class WordList:
|
||||
self.logger.info("[%s] Loaded %d words", self.__name, len(self.words))
|
||||
|
||||
def __escape_word(self, word: str) -> str:
|
||||
return unidecode.unidecode(word.strip()).lower()
|
||||
word = unidecode.unidecode(word.strip()).lower()
|
||||
if " " in word:
|
||||
return ""
|
||||
return word
|
||||
|
||||
def close(self) -> None:
|
||||
pass
|
||||
|
||||
+1
-1
@@ -1,6 +1,6 @@
|
||||
[project]
|
||||
name = "cool-domain"
|
||||
version = "1.0.0"
|
||||
version = "1.1.1"
|
||||
description = "Add your description here"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.14"
|
||||
|
||||
Reference in New Issue
Block a user