mirror of
https://github.com/adulau/providence.git
synced 2024-12-04 07:57:14 +00:00
new: [providence] first skeleton
Signed-off-by: Alexandre Dulaunoy <a@foo.be>
This commit is contained in:
commit
9030065e37
2 changed files with 61 additions and 0 deletions
1
REQUIREMENTS
Normal file
1
REQUIREMENTS
Normal file
|
@ -0,0 +1 @@
|
||||||
|
requests
|
60
bin/providence.py
Normal file
60
bin/providence.py
Normal 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)
|
Loading…
Reference in a new issue