feat: blacklist

cool_domain/__main__.py

@@ -1,16 +1,15 @@
 import logging
 import sys
 
+from cool_domain import PKG_NAME, PKG_VERSION
+from cool_domain.cache import Cache
+from cool_domain.constants import LANGS
+from cool_domain.factory import Factory
+from cool_domain.logs import setup_logs
+from cool_domain.params import parse_parameters
+from cool_domain.tld_list import TLDList
 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:])
@@ -33,3 +32,7 @@ def main() -> int:
             else:
                 logging.getLogger().info(factory.random_domain())
     return 0
+
+
+if __name__ == "__main__":
+    main()

cool_domain/factory.py

@@ -4,8 +4,8 @@ import shutil
 import subprocess
 import typing
 
-from .tld_list import TLDList
+from cool_domain.tld_list import TLDList
-from .word_list import WordList
+from cool_domain.word_list import WordList
 
 
 class FactoryError(Exception):

cool_domain/logs.py

@@ -2,7 +2,7 @@ import enum
 import logging
 import typing
 
-from .params import Parameters
+from cool_domain.params import Parameters
 
 
 class TermColor(enum.StrEnum):

cool_domain/params.py

@@ -5,8 +5,8 @@ import pathlib
 import tempfile
 import typing
 
-from . import PKG_NAME
+from cool_domain import PKG_NAME
-from .constants import LANGS
+from cool_domain.constants import LANGS
 
 
 @dataclasses.dataclass(frozen=True, slots=True)
@@ -19,7 +19,7 @@ class Parameters:
     dig: bool = True
     lang: str = "english"
     cache_file: str = str(pathlib.Path(tempfile.gettempdir()) / ".cool-domain-cache")
-    tld_list_uri: str = "https://publicsuffix.org/list/public_suffix_list.dat"
+    tld_list_uri: str = "https://data.iana.org/TLD/tlds-alpha-by-domain.txt"
     custom_words: str = "words.txt"
 
     @classmethod

cool_domain/tld_list.py

@@ -5,7 +5,8 @@ import typing
 import requests
 import unidecode
 
-from .cache import Cache
+from cool_domain.cache import Cache
+from cool_domain.constants import BLACKLIST
 
 
 class TLDError(Exception):
@@ -29,21 +30,18 @@ class TLDList:
         raw_lines = response.content.decode().split("\n")
         out_lines = []
         for line in raw_lines:
-            stripped_line = line.strip()
+            stripped_line = line.strip().lower()
             if (
                 len(stripped_line)
                 and not stripped_line.startswith("//")
                 and not stripped_line.startswith("*")
+                and not stripped_line.startswith("#")
+                and not stripped_line.startswith("xn--")
+                and "." not in stripped_line
                 and unidecode.unidecode(stripped_line) == stripped_line
+                and stripped_line not in BLACKLIST
             ):
-                if "." in stripped_line:
+                out_lines += [stripped_line]
-                    for other in out_lines:
-                        if stripped_line.endswith(other):
-                            break
-                    else:
-                        out_lines += [stripped_line]
-                else:
-                    out_lines += [stripped_line]
         return out_lines
 
     def load(self) -> None: