From 68a82702865184a18b7d41904b329e86c9bf2ce7 Mon Sep 17 00:00:00 2001
From: Ben <benjamyn.love@gmail.com>
Date: Wed, 31 May 2017 18:14:18 +1000
Subject: [PATCH] Fixed up error with my string spliting :) and improved dig
 functionality

---
 HTTPy.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/HTTPy.py b/HTTPy.py
index a38ff09..78efa8a 100755
--- a/HTTPy.py
+++ b/HTTPy.py
@@ -29,11 +29,11 @@ def whois(inDomain):
     test = lookup.split("\n")
     iterate(test)
 
-def dig(inDomain):
-    dig = subprocess.check_output(["dig", "A", "+short", inDomain])
+def dig(inDomain, record):
+    dig = subprocess.check_output(["dig", record, "+short", inDomain])
     print "========================================================" #ADD REST OF DIGS
     test = dig.split("\n")
-    iterate(test)
+    print record, test[0]
 
 def ipLookup(inDomain):
     ipLook = subprocess.check_output(["geoiplookup", inDomain])
@@ -44,7 +44,7 @@ def getInput():
     global inInput 
     global partDomain
     inInput = raw_input()
-    partDomain = inInput.partition(".")
+    partDomain = inInput.split(".")
 
 
 #if partDomain[2] != "":
@@ -53,8 +53,13 @@ def getInput():
 
 print "Domain: "
 getInput()
-if len(partDomain) == 3:
+if len(partDomain) == 4:
     ipLookup(inInput)
+else:
+    whois(inInput)
+    dig(inInput, "A")
+    dig(inInput, "MX")
+    dig(inInput, "TXT")
 
 #whois(inInput)