ModuleFilter.py

ModuleFilter.py

class ModuleFilter(logging.Filter):
    """
    Filters any logger from outside `root_path` to levels ERROR and above
    """
    
    def __init__(self, root_path: str, *args, **kwargs):
        self.root_path = root_path
        super.__init__(*args, **kwargs)

    def filter(self, record: logging.LogRecord) -> bool:
        if record.pathname.startswith(self.root_path):
            return True
        return record.levelno >= logging.ERROR

# Example usage:

_self_pathname: str = __file__
_proj_root_separator = os.path.sep + "src" + os.path.sep
_proj_root_pathname = _self_pathname[: _self_pathname.index(_proj_root_separator) + len(_proj_root_separator)]

root = logging.getLogger()
for handler in root.handlers:
    handler.addFilter(ModuleFilter(_proj_root_pathname))