feat: added CTRL+C handler

Datalux · Aug 26, 2020 · c2c3c3e · c2c3c3e
1 parent 75b23d7
commit c2c3c3e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -5,6 +5,7 @@
 import argparse
 from src import printcolors as pc
 import sys
+import signal
 
 
 def printlogo():
@@ -63,6 +64,13 @@ def cmdlist():
     print("Get a list of user who commented target's photos")
 
 
+def signal_handler(sig, frame):
+    pc.printout("\nGoodbye!\n", pc.RED)
+    sys.exit(0)
+
+
+signal.signal(signal.SIGINT, signal_handler)
+
 printlogo()
 
 parser = argparse.ArgumentParser(description='Osintgram is a OSINT tool on Instagram. It offers an interactive shell '