36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import logging
|
|
import sys
|
|
|
|
from cool_domain.word_list import WordList
|
|
|
|
from . import PKG_NAME, PKG_VERSION
|
|
from .cache import Cache
|
|
from .constants import LANGS
|
|
from .factory import Factory
|
|
from .logs import setup_logs
|
|
from .params import parse_parameters
|
|
from .tld_list import TLDList
|
|
|
|
|
|
def main() -> int:
|
|
params = parse_parameters(sys.argv[1:])
|
|
setup_logs(params)
|
|
logging.getLogger().info("%s %s", PKG_NAME, PKG_VERSION)
|
|
langs = LANGS
|
|
langs["custom"] = WordList("custom", params.custom_words)
|
|
with (
|
|
Cache(filepath=params.cache_file, no_load=params.cache) as cache,
|
|
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, words=params.words)
|
|
logging.getLogger().info(
|
|
"Searching %s domain%s", params.count, "s" if params.count > 1 else ""
|
|
)
|
|
for _ in range(params.count):
|
|
if params.dig:
|
|
logging.getLogger().info(factory.random_available_domain())
|
|
else:
|
|
logging.getLogger().info(factory.random_domain())
|
|
return 0
|