diff options
| author | user <user@node5.net> | 2025-02-08 21:53:02 +0100 |
|---|---|---|
| committer | user <user@node5.net> | 2025-02-08 21:53:02 +0100 |
| commit | 32ead71f3a69145dc09df9eb09ca0a6b0191749a (patch) | |
| tree | 4a18f0c54c76fc8599009dcb57c7362417c35b3e /src | |
| parent | 14de1a23248086371671fa45f4057fcd5bdd3205 (diff) | |
Fix log lvls, default to debug 4 cli usage
Disable debug in production using environment variable in SystemD service
Diffstat (limited to 'src')
| -rw-r--r-- | src/blog_node5_net.py | 33 | ||||
| -rw-r--r-- | src/log.py | 37 |
2 files changed, 31 insertions, 39 deletions
diff --git a/src/blog_node5_net.py b/src/blog_node5_net.py index 0b7b3de..da2ef84 100644 --- a/src/blog_node5_net.py +++ b/src/blog_node5_net.py @@ -11,14 +11,43 @@ import markdown import markupsafe import yaml -import log import article import db_handler import telegram_handler -logger = logging.getLogger(__name__) # Set the logger name, to the name of the module +class CustomFormatter(logging.Formatter): + grey = "\x1b[90;20m" + blue = "\x1b[34;20m" + yellow = "\x1b[33;20m" + red = "\x1b[31;20m" + bold_red = "\x1b[31;1m" + reset = "\x1b[0m" + format = "%(asctime)s,%(msecs)03d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s" + + FORMATS = { + logging.DEBUG: grey + format + reset, + logging.INFO: blue + format + reset, + logging.WARNING: yellow + format + reset, + logging.ERROR: red + format + reset, + logging.CRITICAL: bold_red + format + reset + } + + def format(self, record): + log_fmt = self.FORMATS.get(record.levelno) + formatter = logging.Formatter(log_fmt) + return formatter.format(record) + + +logger = logging.getLogger(__name__) +debug = 'PROD' not in os.environ +logger.setLevel(logging.DEBUG if debug else logging.WARNING) +logging.getLogger('werkzeug').setLevel(logging.DEBUG if debug else logging.WARNING) +stream_handler = logging.StreamHandler() +stream_handler.setFormatter(CustomFormatter()) +logger.root.addHandler(stream_handler) +logger.info(f"Debug: {'en' if debug else 'dis'}abled") with open(os.path.join('configs', 'config.yml'), 'r') as file: config = yaml.safe_load(file.read()) diff --git a/src/log.py b/src/log.py deleted file mode 100644 index 9e1b0a6..0000000 --- a/src/log.py +++ /dev/null @@ -1,37 +0,0 @@ -import os -import logging - - -class CustomFormatter(logging.Formatter): - grey = "\x1b[90;20m" - blue = "\x1b[34;20m" - yellow = "\x1b[33;20m" - red = "\x1b[31;20m" - bold_red = "\x1b[31;1m" - reset = "\x1b[0m" - format = "%(asctime)s,%(msecs)03d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s" - - FORMATS = { - logging.DEBUG: grey + format + reset, - logging.INFO: blue + format + reset, - logging.WARNING: yellow + format + reset, - logging.ERROR: red + format + reset, - logging.CRITICAL: bold_red + format + reset - } - - def format(self, record): - log_fmt = self.FORMATS.get(record.levelno) - formatter = logging.Formatter(log_fmt) - return formatter.format(record) - - -logger = logging.getLogger(__name__) -debug = os.environ.get('FLASK_DEBUG') == '1' -logger.setLevel(logging.DEBUG if debug else logging.WARNING) -logging.getLogger('werkzeug').setLevel(logging.DEBUG if debug else logging.WARNING) - -stream_handler = logging.StreamHandler() -stream_handler.setFormatter(CustomFormatter()) -logger.root.addHandler(stream_handler) -logger.info(f"Debug: {'en' if debug else 'dis'}abled") - |
