diff options
| author | Yuval Adam <_@yuv.al> | 2025-06-15 13:44:16 +0200 |
|---|---|---|
| committer | Yuval Adam <_@yuv.al> | 2025-06-15 13:44:16 +0200 |
| commit | 2e0755cc68e14bda95abe9472754ae5f4e760a04 (patch) | |
| tree | 4b825dc642cb6eb9a060e54bf8d69288fbee4904 /cli/scan.py | |
| parent | cd812332b9000e58665585e9c3fadc30470a9a6d (diff) | |
Remove all old files before migration
Diffstat (limited to 'cli/scan.py')
| -rw-r--r-- | cli/scan.py | 140 |
1 files changed, 0 insertions, 140 deletions
diff --git a/cli/scan.py b/cli/scan.py deleted file mode 100644 index 318e68b..0000000 --- a/cli/scan.py +++ /dev/null @@ -1,140 +0,0 @@ -import aiohttp -import asyncio -import json -import requests - -from pathlib import Path -from os import makedirs, listdir - -from .base import cli - -ROOT_PATH = Path(__file__).parents[1] - -DATA_DIR = ROOT_PATH / "data" - -TLDS_URL = "https://data.iana.org/TLD/tlds-alpha-by-domain.txt" - -NAMES_ENDPOINT = "https://nameberry.com/nameberry/api/v1/search" - - -class NameScanner: - def fetch_tlds(self): - with open(DATA_DIR / "tlds.txt", "w") as f: - print("Fetching TLDs...") - res = requests.get(TLDS_URL) - tlds = [ - tld.lower() for tld in res.text.strip().split("\n")[1:] if len(tld) < 4 - ] - f.write("\n".join(tlds).strip()) - _TLDS = set(tlds) - - def fetch_homepage(self, domain): - print(f"Fetching {domain}...", end="") - res = requests.get(f"http://{domain}", timeout=5) - if res.ok: - print(f"Found {domain}!") - with open(DATA_DIR / "homepages" / f"{domain}.html", "w") as f: - f.write(res.text) - else: - print("x") - - def fetch_names(self, suffix, count=5000): - res = requests.post( - NAMES_ENDPOINT, - json={ - "starts_with": "", - "ends_with": suffix, - "contains": "", - "syllables": "", - "origin_id": "", - "derivation": "", - "page": 1, - "per_page": count, - }, - ) - if res.ok: - j = res.json() - print(f"found {j['advanced_name_count']}") - makedirs(DATA_DIR / "names", exist_ok=True) - with open(DATA_DIR / "names" / f"{suffix}.json", "w") as f: - f.write(res.text) - - def fetch_all(self): - with open(DATA_DIR / "tlds.txt", "r") as f: - for line in f: - d = line.strip() - print(f"Fetching {d}...", end="") - self.fetch_names(d) - - def consolidate_names(self): - res = set() - for fn in listdir(DATA_DIR / "names"): - with open(DATA_DIR / "names" / fn, "r") as f: - j = json.load(f) - names = j["advanced"] - if names: - suffix = fn.split(".")[0] - for name in names: - n = name["name"].lower().split(suffix)[0] - if n: - res.add(n) - with open(DATA_DIR / "allnames.txt", "w") as out: - for n in res: - out.write(n + f".{suffix}\n") - - async def get_homepage(self, session, domain): - try: - url = f"http://{domain}" - async with session.get(url) as res: - res = await res.text() - if res: - print(f"Got {len(res)} bytes from {domain}") - with open(DATA_DIR / "homepages" / f"{domain}.html", "w") as out: - out.write(res) - return len(res) - except Exception: - return 0 - - async def fetch_homepages(self, N): - with open(DATA_DIR / "allnames.txt", "r") as f: - names = [x.strip() for x in f.readlines()][N * 1000 : 1000 * (N + 1)] - - timeout = aiohttp.ClientTimeout(total=30) - async with aiohttp.ClientSession(timeout=timeout) as session: - tasks = [] - for name in names: - tasks.append(asyncio.ensure_future(self.get_homepage(session, name))) - - _resps = await asyncio.gather(*tasks) - - def find_homepages(self): - res = {} - for fn in listdir(DATA_DIR / "homepages"): - with open(DATA_DIR / "homepages" / fn, "r") as f: - domain = fn[:-5] - name = domain.replace(".", "") - text = f.read() - nip = name in text - github = "github" in text - li = "linkedin" in text - sale = "for sale" in text or "register" in text or "parking" in text - res["domain"] = { - "name": name, - "name_in_page": nip, - "github": github, - "linkedin": li, - } - ge = "✅" if github else " " - le = "✅" if li else " " - ne = "✅" if nip else " " - se = "❌" if sale else " " - print(f"{ne}{ge}{le}{se}{name:20}") - - -@cli.command() -def scan(): - ns = NameScanner() - # for i in range(0, 26): - # asyncio.run(ns.fetch_homepages(i)) - # ns.consolidate_names() - ns.find_homepages() |
