WPIQ/wo/core/logging.py

"""WordOps log module"""


class Log:
    """
        Logs messages with colors for different messages
        according to functions
    """
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

    def error(self, msg, exit=True):
        """
        Logs error into log file
        """
        print(Log.FAIL + msg + Log.ENDC)
        self.app.log.error(Log.FAIL + msg + Log.ENDC)
        if exit:
            self.app.close(1)

    def info(self, msg, end='\n', log=True):
        """
        Logs info messages into log file
        """

        print(Log.OKBLUE + msg + Log.ENDC, end=end)
        if log:
            self.app.log.info(Log.OKBLUE + msg + Log.ENDC)

    def warn(self, msg):
        """
        Logs warning into log file
        """
        print(Log.WARNING + msg + Log.ENDC)
        self.app.log.warning(Log.BOLD + msg + Log.ENDC)

    def debug(self, msg):
        """
        Logs debug messages into log file
        """
        self.app.log.debug(Log.HEADER + msg + Log.ENDC, __name__)

    def wait(self, msg, end='\r', log=True):
        """
        Logs info messages with validation step
        """
        space_to_add = int(31 - len(msg[0:31]))
        space = "                             "
        print(
            Log.OKBLUE + "{0}".format(msg[0:31]) +
            "{0}".format(space[0:space_to_add]) +
            " [" + Log.ENDC + ".." + Log.OKBLUE + "]" + Log.ENDC, end=end)
        if log:
            self.app.log.info(Log.OKBLUE + msg + Log.ENDC)

    def valide(self, msg, end='\n', log=True):
        """
        Logs info messages after validation step
        """
        space_to_add = int(31 - len(msg[0:31]))
        space = "                              "
        print(
            Log.OKBLUE + "{0}".format(msg[0:31]) +
            "{0}".format(space[0:space_to_add]) +
            " [" + Log.ENDC + Log.OKGREEN + "OK" +
            Log.ENDC + Log.OKBLUE + "]" + Log.ENDC, end=end)
        if log:
            self.app.log.info(Log.OKGREEN + msg + Log.ENDC)

    def failed(self, msg, end='\n', log=True):
        """
        Logs info messages after validation step
        """
        space_to_add = int(31 - len(msg[0:31]))
        space = "                             "
        print(
            Log.OKBLUE + "{0}".format(msg[0:31]) +
            "{0}".format(space[0:space_to_add]) +
            " [" + Log.ENDC + Log.FAIL + "KO" +
            Log.ENDC + Log.OKBLUE + "]" + Log.ENDC, end=end)
        if log:
            self.app.log.info(Log.FAIL + msg + Log.ENDC)