2 # vim: set ts=4 sts=4 sw=4 textwidth=92:
6 # PAR2Set-derived types
11 import PAR2Set.ExtractFirstOldRAR
12 import PAR2Set.ExtractFirstNewRAR
16 class RarslaveDetector (object):
18 # A tuple of tuples with the following definition:
19 # (TYPE_NAME, DETECTION_FUNCTION, PAR2Set-derived class)
21 TYPES = ( (PAR2Set.Join.detector, PAR2Set.Join.Join),
22 (PAR2Set.ZIP.detector, PAR2Set.ZIP.ZIP),
23 (PAR2Set.OldRAR.detector, PAR2Set.OldRAR.OldRAR),
24 (PAR2Set.NewRAR.detector, PAR2Set.NewRAR.NewRAR),
25 (PAR2Set.ExtractFirstOldRAR.detector, PAR2Set.ExtractFirstOldRAR.ExtractFirstOldRAR),
26 (PAR2Set.ExtractFirstNewRAR.detector, PAR2Set.ExtractFirstNewRAR.ExtractFirstNewRAR),
29 def __init__ (self, dir, p2file):
31 # The real "meat" of the class
34 self.basename = rsutil.common.get_basename (p2file)
36 # Find files that match by name only
37 self.name_matched_files = rsutil.common.find_name_matches (self.dir, self.basename)
39 # Find all par2 files for this set using name matches
40 self.all_p2files = rsutil.common.find_par2_files (self.name_matched_files)
42 # Try to get the protected files for this set
43 self.prot_matched_files = rsutil.common.parse_all_par2 (self.dir, self.p2file, self.all_p2files)
45 def runMatchingTypes (self):
46 # Now tries to run every type of PAR2Set-derived class for which the detector
47 # detects that the class is valid.
51 for (detector, classname) in self.TYPES:
52 if detector (self.name_matched_files, self.prot_matched_files):
53 # The detector matched, so we're up and running!
55 p2set = classname (self.dir, self.p2file)
58 logging.debug ('Detected type: %s' % p2set)
60 # Try to have rarslave do it's thing
63 # If something already worked, there is no need to continue,
64 # since we've already finished!
65 if ret == rsutil.common.SUCCESS:
68 logging.error ('Detected type failed for: %s' % self.p2file)
70 # Make sure we detected at least one valid type
72 logging.warning ('Unable to determine type: %s' % self.p2file)
73 logging.debug ('The following information will help in writing a detector:')
74 logging.debug ('name_matches: %s' % self.name_matched_files)
75 logging.debug ('prot_matches: %s' % self.prot_matched_files)
76 return -rsutil.common.EDETECT
78 # Make sure that something worked
79 if ret != rsutil.common.SUCCESS:
80 logging.critical ('All types failed for: %s' % self.p2file)
87 if __name__ == '__main__':