sh_cpl/src/sh_edraft/logging/logger.py

import datetime
import os
import traceback
from string import Template

from sh_edraft.hosting.base.application_runtime_base import ApplicationRuntimeBase
from sh_edraft.logging.base.logger_base import LoggerBase
from sh_edraft.logging.model.logging_settings import LoggingSettings
from sh_edraft.logging.model.logging_level import LoggingLevel
from sh_edraft.time.model.time_format_settings import TimeFormatSettings
from sh_edraft.console.console import Console
from sh_edraft.console.model.foreground_color import ForegroundColor


class Logger(LoggerBase):

    def __init__(self, logging_settings: LoggingSettings, time_format: TimeFormatSettings, app_runtime: ApplicationRuntimeBase):
        LoggerBase.__init__(self)

        self._app_runtime = app_runtime
        self._log_settings: LoggingSettings = logging_settings
        self._time_format_settings: TimeFormatSettings = time_format

        self._log = Template(self._log_settings.filename).substitute(
            date_time_now=self._app_runtime.date_time_now.strftime(self._time_format_settings.date_time_format),
            start_time=self._app_runtime.start_time.strftime(self._time_format_settings.date_time_log_format)
        )
        self._path = self._log_settings.path
        self._level = self._log_settings.level
        self._console = self._log_settings.console

        self.create()

    def _get_datetime_now(self) -> str:
        try:
            return datetime.datetime.now().strftime(self._time_format_settings.date_time_format)
        except Exception as e:
            self.error(__name__, 'Cannot get time', ex=e)

    def _get_date(self) -> str:
        try:
            return datetime.datetime.now().strftime(self._time_format_settings.date_format)
        except Exception as e:
            self.error(__name__, 'Cannot get date', ex=e)

    def create(self) -> None:
        """ path """
        try:
            # check if log file path exists
            if not os.path.exists(self._path):
                os.mkdir(self._path)
        except Exception as e:
            self._fatal_console(__name__, 'Cannot create log dir', ex=e)

        """ create new log file """
        try:
            # open log file, create if not exists
            path = f'{self._path}{self._log}'
            f = open(path, "w+")
            Console.write_line(f'[{__name__}]: Using log file: {path}')
            f.close()
        except Exception as e:
            self._fatal_console(__name__, 'Cannot open log file', ex=e)

    def _append_log(self, string):
        try:
            # open log file and append always
            if not os.path.isdir(self._path):
                self._fatal_console(__name__, 'Log directory not found')

            with open(self._path + self._log, "a+", encoding="utf-8") as f:
                f.write(string + '\n')
                f.close()
        except Exception as e:
            self._fatal_console(__name__, f'Cannot append log file, message: {string}', ex=e)

    def _get_string(self, name: str, level: LoggingLevel, message: str) -> str:
        log_level = level.name
        return f'<{self._get_datetime_now()}> [ {log_level} ] [ {name} ]: {message}'

    def header(self, string: str):
        # append log and print message
        self._append_log(string)
        Console.set_foreground_color(ForegroundColor.default)
        Console.write_line(string)
        Console.set_foreground_color(ForegroundColor.default)

    def trace(self, name: str, message: str):
        output = self._get_string(name, LoggingLevel.TRACE, message)

        # check if message can be written to log
        if self._level.value >= LoggingLevel.TRACE.value:
            self._append_log(output)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.TRACE.value:
            Console.set_foreground_color(ForegroundColor.green)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

    def debug(self, name: str, message: str):
        output = self._get_string(name, LoggingLevel.DEBUG, message)

        # check if message can be written to log
        if self._level.value >= LoggingLevel.DEBUG.value:
            self._append_log(output)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.DEBUG.value:
            Console.set_foreground_color(ForegroundColor.green)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

    def info(self, name: str, message: str):
        output = self._get_string(name, LoggingLevel.INFO, message)

        # check if message can be written to log
        if self._level.value >= LoggingLevel.INFO.value:
            self._append_log(output)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.INFO.value:
            Console.set_foreground_color(ForegroundColor.green)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

    def warn(self, name: str, message: str):
        output = self._get_string(name, LoggingLevel.WARN, message)

        # check if message can be written to log
        if self._level.value >= LoggingLevel.WARN.value:
            self._append_log(output)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.WARN.value:
            Console.set_foreground_color(ForegroundColor.yellow)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

    def error(self, name: str, message: str, ex: Exception = None):
        output = ''
        if ex is not None:
            tb = traceback.format_exc()
            self.error(name, message)
            output = self._get_string(name, LoggingLevel.ERROR, f'{ex} -> {tb}')
        else:
            output = self._get_string(name, LoggingLevel.ERROR, message)

        # check if message can be written to log
        if self._level.value >= LoggingLevel.ERROR.value:
            self._append_log(output)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.ERROR.value:
            Console.set_foreground_color(ForegroundColor.red)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

    def fatal(self, name: str, message: str, ex: Exception = None):
        output = ''
        if ex is not None:
            tb = traceback.format_exc()
            self.error(name, message)
            output = self._get_string(name, LoggingLevel.FATAL, f'{ex} -> {tb}')
        else:
            output = self._get_string(name, LoggingLevel.FATAL, message)

        # check if message can be written to log
        if self._level.value >= LoggingLevel.FATAL.value:
            self._append_log(output)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.FATAL.value:
            Console.set_foreground_color(ForegroundColor.red)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

        exit()

    def _fatal_console(self, name: str, message: str, ex: Exception = None):
        output = ''
        if ex is not None:
            tb = traceback.format_exc()
            self.error(name, message)
            output = self._get_string(name, LoggingLevel.ERROR, f'{ex} -> {tb}')
        else:
            output = self._get_string(name, LoggingLevel.ERROR, message)

        # check if message can be shown in console
        if self._console.value >= LoggingLevel.FATAL.value:
            Console.set_foreground_color(ForegroundColor.red)
            Console.write_line(output)
            Console.set_foreground_color(ForegroundColor.default)

        exit()