[RARSLAVE] Config-ify rarslave

[rarslave2.git] / rarslave.py

diff --git a/rarslave.py b/rarslave.py
index 93debab..3f251f3 100644 (file)
--- a/rarslave.py
+++ b/rarslave.py
@@ -3,10 +3,12 @@
 
 import re, os, sys
 import par2parser
+import RarslaveConfig
 
 # Global Variables
 (TYPE_OLDRAR, TYPE_NEWRAR, TYPE_ZIP, TYPE_NOEXTRACT) = range (4)
 (ECHECK, EEXTRACT, EDELETE) = range(1,4)
+config = RarslaveConfig.RarslaveConfig()
 
 class RarslaveExtractor (object):
 
@@ -23,11 +25,11 @@ class RarslaveExtractor (object):
 
                self.heads.append (os.path.join (dir, head))
 
-       def extract (self, todir):
+       def extract (self, todir=None):
                # Extract all heads of this set
 
                # Create the directory $todir if it doesn't exist
-               if not os.path.isdir (todir):
+               if todir != None and not os.path.isdir (todir):
                        # TODO: LOGGER
                        try:
                                os.makedirs (todir)
@@ -44,13 +46,17 @@ class RarslaveExtractor (object):
 
                # Call the extraction function on each head
                for h in self.heads:
-                       extraction_func (h, todir)
+                       if todir == None:
+                               # Run in the head's directory
+                               extraction_func (h, os.path.dirname (h))
+                       else:
+                               extraction_func (h, todir)
 
        def __extract_rar (self, file, todir):
                assert os.path.isfile (file)
                assert os.path.isdir (todir)
 
-               RAR_CMD = 'unrar x -o+ -- '
+               RAR_CMD = config.get_value ('commands', 'unrar')
 
                cmd = '%s \"%s\"' % (RAR_CMD, file)
                ret = run_command (cmd, todir)
@@ -60,7 +66,7 @@ class RarslaveExtractor (object):
                        return -EEXTRACT
 
        def __extract_zip (self, file, todir):
-               ZIP_CMD = 'unzip \"%s\" -d \"%s\"'
+               ZIP_CMD = config.get_value ('commands', 'unzip')
 
                cmd = ZIP_CMD % (file, todir)
                ret = run_command (cmd)
@@ -72,7 +78,9 @@ class RarslaveExtractor (object):
        def __extract_noextract (self, file, todir):
                # Just move this file to the $todir, since no extraction is needed
                # FIXME: NOTE: mv will fail by itself if you're moving to the same dir!
-               cmd = 'mv \"%s\" \"%s\"' % (file, todir)
+               NOEXTRACT_CMD = config.get_value ('commands', 'noextract')
+
+               cmd = NOEXTRACT_CMD % (file, todir)
                ret = run_command (cmd)
 
                # Check error code
@@ -96,7 +104,7 @@ class RarslaveRepairer (object):
        def checkAndRepair (self):
                # Form the command:
                # par2repair -- PAR2 PAR2_EXTRA [JOIN_FILES]
-               PAR2_CMD = 'par2repair -- '
+               PAR2_CMD = config.get_value ('commands', 'par2repair')
 
                # Get set up
                basename = get_basename (self.file)
@@ -155,7 +163,7 @@ def get_filename (f):
 def get_basename (name):
        """Strips most kinds of endings from a filename"""
 
-       regex = '^(.+)\.(par2|vol\d+\+\d+|\d\d\d|part\d+|rar|zip|avi|mp4|mkv|ogm)$'
+       regex = config.get_value ('regular expressions', 'basename_regex')
        r = re.compile (regex, re.IGNORECASE)
        done = False
 
@@ -185,7 +193,8 @@ def find_likely_files (name, dir):
 def find_par2_files (files):
        """Find all par2 files in the list $files"""
 
-       regex = re.compile ('^.*\.par2$', re.IGNORECASE)
+       PAR2_REGEX = config.get_value ('regular expressions', 'par2_regex')
+       regex = re.compile (PAR2_REGEX, re.IGNORECASE)
        return [f for f in files if regex.match (f)]
 
 def find_all_par2_files (dir):
@@ -301,7 +310,8 @@ def is_noextract (files):
 def find_deleteable_files (files):
        # Deleteable types regex should come from the config
        dfiles = []
-       dregex = re.compile ('^.*\.(par2|\d|\d\d\d|rar|r\d\d|zip)$', re.IGNORECASE)
+       DELETE_REGEX = config.get_value ('regular expressions', 'delete_regex')
+       dregex = re.compile (DELETE_REGEX, re.IGNORECASE)
 
        return [f for f in files if dregex.match (f)]
 
@@ -353,15 +363,17 @@ class PAR2Set (object):
                        return -ECHECK
 
                # Extraction Stage
+               EXTRACT_DIR = config.get_value ('directories', 'extract_directory')
                extractor = find_extraction_heads (self.dir, self.likely_files)
-               ret = extractor.extract ('extract_dir') # FIXME: Get it from the config
+               ret = extractor.extract (EXTRACT_DIR)
 
                if ret: # FAILURE
                        return -EEXTRACT
 
                # Deletion Stage
+               DELETE_INTERACTIVE = config.get_value ('options', 'interactive')
                deleteable_files = find_deleteable_files (self.likely_files)
-               ret = delete_list (deleteable_files)
+               ret = delete_list (deleteable_files, DELETE_INTERACTIVE)
 
                if ret: # FAILURE
                        return -EDELETE