Fixed whois info being fucked

This commit is contained in:
Benjamyn Love 2018-03-10 18:46:47 +11:00
parent 84b909e7cb
commit c880477724

19
main.py
View File

@ -7,12 +7,10 @@ import json
from pprint import pprint from pprint import pprint
from libs import config from libs import config
running = 0 running = 1
prompt = '$ ' prompt = ': '
indent = '~ ' indent = '~ '
whoisrecs = ["NotFound", "DomainName", "Registrar", "Status", "UpdatedDate", "CreationDate", "ExpirationDate", "RegistrantName", "EligibilityName", "Eligibilitytype", "NameServer"] whoisrecs = ["NotFound", "DomainName", "Registrar", "Status", "UpdatedDate", "CreationDate", "ExpirationDate", "RegistrantName", "EligibilityName", "Eligibilitytype", "NameServer"]
subdomains = ["www", "mail", "ftp"]
dnsrecords = ["A", "MX", "TXT", "NS", "SOA"]
def domainWhois(inData): def domainWhois(inData):
whoisData = whois.whois(inData).query() whoisData = whois.whois(inData).query()
@ -23,10 +21,10 @@ def signal_handler(singal, frame):
quit() quit()
def printWhois(resultData): def printWhois(resultData):
for x in range(0, len(whoisrecs)):
try: try:
for recType in whoisrecs: for item in resultData[whoisrecs[x]]:
for item in resultData[recType]: print(indent + '%s: %s' % (whoisrecs[x], item))
print(indent + '%s: %s' % (recType, item))
except Exception as e: except Exception as e:
pass pass
@ -44,7 +42,10 @@ def parseInput(inData):
signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGINT, signal_handler)
jsonData = config.readConf() jsonData = config.readConf()
pprint(jsonData) #pprint(jsonData)
subdomains = jsonData["subdomains"]
dnsrecords = jsonData["dnsrecords"]
print "root" print "root"
for rdata in dns.resolver.query('yahoo.com', 'A') : for rdata in dns.resolver.query('yahoo.com', 'A') :
@ -58,4 +59,4 @@ for sub in subdomains:
pass pass
while running == 1: while running == 1:
parseInput(raw_input("Please Enter a domain" + prompt) ) parseInput(raw_input("\nPlease Enter a domain" + prompt) )