|
| 1 | +# -*- coding: utf-8 -*- |
| 2 | +# |
| 3 | +# GrimoireLab logging setup. |
| 4 | +# |
| 5 | +# Logging is configured using 'structlog', 'django-structlog', |
| 6 | +# and the standard library 'logging' module. |
| 7 | +# |
| 8 | +# More info: |
| 9 | +# |
| 10 | +# https://django-structlog.readthedocs.io/en/latest/ |
| 11 | +# https://docs.djangoproject.com/en/4.2/topics/logging/#configuring-logging |
| 12 | +# |
| 13 | + |
| 14 | +import structlog |
| 15 | + |
| 16 | + |
| 17 | +# Default column styles for the console renderer. |
| 18 | + |
| 19 | +styles = structlog.dev._ColorfulStyles |
| 20 | + |
| 21 | +logger_name_formatter = structlog.dev.KeyValueColumnFormatter( |
| 22 | + key_style=None, |
| 23 | + value_style=styles.bright + styles.logger_name, |
| 24 | + reset_style=styles.reset, |
| 25 | + value_repr=str, |
| 26 | + prefix="[", |
| 27 | + postfix="]" |
| 28 | +) |
| 29 | + |
| 30 | +console_columns = [ |
| 31 | + structlog.dev.Column( |
| 32 | + "timestamp", |
| 33 | + structlog.dev.KeyValueColumnFormatter( |
| 34 | + key_style=None, |
| 35 | + value_style=styles.timestamp, |
| 36 | + reset_style=styles.reset, |
| 37 | + value_repr=str |
| 38 | + ), |
| 39 | + ), |
| 40 | + structlog.dev.Column( |
| 41 | + "level", |
| 42 | + structlog.dev.LogLevelColumnFormatter( |
| 43 | + structlog.dev.ConsoleRenderer.get_default_level_styles(), |
| 44 | + reset_style=styles.reset, |
| 45 | + ), |
| 46 | + ), |
| 47 | + structlog.dev.Column("logger", logger_name_formatter), |
| 48 | + structlog.dev.Column("logger_name", logger_name_formatter), |
| 49 | + structlog.dev.Column( |
| 50 | + "event", |
| 51 | + structlog.dev.KeyValueColumnFormatter( |
| 52 | + key_style=None, |
| 53 | + value_style=styles.bright, |
| 54 | + reset_style=styles.reset, |
| 55 | + value_repr=str |
| 56 | + ), |
| 57 | + ), |
| 58 | + # Default formatter for all keys not explicitly mentioned. |
| 59 | + structlog.dev.Column( |
| 60 | + "", |
| 61 | + structlog.dev.KeyValueColumnFormatter( |
| 62 | + key_style=styles.kv_key, |
| 63 | + value_style=styles.kv_value, |
| 64 | + reset_style=styles.reset, |
| 65 | + value_repr=str |
| 66 | + ), |
| 67 | + ) |
| 68 | +] |
| 69 | + |
| 70 | + |
| 71 | +# Configuration of chain processors for logs not generated by structlog. |
| 72 | +# This will add default fields such the log level, timestamp, etc. |
| 73 | + |
| 74 | +pre_chain_processors = [ |
| 75 | + structlog.processors.TimeStamper(fmt="iso", utc=True), |
| 76 | + structlog.stdlib.add_log_level, |
| 77 | + structlog.stdlib.add_logger_name, |
| 78 | + structlog.stdlib.ExtraAdder() |
| 79 | +] |
| 80 | + |
| 81 | + |
| 82 | +# Default logging configuration for GrimoireLab |
| 83 | + |
| 84 | +_GRIMOIRELAB_LOGGING_CONFIG = { |
| 85 | + "version": 1, |
| 86 | + "disable_existing_loggers": False, |
| 87 | + "formatters": { |
| 88 | + "plain": { |
| 89 | + "()": structlog.stdlib.ProcessorFormatter, |
| 90 | + "processors": [ |
| 91 | + structlog.stdlib.ProcessorFormatter.remove_processors_meta, |
| 92 | + structlog.dev.ConsoleRenderer(columns=console_columns), |
| 93 | + ], |
| 94 | + "foreign_pre_chain": pre_chain_processors, |
| 95 | + }, |
| 96 | + "json": { |
| 97 | + "()": structlog.stdlib.ProcessorFormatter, |
| 98 | + "processors": [ |
| 99 | + structlog.stdlib.ProcessorFormatter.remove_processors_meta, |
| 100 | + structlog.processors.JSONRenderer(), |
| 101 | + ], |
| 102 | + "foreign_pre_chain": pre_chain_processors, |
| 103 | + }, |
| 104 | + "not_structured": { |
| 105 | + "format": "[%(asctime)s - %(name)s - %(levelname)s] - %(message)s" |
| 106 | + }, |
| 107 | + }, |
| 108 | + "handlers": { |
| 109 | + "default": { |
| 110 | + "class": "logging.StreamHandler", |
| 111 | + "formatter": "plain", |
| 112 | + }, |
| 113 | + "json": { |
| 114 | + "class": "logging.StreamHandler", |
| 115 | + "formatter": "json", |
| 116 | + }, |
| 117 | + "scheduler": { |
| 118 | + "class": "logging.StreamHandler", |
| 119 | + "formatter": "not_structured", |
| 120 | + }, |
| 121 | + } |
| 122 | +} |
| 123 | + |
| 124 | + |
| 125 | +def configure_grimoirelab_logging( |
| 126 | + json_mode: bool = False, |
| 127 | + debug: bool = False |
| 128 | +) -> None: |
| 129 | + """ |
| 130 | + Set up the GrimoireLab logging settings. |
| 131 | +
|
| 132 | + :param json_mode: If True, use JSON format for logging. |
| 133 | + :param debug: If True, set logging level to DEBUG. |
| 134 | + """ |
| 135 | + structlog.configure( |
| 136 | + processors=[ |
| 137 | + structlog.contextvars.merge_contextvars, |
| 138 | + structlog.stdlib.filter_by_level, |
| 139 | + structlog.stdlib.PositionalArgumentsFormatter(), |
| 140 | + structlog.processors.TimeStamper(fmt="iso", utc=True), |
| 141 | + structlog.stdlib.add_log_level, |
| 142 | + structlog.stdlib.add_logger_name, |
| 143 | + structlog.processors.StackInfoRenderer(), |
| 144 | + structlog.processors.format_exc_info, |
| 145 | + structlog.processors.UnicodeDecoder(), |
| 146 | + structlog.stdlib.ProcessorFormatter.wrap_for_formatter, |
| 147 | + ], |
| 148 | + logger_factory=structlog.stdlib.LoggerFactory(), |
| 149 | + wrapper_class=structlog.stdlib.BoundLogger, |
| 150 | + cache_logger_on_first_use=True, |
| 151 | + ) |
| 152 | + |
| 153 | + logging_settings = dict(_GRIMOIRELAB_LOGGING_CONFIG) |
| 154 | + logging_settings["loggers"] = { |
| 155 | + "": { |
| 156 | + "handlers": ["json"] if json_mode else ["default"], |
| 157 | + "level": "DEBUG" if debug else "INFO", |
| 158 | + }, |
| 159 | + "grimoirelab.core.scheduler": { |
| 160 | + "handlers": ["scheduler"], |
| 161 | + } |
| 162 | + } |
| 163 | + |
| 164 | + return logging_settings |
0 commit comments