import logging

from gunicorn.glogging import Logger


class HealthCheckFilter(logging.Filter):
    def filter(self, record):
        message = record.getMessage()
        return "kube-probe" not in message


class CustomLogger(Logger):
    def setup(self, cfg):
        super().setup(cfg)
        self.access_log.addFilter(HealthCheckFilter())


logger_class = CustomLogger