From 4eaf6bb6ef421adfcdadb9e1ede8da93946f38cd Mon Sep 17 00:00:00 2001
From: Benjamyn Love <benjamyn.love@gmail.com>
Date: Wed, 28 Feb 2018 18:42:18 +1100
Subject: [PATCH] Added sig handling and fixed some shit

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

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