Compatibility with python3.5, remove sandbox on keyboard interrupt

1 files changed, 7 insertions, 3 deletions

diff --git a/main.py b/main.py
index 97f4702..85111c0 100755
--- a/main.py
+++ b/main.py
@@ -2,15 +2,19 @@
 
 import sys
 import argparse
+import shutil
 
 import utils
 import config
 import suites
 
 def main():
-    suites.suite_quote()
-    suites.suite_echo()
-    suites.suite_redirection()
+    try:
+        suites.suite_quote()
+        suites.suite_echo()
+        suites.suite_redirection()
+    except KeyboardInterrupt:
+        shutil.rmtree(config.SANDBOX_PATH)
 
 
 if __name__ == "__main__":