Add feature clear cache for a given site

Author: coderick14

README.md

@@ -51,6 +51,8 @@ optional arguments:
   --set-default-contest DEFAULT_CONTEST
                         Name of default contest to be used when -c flag is not
                         specified
+  --clear-cache         Clear cached test cases for a given site. Takes
+                        default site if -s flag is omitted
 
 ```
 During installation, the default site is set to `codeforces`. You can change it anytime using the above mentioned flags.  

README.rst

@@ -73,6 +73,8 @@ Usage
       --set-default-contest DEFAULT_CONTEST
                             Name of default contest to be used when -c flag is not
                             specified
+      --clear-cache         Clear cached test cases for a given site. Takes
+                            default site if -s flag is omitted
 
 During installation, the default site is set to ``codeforces``. You
 can change it anytime using the above mentioned flags.

acedit/main.py

@@ -6,10 +6,16 @@
 
 
 def validate_args(args):
+    """
+    Method to check valid combination of flags
+    """
 
     if args['default_site'] is not None or args['default_contest'] is not None:
         return
 
+    if args['clear_cache']:
+        return
+
     if not args['site'] == 'spoj' and args['contest'] is None:
         print 'Please specify contest code or set a default contest.'
         sys.exit(0)
@@ -31,10 +37,14 @@ def main():
             # set default site
             util.Utilities.set_constants('default_site', args['default_site'])
 
-        if args['default_contest']:
+        elif args['default_contest']:
             # set default contest
             util.Utilities.set_constants('default_contest', args['default_contest'])
 
+        elif args['clear_cache']:
+            # clear cached test cases
+            util.Utilities.clear_cache(args['site'])
+
         elif args['source']:
             # run code
             util.Utilities.run_solution(args)

acedit/util.py

@@ -66,7 +66,12 @@ def parse_flags(supported_sites):
                             dest='default_contest',
                             help='Name of default contest to be used when -c flag is not specified')
 
-        parser.set_defaults(force=False)
+        parser.add_argument('--clear-cache',
+                            dest='clear_cache',
+                            action='store_true',
+                            help='Clear cached test cases for a given site. Takes default site if -s flag is omitted')
+
+        parser.set_defaults(force=False, clear_cache=False)
 
         args = parser.parse_args()
 
@@ -94,6 +99,7 @@ def parse_flags(supported_sites):
 
         flags['problem'] = args.problem
         flags['force'] = args.force
+        flags['clear_cache'] = args.clear_cache
         flags['source'] = args.source_file
         flags['default_site'] = args.default_site
         flags['default_contest'] = args.default_contest
@@ -138,6 +144,24 @@ def check_cache(site, contest, problem):
                                      contest, problem))
             return False
 
+    @staticmethod
+    def clear_cache(site):
+        """
+        Method to clear cached test cases
+        """
+
+        confirm = raw_input(
+            'Remove entire cache for site %s? (y/N) : ' % (site))
+        if confirm == 'y':
+            from shutil import rmtree
+            try:
+                rmtree(os.path.join(Utilities.cache_dir, site))
+            except:
+                print 'Some error occured. Try again.'
+                return
+            os.makedirs(os.path.join(Utilities.cache_dir, site))
+            print 'Done.'
+
     @staticmethod
     def store_files(site, contest, problem, inputs, outputs):
         """

setup.py

@@ -22,7 +22,7 @@
 
     packages=['acedit'],
 
-    version='1.0.4',
+    version='1.0.5',
 
     description='Download and test against sample test cases from any competitive programming website',