WebSVN – programming – Diff – /projects/rarslave/rarslave.py

+#
 # - 2005-11-06
 # - Fixed the rar command so that it can extract files whose names begin
 #   with a hyphen.
+#
+# - 2006-03-08
+# - Make an interactive mode which asks the user before deleting files.
+#
 ################################################################################
 # REQUIREMENTS:
+#
 # This code requires the programs cfv, par2repair, lxsplit, and rar to be able
         self.__defaults = {
             ('directories', 'working_directory') : '~/downloads/usenet',
             ('options', 'recursive') : True,
             ('options', 'check_required_programs') : False,
             ('options', 'extract_with_full_path') : False,
+            ('options', 'interactive') : False,
             ('regular expressions', 'par2_regex') : '.*\.par2$',
             ('regular expressions', 'video_file_regex') : '.*\.(avi|ogm|mkv|mp4)$',
             ('regular expressions', 'temp_repair_regex') : '.*\.1$',
             ('regular expressions', 'remove_regex') : '^.*\.(rar|r\d\d)$' }
         self.__user_config = self.__read_config()
 # This is the global config variable.
 config = rarslave_config()
+# This is the global options variable. (to be set later)
+options = None
 ################################################################################
 # The rarslave_output class
+#
 # This class handles the nice output summary which is printed at the end
 # of a run
         print
         print '=== extra_pars ==='
         for f in self.extra_pars:
             print f
         print
         print '=== files ==='
         for f in self.files:
             print f
         if not self.extracted:
             print 'Did not extract yet, not removing currentset'
             return
+        files_to_remove = []
         # remove the main par
-        os.remove(self.parfile)
+        files_to_remove.append(self.parfile)
         # remove all of the extra pars
         for i in self.extra_pars:
-            os.remove(i)
+            files_to_remove.append(i)
         # remove any rars that are associated (leave EVERYTHING else)
         # This regex matches both old and new style rar(s) by default.
         regex = re.compile(
                 config.get_value('regular expressions', 'remove_regex'),
                 re.IGNORECASE)
         for i in self.files:
             if regex.match(i):
-                os.remove(i)
+                files_to_remove.append(i)
         # remove any .{001,002,...} files (from parjoin)
         if self.used_parjoin:
             for i in os.listdir(os.getcwd()):
                 if i != self.files[0] and self.files[0] in i:
-                    os.remove(i)
+                    files_to_remove.append(i)
         # remove any temp repair files
         regex = re.compile(
                 config.get_value('regular expressions', 'temp_repair_regex'),
                 re.IGNORECASE)
-        [os.remove(f) for f in os.listdir(os.getcwd()) if regex.match(f)]
+        [files_to_remove.append(f) for f in os.listdir(os.getcwd()) if regex.match(f)]
+        # interactively remove files
+        if options.interactive:
+            print # blank line
+            for f in files_to_remove:
+                print f
+            print '========================================'
+            done = False
+            while not done:
+                s = raw_input("Delete files [y,n]: ")
+                s.lower()
+                if s == 'y' or s == 'yes':
+                    done = True
+                    self.__remove_list_of_files(files_to_remove)
+                elif s == 'n' or s == 'no':
+                    done = True
+                    print 'Not removing files'
+                else:
+                    print 'Bad selection, try again...'
+        else:
+            self.__remove_list_of_files(files_to_remove)
+    def __remove_list_of_files(self, files_to_remove):
+        """Remove all files in the list"""
+        for f in files_to_remove:
+            os.remove(f)
     def __get_extract_file(self):
         """Find the first extractable file"""
         for i in self.files:
             if os.path.splitext(i)[1] == '.rar':
             cur.files = filenames
             parsets.append(cur)
     return parsets
-def directory_worker(dir, options):
+def directory_worker(dir):
     """Attempts to find, verify, and extract every parset in the directory
     given as a parameter"""
     cwd = os.getcwd()
     os.chdir(dir)
     parser.add_option('-o', '--output-debug-info',
                        action='store_true', dest='debug_info',
                        default=False,
                        help="Output debug info for every parset, then exit")
+    parser.add_option('-i', '--interactive', dest='interactive', action='store_true',
+                      default=config.get_value('options', 'interactive'),
+                      help="Confirm before removing files")
     # Parse the given options
+    global options
     (options, args) = parser.parse_args()
     # Fix up the working directory
     options.work_dir = os.path.abspath(os.path.expanduser(options.work_dir))
         config.write_config()
     # Run rarslave!
     if options.recursive:
         for root, dirs, files in os.walk(options.work_dir):
-            directory_worker(root, options)
+            directory_worker(root)
     else:
-        directory_worker(options.work_dir, options)
+        directory_worker(options.work_dir)
     # Print the results
     output.print_results_table()
 if __name__ == '__main__':

Subversion Repositories programming

(root)/projects/rarslave/rarslave.py – Rev 220 → 275