Expose context in OP initialize and finalize

Expose the context to the initialize and finalize functions for Output
Processors. This was found to be necessary for the upcoming PostgreSQL
Output Processor.

4 files changed, 4 insertions, 4 deletions

diff --git a/wa/output_processors/cpustates.py b/wa/output_processors/cpustates.py
index 7ab022df..e5545c59 100755
--- a/wa/output_processors/cpustates.py
+++ b/wa/output_processors/cpustates.py
@@ -85,7 +85,7 @@ class CpuStatesProcessor(OutputProcessor):
                   """),
     ]
 
-    def initialize(self):
+    def initialize(self, context):
         self.iteration_reports = OrderedDict()  # pylint: disable=attribute-defined-outside-init
 
     def process_job_output(self, output, target_info, run_output):  # pylint: disable=unused-argument
diff --git a/wa/output_processors/csvproc.py b/wa/output_processors/csvproc.py
index 884f6542..08432f7a 100644
--- a/wa/output_processors/csvproc.py
+++ b/wa/output_processors/csvproc.py
@@ -56,7 +56,7 @@ class CsvReportProcessor(OutputProcessor):
                   'use_all_classifiers is True'
             raise ConfigError(msg)
 
-    def initialize(self):
+    def initialize(self, context):
         # pylint: disable=attribute-defined-outside-init
         self.outputs_so_far = []
         self.artifact_added = False
diff --git a/wa/output_processors/sqlite.py b/wa/output_processors/sqlite.py
index 1064a4a3..30c066d5 100644
--- a/wa/output_processors/sqlite.py
+++ b/wa/output_processors/sqlite.py
@@ -108,7 +108,7 @@ class SqliteResultProcessor(OutputProcessor):
 
     ]
 
-    def initialize(self):
+    def initialize(self, context):
         self._last_spec = None
         self._run_oid = None
         self._spec_oid = None
diff --git a/wa/output_processors/targz.py b/wa/output_processors/targz.py
index 27087ab9..3b573e41 100644
--- a/wa/output_processors/targz.py
+++ b/wa/output_processors/targz.py
@@ -56,7 +56,7 @@ class TargzProcessor(OutputProcessor):
                   '''),
     ]
 
-    def initialize(self):
+    def initialize(self, context):
         if self.delete_output:
             self.logger.debug('Registering RUN_FINALIZED handler.')
             signal.connect(self.delete_output_directory, signal.RUN_FINALIZED, priority=-100)