PyPrintLogger - A python module to use in built print function as logger

PyPrintLogger

#!/usr/bin/env python

# PyPrintLogger is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# PyPrintLogger is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with PyPrintLogger.  If not, see <http://www.gnu.org/licenses/>.

import builtins
import inspect

from time import strftime


class Logger:
    NONE = 0
    DEBUG = 1
    INFO = 2
    ERROR = 3
    SEVERE = 4

    def __init__(self, mode):
        self.mode = mode
        self.log_file = None
        self._original_print = builtins.print

    def initialize(self):
        builtins.print = self._new_print

    def _can_print(self, msg):
        if self.mode == Logger.NONE:
            return False

        if msg.startswith("DEBUG:"):
            return Logger.DEBUG >= self.mode
        elif msg.startswith("INFO:"):
            return Logger.INFO >= self.mode
        elif msg.startswith("ERROR:"):
            return Logger.ERROR >= self.mode
        elif msg.startswith("SEVERE:"):
            return Logger.SEVERE >= self.mode
        else:
            return True

    def _new_print(self, *args, **kwargs):
        args1 = [str(arg) for arg in args]
        if 'file' not in kwargs:
            try:
                msg = kwargs['sep'].join(args1)
            except:
                msg = ''.join(args1)
            if not self._can_print(msg):
                return

            frm = inspect.currentframe()
            line = ['[', strftime("%Y-%m-%d %H:%M:%S"), "Module: %s" % frm.f_back.f_globals['__name__'],
                    "Line: %s ] " % frm.f_back.f_lineno]
            # kwargs['file'] = self.log_file
            self._original_print(' '.join(line), *args, **kwargs)
        else:
            self._original_print(*args, **kwargs)