Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 18 additions & 4 deletions bittensor/utils/btlogging/loggingmachine.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,13 +120,21 @@ class LoggingMachine(StateMachine, Logger):
| Info.to(Warning)
)

disable_trace = Trace.to(Default)
def disable_trace(self):
"""Disables trace logging by transitioning back to Default."""
self.enable_default()

disable_debug = Debug.to(Default)
def disable_debug(self):
"""Disables debug logging by transitioning back to Default."""
self.enable_default()

disable_warning = Warning.to(Default)
def disable_warning(self):
"""Disables warning logging by transitioning back to Default."""
self.enable_default()

disable_info = Info.to(Default)
def disable_info(self):
"""Disables info logging by transitioning back to Default."""
self.enable_default()

disable_logging = (
Trace.to(Disabled)
Expand Down Expand Up @@ -405,6 +413,7 @@ def before_enable_default(self):
"""Logs status before enable Default."""
self._logger.info("Enabling default logging (Warning level)")
self._logger.setLevel(stdlogging.WARNING)
self._stream_formatter.set_trace(False)
for logger in all_loggers():
if logger.name in self._primary_loggers:
continue
Expand All @@ -417,6 +426,7 @@ def after_enable_default(self):
def before_enable_warning(self):
"""Logs status before enable Warning."""
self._logger.info("Enabling warning.")
self._logger.setLevel(stdlogging.WARNING)
self._stream_formatter.set_trace(True)
for logger in all_loggers():
logger.setLevel(stdlogging.WARNING)
Expand All @@ -430,6 +440,7 @@ def before_enable_info(self):
"""Logs status before enable info."""
self._logger.info("Enabling info logging.")
self._logger.setLevel(stdlogging.INFO)
self._stream_formatter.set_trace(True)
for logger in all_loggers():
if logger.name in self._primary_loggers:
continue
Expand All @@ -443,6 +454,7 @@ def after_enable_info(self):
def before_enable_trace(self):
"""Logs status before enable Trace."""
self._logger.info("Enabling trace.")
self._logger.setLevel(stdlogging.TRACE)
self._stream_formatter.set_trace(True)
for logger in all_loggers():
logger.setLevel(stdlogging.TRACE)
Expand All @@ -465,12 +477,14 @@ def after_disable_trace(self):
def before_enable_debug(self):
"""Logs status before enable Debug."""
self._logger.info("Enabling debug.")
self._logger.setLevel(stdlogging.DEBUG)
self._stream_formatter.set_trace(True)
for logger in all_loggers():
logger.setLevel(stdlogging.DEBUG)

def before_enable_console(self):
"""Logs status before enable Console."""
self._logger.setLevel(stdlogging.DEBUG)
self._stream_formatter.set_trace(True)
for logger in all_loggers():
logger.setLevel(stdlogging.DEBUG)
Expand Down