new: [providence] first skeleton

Signed-off-by: Alexandre Dulaunoy <a@foo.be>
This commit is contained in:
Alexandre Dulaunoy 2023-01-07 15:52:19 +01:00
commit 9030065e37
Signed by: adulau
GPG key ID: 09E2CD4944E6CBCD
2 changed files with 61 additions and 0 deletions

1
REQUIREMENTS Normal file
View file

@ -0,0 +1 @@
requests

60
bin/providence.py Normal file
View file

@ -0,0 +1,60 @@
import json
import sys
import argparse
from pathlib import Path
import os
import requests
parser = argparse.ArgumentParser(
description="providence is an open source to find company domain names based on a given company name",
epilog="More info: https://github.com/adulau/providence",
)
parser.add_argument("-v", help="increase output verbosity", action="store_true")
parser.add_argument("-n", "--name", type=str, help="Company name to find")
parser.add_argument(
"-t",
"--tld",
type=str,
help="Limit to a specific TLD (if not, all known TLDs are tested)",
)
args = parser.parse_args()
homedir = str(Path.home())
if not args.name:
parser.print_usage()
parser.exit()
def cache_suffixes(url="https://publicsuffix.org/list/public_suffix_list.dat"):
tlds = set()
cachedfile = os.path.join(homedir, ".providence-suffix")
if not os.path.exists(cachedfile):
r = requests.get(url)
with open(cachedfile, 'wb') as fd:
fd.write(r.content)
with open(cachedfile, 'r') as fd:
for line in fd:
tld = line.rstrip()
if not tld.startswith("//"):
tlds.add(tld)
return tlds
def guess_name(name=None, tlds=None):
if name is None:
return False
for t in tlds:
print(f"{name}.{t}")
if not args.tld:
tlds = cache_suffixes()
else:
tlds = [args.tld]
guess_name(name=args.name, tlds=tlds)