logger.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. import os
  2. import logging
  3. import tempfile
  4. from . import constants
  5. LOG_FILE = None
  6. LOGGER = None
  7. LEVELS = {
  8. constants.DEBUG: logging.DEBUG,
  9. constants.INFO: logging.INFO,
  10. constants.WARNING: logging.WARNING,
  11. constants.ERROR: logging.ERROR,
  12. constants.CRITICAL: logging.CRITICAL
  13. }
  14. def init(filename, level=constants.DEBUG):
  15. """Initialize the logger.
  16. :param filename: base name of the log file
  17. :param level: logging level (Default value = DEBUG)
  18. """
  19. global LOG_FILE
  20. LOG_FILE = os.path.join(tempfile.gettempdir(), filename)
  21. with open(LOG_FILE, 'w'):
  22. pass
  23. global LOGGER
  24. LOGGER = logging.getLogger('Three.Export')
  25. LOGGER.setLevel(LEVELS[level])
  26. stream = logging.StreamHandler()
  27. stream.setLevel(LEVELS[level])
  28. format_ = '%(asctime)s - %(name)s - %(levelname)s: %(message)s'
  29. formatter = logging.Formatter(format_)
  30. stream.setFormatter(formatter)
  31. file_handler = logging.FileHandler(LOG_FILE)
  32. file_handler.setLevel(LEVELS[level])
  33. file_handler.setFormatter(formatter)
  34. LOGGER.addHandler(stream)
  35. LOGGER.addHandler(file_handler)
  36. def info(*args):
  37. LOGGER.info(*args)
  38. def debug(*args):
  39. LOGGER.debug(*args)
  40. def warning(*args):
  41. LOGGER.warning(*args)
  42. def error(*args):
  43. LOGGER.error(*args)
  44. def critical(*args):
  45. LOGGER.critical(*args)