diff --git a/cool_domain/factory.py b/cool_domain/factory.py index 4b6ba7e..7ab5938 100644 --- a/cool_domain/factory.py +++ b/cool_domain/factory.py @@ -14,7 +14,7 @@ class FactoryError(Exception): class Factory: - __slots__ = ["__found", "__tld_list", "__word_list", "logger", "__words"] + __slots__ = ["__found", "__tld_list", "__word_list", "__words", "logger"] def __init__(self, tld_list: TLDList, word_list: WordList, words: int) -> None: self.logger: logging.Logger = logging.getLogger(self.__class__.__name__) @@ -71,7 +71,7 @@ class Factory: 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 diff --git a/cool_domain/word_list.py b/cool_domain/word_list.py index 249bf5f..58c8e87 100644 --- a/cool_domain/word_list.py +++ b/cool_domain/word_list.py @@ -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