From 2fadf31a822ab204503da0c37600e3e1d233c93a Mon Sep 17 00:00:00 2001 From: Charles Date: Tue, 16 Jun 2020 13:06:33 +0200 Subject: Compatibility with python3.5, remove sandbox on keyboard interrupt --- main.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'main.py') 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__": -- cgit