utils.py 616 B

12345678910111213141516171819202122232425
  1. import tempfile
  2. class WrapLogger():
  3. """
  4. Used to convert a Logger into file streams. Adds easy integration
  5. of Logger into subprocess, which takes file parameters for stdout
  6. and stderr.
  7. Use:
  8. (out, err) = WrapLogger(logger, logging.INFO), WrapLogger(logger, logging.ERROR)
  9. subprocess.Popen(command, stdout=out, stderr=err)
  10. """
  11. def __init__(self, logger, level):
  12. self.logger = logger
  13. self.level = level
  14. self.file = tempfile.TemporaryFile()
  15. def write(self, message):
  16. self.logger.log(self.level, message)
  17. def __getattr__(self, name):
  18. return getattr(self.file, name)