Source code for fedlab_core.utils.logger

import logging
import sys

logging.basicConfig(stream=sys.stdout, level=logging.INFO,
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')


class logger(object):
    """record cmd info to file and print it to cmd at the same time"""

    def __init__(self, log_file, client_name):
        self.logger = logging.getLogger(client_name)
        handler = logging.FileHandler(log_file, mode='w')
        handler.setLevel(level=logging.INFO)
        formatter = logging.Formatter(
            '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        handler.setFormatter(formatter)
        self.logger.addHandler(handler)

    def info(self, log_str):
        """"""
        self.logger.info(log_str)

    def error(self, error_str, log_str):
        """"""
        raise NotImplementedError()

    def warning(self, warning_str):
        """"""
        raise NotImplementedError()