From c8804777248c95d5d07d0b9a9be199dc1a4a034d Mon Sep 17 00:00:00 2001
From: Benjamyn Love <benjamyn.love@gmail.com>
Date: Sat, 10 Mar 2018 18:46:47 +1100
Subject: [PATCH] Fixed whois info being fucked

---
 main.py | 25 +++++++++++++------------
 1 file changed, 13 insertions(+), 12 deletions(-)

diff --git a/main.py b/main.py
index 6b2712e..b7b2d28 100755
--- a/main.py
+++ b/main.py
@@ -7,12 +7,10 @@ import json
 from pprint import pprint
 from libs import config
 
-running = 0
-prompt = '$ '
+running = 1
+prompt = ': '
 indent = '~ '
 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):
 	whoisData = whois.whois(inData).query()
@@ -23,12 +21,12 @@ def signal_handler(singal, frame):
     quit()
 
 def printWhois(resultData):
-    try:
-        for recType in whoisrecs:        
-            for item in resultData[recType]:
-                print(indent + '%s: %s' % (recType, item))
-    except Exception as e:
-        pass
+    for x in range(0, len(whoisrecs)):        
+        try:
+            for item in resultData[whoisrecs[x]]:
+                print(indent + '%s: %s' % (whoisrecs[x], item))
+        except Exception as e:
+            pass
 
 def parseInput(inData):
     if inData.lower() == "quit":
@@ -44,7 +42,10 @@ def parseInput(inData):
 signal.signal(signal.SIGINT, signal_handler)
 
 jsonData = config.readConf()
-pprint(jsonData)
+#pprint(jsonData)
+subdomains = jsonData["subdomains"]
+dnsrecords = jsonData["dnsrecords"]
+
 
 print "root"
 for rdata in dns.resolver.query('yahoo.com', 'A') :
@@ -58,4 +59,4 @@ for sub in subdomains:
 		pass
 
 while running == 1:
-    parseInput(raw_input("Please Enter a domain" + prompt) )
+    parseInput(raw_input("\nPlease Enter a domain" + prompt) )