logger.py

# copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import logging
logging.basicConfig()

DEBUG = logging.DEBUG  # 10
INFO = logging.INFO  # 20
WARN = logging.WARN  # 30
ERROR = logging.ERROR  # 40


class Logger(object):
    """
    Logger
    """

    def __init__(self, level=DEBUG):
        self.init(level)

    def init(self, level=DEBUG):
        """
        init
        """
        self._logger = logging.getLogger()
        self._logger.setLevel(level)

    def info(self, fmt, *args):
        """info"""
        self._logger.info(fmt, *args)

    def warning(self, fmt, *args):
        """warning"""
        self._logger.warning(fmt, *args)

    def error(self, fmt, *args):
        """error"""
        self._logger.error(fmt, *args)


_logger = Logger()


def init(level=DEBUG):
    """init for external"""
    _logger.init(level)


def info(fmt, *args):
    """info"""
    _logger.info(fmt, *args)


def warning(fmt, *args):
    """warn"""
    _logger.warning(fmt, *args)


def error(fmt, *args):
    """error"""
    _logger.error(fmt, *args)


def advertisement():
    copyright = "PaddleClas is powered by PaddlePaddle !"
    info = "For more info please go to the following website."
    website = "https://github.com/PaddlePaddle/PaddleClas"
    AD_LEN = 55

    _logger.info("\n{0}\n{1}\n{2}\n{3}\n{4}\n{5}\n{6}\n{7}\n".format(
        "=" * (AD_LEN + 4),
        "=={}==".format(copyright.center(AD_LEN)),
        "=" * (AD_LEN + 4),
        "=={}==".format(' ' * AD_LEN),
        "=={}==".format(info.center(AD_LEN)),
        "=={}==".format(' ' * AD_LEN),
        "=={}==".format(website.center(AD_LEN)),
        "=" * (AD_LEN + 4), ))