1
0
Fork 0
mirror of synced 2024-06-25 18:31:09 -04:00
dotbot/dotbot/messenger/messenger.py
Anish Athalye 7200832465 Fix source compatibility with Python 3
According to PEP 394, `python` should only be used in the shebang line
for scripts that are source compatible with both Python 2 and Python 3.

In previous versions of Dotbot, on certain systems where `python`
referred to `python3`, running Dotbot would throw an exception due to a
SyntaxError. This can be fixed by making Dotbot source compatible with
both Python 2 and Python 3.
2014-06-14 23:56:33 -07:00

61 lines
1.6 KiB
Python

import sys
from ..util.singleton import Singleton
from .color import Color
from .level import Level
class Messenger(object):
__metaclass__ = Singleton
def __init__(self, level = Level.LOWINFO):
self.set_level(level)
def set_level(self, level):
self._level = level
def log(self, level, message):
if (level >= self._level):
print('%s%s%s' % (self._color(level), message, self._reset()))
def debug(self, message):
self.log(Level.DEBUG, message)
def lowinfo(self, message):
self.log(Level.LOWINFO, message)
def info(self, message):
self.log(Level.INFO, message)
def warning(self, message):
self.log(Level.WARNING, message)
def error(self, message):
self.log(Level.ERROR, message)
def _color(self, level):
'''
Get a color (terminal escape sequence) according to a level.
'''
if not sys.stdout.isatty():
return ''
elif level < Level.DEBUG:
return ''
elif Level.DEBUG <= level < Level.LOWINFO:
return Color.YELLOW
elif Level.LOWINFO <= level < Level.INFO:
return Color.BLUE
elif Level.INFO <= level < Level.WARNING:
return Color.GREEN
elif Level.WARNING <= level < Level.ERROR:
return Color.MAGENTA
elif Level.ERROR <= level:
return Color.RED
def _reset(self):
'''
Get a reset color (terminal escape sequence).
'''
if not sys.stdout.isatty():
return ''
else:
return Color.RESET