Added sig handling and fixed some shit

This commit is contained in:
Benjamyn Love 2018-02-28 18:42:18 +11:00
parent 9435fe0026
commit 4eaf6bb6ef

15
main.py
View File

@ -1,20 +1,25 @@
#!/usr/bin/python2.7 #!/usr/bin/python2.7
import whois import whois
import readline import readline
import signal
running = 1 running = 1
prompt = '$ ' prompt = '$ '
def parseInput(inData): def signal_handler(singal, frame):
global running #print('\n')
if inData.lower() == "quit": quit()
running = 0
pass
def parseInput(inData):
if inData.lower() == "quit":
quit()
#whoisData = whois.whois("benjamyn-testing.com").query() #whoisData = whois.whois("benjamyn-testing.com").query()
#result = whois.Parser("benjamyn-testing.com", whoisData[1]).parse() #result = whois.Parser("benjamyn-testing.com", whoisData[1]).parse()
#print(result) #print(result)
signal.signal(signal.SIGINT, signal_handler)
while running == 1: while running == 1:
parseInput(raw_input("Please Enter a domain" + prompt) ) parseInput(raw_input("Please Enter a domain" + prompt) )