Allow overriding CSV writer.

QuLogic · QuLogic · commit de159e258e40 · 2014-08-12T16:13:19.000-04:00
diff --git a/abbyy2csv.py b/abbyy2csv.py
@@ -67,13 +67,22 @@ def run(self):
         self.logger.debug('Reading file %s ...' % (self.input.name, ))
 
         content = etree.parse(self.input)
-        self.writer = csv.writer(self.output)
+        if self.output:
+            self.writer = csv.writer(self.output)
         for elem in content.iter(PAGE):
             self.processPage(elem)
 
         self.logger.info('Processed %d pages ...' % (self.pages, ))
         self.logger.info('Processed %d lines ...' % (self.total_lines, ))
 
+    def processResults(self, lines):
+        '''
+        Process results and place them in CSV.
+
+        Override this method to do other things with the results.
+        '''
+        self.writer.writerows(lines)
+
     def analyzeCoverPage(self, objs):
         '''
         Analyze a portrait page, which is probably a cover.
@@ -268,7 +277,7 @@ def processPage(self, page):
         else:
             lines = self.analyzePage(page_objs)
 
-        self.writer.writerows(lines)
+        self.processResults(lines)
 
         self.logger.info('    Max columns: %d' % (max(len(x) for x in lines)))
         self.logger.info('    New rows: %d' % (len(lines)))
