utils.py 993 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  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)
  19. class Header():
  20. """
  21. """
  22. def __init__(self, message, top='-', bottom='-'):
  23. self.message = message
  24. self.top = top
  25. self.bottom = bottom
  26. def __str__(self):
  27. topheader = self.top * 80
  28. topheader = topheader[:80]
  29. bottomheader = self.bottom * 80
  30. bottomheader = bottomheader[:80]
  31. return "%s\n %s\n%s" % (topheader, self.message, bottomheader)