safelog.py

import logging

class SafeLog:
    def __init__(self, log: logging.Logger):
        self.log = log
        self.safe = False

    def __enter__(self):
        self.safe = True
        return self

    def __getattr__(self, key):
        if not self.safe:
            self.log.warn("Using unsafe log")
        return getattr(self.log, key)

    def __exit__(self, type, value, traceback):
        self.safe = False
        if type or value or traceback:
            # Log the exception and prevent from propagate
            self.log.exception("Error writing to log")
            return True
        return False
 
# Example usage:
 
log = logging.getLogger()
 
safelog = SafeLog(log)

with safelog: safelog.info(f"this will not throw {di['key']}")