Merge pull request 'merge develop branch with useful additions.' (#2) from develop into main

Reviewed-on: https://git.faulty.nl/didier/pylogger/pulls/2
This commit is contained in:
Didier 2022-07-29 11:14:22 +00:00
commit 82f951027d
9 changed files with 167 additions and 131 deletions

1
.gitignore vendored
View File

@ -1 +1,2 @@
venv/
*.log

View File

@ -1,11 +1,12 @@
from lib import BaseLogger
from lib import Logger
from .lib.Logger import Logger
_logger = None
def get_or_create_logger() -> Logger:
def get_or_create_logger(logger: Logger = None) -> Logger:
global _logger
if _logger is None:
_logger = Logger()
if not logger:
raise Exception("Global logger not found.")
_logger = logger
return _logger

View File

@ -1,36 +0,0 @@
from .Log import Log
from .LogLevels import basic_levels, Level
from .console import util
import inspect
class BaseLogger:
def __init__(self, name: str = "Generic", do_colors = True):
self.name = name
self.do_colors = do_colors
def _log(self, log: Log):
l_pre = f"{log.level.short} [{log.format_time('%Y-%m-%d %H:%M:%S')}] <{log.class_str() or log.file}>"
l_str = f"{log.format_text()}"
if self.do_colors:
print(f"{log.level.color}{l_pre}{util.reset} {l_str}")
else:
print(f'{l_pre} {l_str}')
def custom(self, level: Level, text: str, *args, _class=None, _file=inspect.stack()[1].filename):
self._log(Log(level, text, *args, _class=None, _file=inspect.stack()[1].filename))
def debug(self, text: str, *args, _class=None):
self._log(Log(basic_levels['DEBUG'], text, *args, _class=_class, _file=inspect.stack()[1].filename))
def info(self, text: str, *args, _class=None):
self._log(Log(basic_levels['INFO'], text, *args, _class=_class, _file=inspect.stack()[1].filename))
def warn(self, text: str, *args, _class=None):
self._log(Log(basic_levels['WARNING'], text, *args, _class=_class, _file=inspect.stack()[1].filename))
def err(self, text: str, *args, _class=None):
self._log(Log(basic_levels['ERROR'], text, *args, _class=_class, _file=inspect.stack()[1].filename))
def fatal(self, text: str, *args, _class=None):
self._log(Log(basic_levels['FATAL'], text, *args, _class=_class, _file=inspect.stack()[1].filename))

View File

@ -5,7 +5,7 @@ from .LogLevels import Level
class Log:
def __init__(self, level: Level, text: str, *args, _class=None, _file=inspect.stack()[1].filename):
def __init__(self, level: Level, text: str, *args, _object=None, _file=inspect.stack()[1].filename):
"""
Create a log message.
"""
@ -14,17 +14,17 @@ class Log:
self.args = args
self.created_on = time.time()
self.file = _file
self._class = _class
self._object = _object
def format_time(self, fmt: str):
return datetime.datetime.fromtimestamp(self.created_on).strftime(fmt)
def class_str(self):
if self._class:
prefix = f'{type(self._class).__name__}'
if hasattr(self._class, 'name'):
if self._object:
prefix = f'{type(self._object).__name__}'
if hasattr(self._object, 'name'):
prefix += ":"
prefix += self._class.name
prefix += self._object.name
return prefix
else:
return None

View File

@ -1,7 +1,7 @@
from lib import console
from . import console
class Level:
def __init__(self, name: str, weight: int, short_name: str or None = None, _color: str = console.fg.white):
def __init__(self, name: str, weight: int, short_name: str or None = None, _color: str = console.FG.white):
"""
Create a logging level.
"""
@ -15,9 +15,9 @@ class Level:
basic_levels = {
'FATAL': Level('FATAL', 10, _color=(console.fg.red + console.util.bold)),
'ERROR': Level('ERROR', 3, _color=console.fg.red),
'WARNING': Level('WARNING', 2, _color=console.fg.yellow),
'INFO': Level('INFO', 1, _color=console.fg.blue),
'DEBUG': Level('DEBUG', 0, _color=console.fg.green)
'FATAL': Level('FATAL', 10, _color=(console.FG.red + console.Utils.bold)),
'ERROR': Level('ERROR', 3, _color=console.FG.red),
'WARNING': Level('WARNING', 2, _color=(console.FG.yellow + console.Utils.bold + console.Utils.underline)),
'INFO': Level('INFO', 1, _color=console.FG.blue),
'DEBUG': Level('DEBUG', 0, _color=console.FG.green)
}

View File

@ -1,4 +1,41 @@
from .BaseLogger import BaseLogger
import sys
from .Log import Log
from .LogLevels import basic_levels, Level
from .console import Utils
import inspect
class Logger(BaseLogger):
pass
class Logger:
def __init__(self, name: str = "Generic", do_colors = True, output = sys.stdout):
self.name = name
self.do_colors = do_colors
self.output = output
def _print(self, data: str, terminator: str = '\r\n'):
self.output.write(data + terminator)
def _log(self, log: Log):
l_pre = f"{log.level.short} [{log.format_time('%Y-%m-%d %H:%M:%S')}] <{log.class_str() or log.file}>"
l_str = f"{log.format_text()}"
if self.do_colors:
self._print(f"{log.level.color}{l_pre}{Utils.reset} {l_str}")
else:
self._print(f'{l_pre} {l_str}')
def custom(self, level: Level, text: str, *args, _object=None, _file=inspect.stack()[1].filename):
self._log(Log(level, text, *args, _object=None, _file=inspect.stack()[1].filename))
def debug(self, text: str, *args, _object=None):
self._log(Log(basic_levels['DEBUG'], text, *args, _object=_object, _file=inspect.stack()[1].filename))
def info(self, text: str, *args, _object=None):
self._log(Log(basic_levels['INFO'], text, *args, _object=_object, _file=inspect.stack()[1].filename))
def warn(self, text: str, *args, _object=None):
self._log(Log(basic_levels['WARNING'], text, *args, _object=_object, _file=inspect.stack()[1].filename))
def err(self, text: str, *args, _object=None):
self._log(Log(basic_levels['ERROR'], text, *args, _object=_object, _file=inspect.stack()[1].filename))
def fatal(self, text: str, *args, _object=None):
self._log(Log(basic_levels['FATAL'], text, *args, _object=_object, _file=inspect.stack()[1].filename))

12
src/lib/MultiLogger.py Normal file
View File

@ -0,0 +1,12 @@
from .Logger import Logger
from .Log import Log
class MultiLogger(Logger):
def __init__(self, sub_loggers: list[Logger], _name="Generic_MultiLogger"):
super().__init__(name=_name)
self._sub_loggers = sub_loggers
def _log(self, log: Log):
for sl in self._sub_loggers:
sl._log(log)

View File

@ -1,7 +1,8 @@
from sys import stdout, stdin
from time import sleep
class fg:
class FG:
black = "\u001b[30m"
red = "\u001b[31m"
green = "\u001b[32m"
@ -13,7 +14,8 @@ class fg:
def rgb(r, g, b): return f"\u001b[38;2;{r};{g};{b}m"
class bg:
class BG:
black = "\u001b[40m"
red = "\u001b[41m"
green = "\u001b[42m"
@ -26,14 +28,14 @@ class bg:
def rgb(r, g, b): return f"\u001b[48;2;{r};{g};{b}m"
class util:
class Utils:
reset = "\u001b[0m"
bold = "\u001b[1m"
underline = "\u001b[4m"
reverse = "\u001b[7m"
clear = "\u001b[2J"
clearline = "\u001b[2K"
clear_line = "\u001b[2K"
up = "\u001b[1A"
down = "\u001b[1B"

View File

@ -1,8 +1,27 @@
from lib.BaseLogger import BaseLogger
from lib.Logger import Logger
from lib.MultiLogger import MultiLogger
from lib.LogLevels import Level
from lib.Log import Log
from lib import console
_LOGGER = BaseLogger("Generic")
_LOGGER = Logger("Generic")
_LOGGER.info("Started Loco {}", "v1")
_LOGGER.info("This is a simple log {}", "message.")
custom_level = Level("FAULTY_LEVEL", 100, "*", (console.Utils.underline + console.Utils.bold + console.FG.cyan))
_LOGGER.custom(custom_level, "ok")
_temp_logger = Logger("tmp_logger", False, open("logs.log", 'a+'))
_MultiLogger = MultiLogger([_LOGGER, _temp_logger], "Potato")
_MultiLogger.warn("OK")
class User:
name = "faulty"
pass
u = User()
_LOGGER.info("logged in.", _object=u)