def setup(self):
self.src_filename = 'source file name'
self.tee = sg._LoggingTee(self.src_filename)
self.output_file = self.tee.output
def log_collector_wrap(log_collector):
"""Wrap our log_collector."""
src_filename = 'source file name'
tee = sg._LoggingTee(src_filename)
output_file = tee.output
yield log_collector, src_filename, tee, output_file