fix: space in words
This commit is contained in:
@@ -14,7 +14,7 @@ class FactoryError(Exception):
|
|||||||
|
|
||||||
|
|
||||||
class Factory:
|
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:
|
def __init__(self, tld_list: TLDList, word_list: WordList, words: int) -> None:
|
||||||
self.logger: logging.Logger = logging.getLogger(self.__class__.__name__)
|
self.logger: logging.Logger = logging.getLogger(self.__class__.__name__)
|
||||||
@@ -71,7 +71,7 @@ class Factory:
|
|||||||
tld_escaped = tld.replace(".", "")
|
tld_escaped = tld.replace(".", "")
|
||||||
if word.endswith(tld_escaped) and len(word) > len(tld_escaped):
|
if word.endswith(tld_escaped) and len(word) > len(tld_escaped):
|
||||||
domain = word[: -len(tld_escaped)] + "." + tld
|
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)
|
self.__found.add(domain)
|
||||||
return domain
|
return domain
|
||||||
return None
|
return None
|
||||||
|
|||||||
@@ -22,7 +22,10 @@ class WordList:
|
|||||||
self.logger.info("[%s] Loaded %d words", self.__name, len(self.words))
|
self.logger.info("[%s] Loaded %d words", self.__name, len(self.words))
|
||||||
|
|
||||||
def __escape_word(self, word: str) -> str:
|
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:
|
def close(self) -> None:
|
||||||
pass
|
pass
|
||||||
|
|||||||
Reference in New Issue
Block a user