diff --git a/HTTPy.py b/HTTPy.py index 867111f..45c7b1b 100755 --- a/HTTPy.py +++ b/HTTPy.py @@ -18,7 +18,7 @@ class bcolors: inInput = "" partDomain = "" running = 1 -pingOn = "P" +pingOn = "p" whoOn = "W" phrases = ["ping", "whois", "quit", "clear", "help", "nameserver"] @@ -201,12 +201,17 @@ while running == 1: if whoOn == "W": whois(inInput) #Condense dig calls - test1, test2 = newdig(inInput, "A") - iterate(test1, test2) - dig(inInput, "MX") - dig(inInput, "TXT") - test = dig(inInput, "NS") - dig(inInput, "SRV") - soaRec = dig(inInput, "SOA") - print compareSOA(test, inInput, soaRec) + aRec, aType = newdig(inInput, "A") + iterate(aRec, aType) + mxRec, mxType = newdig(inInput, "MX") + iterate(mxRec, mxType) + txtRec, txtType = newdig(inInput, "TXT") + iterate(txtRec, txtType) + nsRec, nsType = newdig(inInput, "NS") + iterate(nsRec, nsType) + srvRec, srvType = newdig(inInput, "SRV") + iterate(srvRec, srvType) + soaRec, soaType = newdig(inInput, "SOA") + iterate(soaRec, soaType) + print compareSOA(nsRec, inInput, soaRec) print '\n' \ No newline at end of file