Add method to handle keyboard interrupt

Author: coderick14

ACedIt/main.py

@@ -44,7 +44,7 @@ def main():
 
     except KeyboardInterrupt:
         # Clean up files here
-        print 'Interruped manually. Exiting gracefully.'
+        util.Utilities.handle_kbd_interrupt(args)
 
 
 if __name__ == '__main__':

ACedIt/util.py

@@ -110,7 +110,7 @@ def check_cache(site, contest, problem):
         """
 
         Utilities.create_workdir_structure(site, contest)
-        
+
         if problem is None:
             if not os.path.isdir(os.path.join(Utilities.cache_dir, site, contest)):
                 os.makedirs(os.path.join(Utilities.cache_dir, site,
@@ -203,6 +203,26 @@ def cleanup(num_cases):
             if os.path.isfile('temp_output' + str(i)):
                 os.remove('temp_output' + str(i))
 
+    @staticmethod
+    def handle_kbd_interrupt(args):
+        """
+        Method to handle keyboard interrupt
+        """
+        from shutil import rmtree
+        print 'Interrupted manually. Cleaning up...'
+
+        if args['problem'] is not None:
+            path = os.path.join(Utilities.cache_dir, args['site'], args['contest'], args['problem'])
+            if os.path.isdir(path):
+                rmtree(path)
+        else:
+            path = os.path.join(Utilities.cache_dir, args['site'], args['contest'])
+            if os.path.isdir(path):
+                rmtree(path)
+
+        print 'Done. Exiting gracefully.'
+
+
     @staticmethod
     def run_solution(problem):
         """