-
-
Notifications
You must be signed in to change notification settings - Fork 407
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Moved Simulation & Montecarlo Logging Frameworks to `tardis/io/logger…
…` folder (#1684) * Moved Logging Frameworks to tardis/io/logger folder * Fixing some errors with the tests * Changed simulation_logger.py to logger.py * Changed implementation to target only TARDIS loggers Removed list_of_filters to check the filters from the logger itself * Removed list creation in logging_state & Moved in tardis_logger()
- Loading branch information
1 parent
7d3899e
commit 1af3a40
Showing
11 changed files
with
267 additions
and
241 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,163 @@ | ||
| import logging | ||
| import sys | ||
| import logging | ||
| import warnings | ||
| import pyne.data | ||
| from tardis.io.logger.colored_logger import ColoredFormatter, formatter_message | ||
|
|
||
| warnings.filterwarnings("ignore", category=pyne.utils.QAWarning) | ||
|
|
||
| FORMAT = "[$BOLD%(name)-20s$RESET][%(levelname)-18s] %(message)s ($BOLD%(filename)s$RESET:%(lineno)d)" | ||
| COLOR_FORMAT = formatter_message(FORMAT, True) | ||
|
|
||
| logging.captureWarnings(True) | ||
| logger = logging.getLogger("tardis") | ||
| logger.setLevel(logging.INFO) | ||
|
|
||
| console_handler = logging.StreamHandler(sys.stdout) | ||
| console_formatter = ColoredFormatter(COLOR_FORMAT) | ||
| console_handler.setFormatter(console_formatter) | ||
|
|
||
| logger.addHandler(console_handler) | ||
| logging.getLogger("py.warnings").addHandler(console_handler) | ||
|
|
||
| LOGGING_LEVELS = { | ||
| "NOTSET": logging.NOTSET, | ||
| "DEBUG": logging.DEBUG, | ||
| "INFO": logging.INFO, | ||
| "WARNING": logging.WARNING, | ||
| "ERROR": logging.ERROR, | ||
| "CRITICAL": logging.CRITICAL, | ||
| } | ||
| DEFAULT_LOG_STATE = "CRITICAL" | ||
|
|
||
|
|
||
| class FilterLog(object): | ||
| """ | ||
| Filter Log Class for Filtering Logging Output | ||
| to a particular level | ||
| Parameters | ||
| ---------- | ||
| log_level : logging object | ||
| allows to have a filter for the | ||
| particular log_level | ||
| """ | ||
|
|
||
| def __init__(self, log_level): | ||
| self.log_level = log_level | ||
|
|
||
| def filter(self, log_record): | ||
| """ | ||
| filter() allows to set the logging level for | ||
| all the record that are being parsed & hence remove those | ||
| which are not of the particular level | ||
| Parameters | ||
| ---------- | ||
| log_record : logging.record | ||
| which the paricular record upon which the | ||
| filter will be applied | ||
| Returns | ||
| ------- | ||
| boolean : True, if the current log_record has the | ||
| level that of the specified log_level | ||
| False, if the current log_record doesn't have the | ||
| same log_level as the specified one | ||
| """ | ||
| return log_record.levelno == self.log_level | ||
|
|
||
|
|
||
| def logging_state(log_state, tardis_config, specific): | ||
| """ | ||
| Function to set the logging configuration for the simulation output | ||
| Called from within run_tardis() | ||
| Configured via functional arguments passed through run_tardis() - log_state & specific | ||
| Configured via YAML parameters under `debug` section - logging_level & specific_logging | ||
| Parameters | ||
| ---------- | ||
| log_state: str | ||
| Allows to input the log level for the simulation | ||
| Uses Python logging framework to determine the messages that will be output | ||
| specific: boolean | ||
| Allows to set specific logging levels. Logs of the `log_state` level would be output. | ||
| """ | ||
|
|
||
| if "debug" in tardis_config: | ||
| specific = ( | ||
| tardis_config["debug"]["specific"] if specific is None else specific | ||
| ) | ||
|
|
||
| logging_level = ( | ||
| log_state if log_state else tardis_config["debug"]["log_state"] | ||
| ) | ||
|
|
||
| # Displays a message when both log_state & tardis["debug"]["log_state"] are specified | ||
| if log_state and tardis_config["debug"]["log_state"]: | ||
| print( | ||
| "log_state is defined both in Functional Argument & YAML Configuration {debug section}" | ||
| ) | ||
| print( | ||
| f"log_state = {log_state.upper()} will be used for Log Level Determination\n" | ||
| ) | ||
|
|
||
| else: | ||
| if log_state: | ||
| logging_level = log_state | ||
| else: | ||
| tardis_config["debug"] = {"log_state": DEFAULT_LOG_STATE} | ||
| logging_level = tardis_config["debug"]["log_state"] | ||
|
|
||
| if specific: | ||
| specific = specific | ||
|
|
||
| logging_level = logging_level.upper() | ||
| if not logging_level in LOGGING_LEVELS: | ||
| raise ValueError( | ||
| f"Passed Value for log_state = {logging_level} is Invalid. Must be one of the following {list(LOGGING_LEVELS.keys())}" | ||
| ) | ||
|
|
||
| # Getting the TARDIS logger & all its children loggers | ||
| logger = logging.getLogger("tardis") | ||
|
|
||
| # Creating a list for Storing all the Loggers which are derived from TARDIS | ||
| tardis_loggers = tardis_logger() | ||
|
|
||
| if logging_level in LOGGING_LEVELS: | ||
| logger.setLevel(LOGGING_LEVELS[logging_level]) | ||
|
|
||
| if logger.filters: | ||
| for filter in logger.filters: | ||
| for logger in tardis_loggers: | ||
| logger.removeFilter(filter) | ||
|
|
||
| if specific: | ||
| filter_log = FilterLog(LOGGING_LEVELS[logging_level]) | ||
| for logger in tardis_loggers: | ||
| logger.addFilter(filter_log) | ||
| else: | ||
| for filter in logger.filters: | ||
| for logger in tardis_loggers: | ||
| logger.removeFilter(filter) | ||
|
|
||
|
|
||
| def tardis_logger(): | ||
| """ | ||
| Generates the list of the loggers which are derived from TARDIS | ||
| All loggers which are of the form `tardis.module_name` are added to the list | ||
| Parameters | ||
| ---------- | ||
| list_for_loggers : list | ||
| List for storing the loggers derived from TARDIS | ||
| Returns | ||
| ------- | ||
| list_for_loggers : list | ||
| """ | ||
| list_for_loggers = [] | ||
| for name in logging.root.manager.loggerDict: | ||
| if not name.find("tardis"): | ||
| list_for_loggers.append(logging.getLogger(name)) | ||
| return list_for_loggers |
File renamed without changes.
Empty file.
Oops, something went wrong.