A decorator/context manager to route logging to redis

redis_logger.py

# Usage

log_key = 'smth:%d:log' % some_id
with extra_logging_handler('name', RedisHandler(key=log_key)):
    # Do your thing
    # ...


# Implementation

import logging
from contextlib import contextmanager
import redis

redis_client = redis.StrictRedis(**redis_conf)

@contextmanager
def extra_logging_handler(name, handler):
    logger = logging.getLogger(name)
    logger.addHandler(handler)
    try:
        yield
    except Exception as e:
        logger.error(e, exc_info=True)
    finally:
        logger.removeHandler(handler)

class RedisHandler(logging.Handler):
    def __init__(self, key):
        super(RedisHandler, self).__init__()
        self.key = key

    def emit(self, record):
        # NOTE: other option is saving json.dumps(record.__data__)
        redis_client.rpush(self.key, self.format(record))