From d1deaad03f8f53a26edd95dd9cd4a827c2801118 Mon Sep 17 00:00:00 2001 From: Klemek Date: Mon, 15 Jun 2026 14:31:21 +0200 Subject: [PATCH] fix: space in words --- cool_domain/factory.py | 4 ++-- cool_domain/word_list.py | 5 ++++- 2 files changed, 6 insertions(+), 3 deletions(-) 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