logwriter

#
# Copyright 2009 Vinay Sajip
#
import logging

logger = logging.getLogger(__name__)

class LogWriter:
    def __init__(self, logger):
        self.logger = logger

    def as_stream(self, levelname):
        level = logging.getLevelName(levelname.upper())
        logger = self.logger

        class StreamProxy:
            def write(self, s):
                if s != '\n':
                    logger.log(level, s)

        return StreamProxy()

def main():
    logging.basicConfig(level=logging.DEBUG)
    lw = LogWriter(logger)
    info = lw.as_stream('info')
    debug = lw.as_stream('debug')
    print >> info, 'Info!',
    print >> debug, 'Debug!',

if __name__ == '__main__':
    main()