Stream write logger

stream_logger.py

import traceback


class WriteLogger(object):
    def __init__(self, stream):
        self.stream = stream

    def write(self, str):
        stack = traceback.extract_stack()
        stack_lines = traceback.format_list(stack[:-1])
        self.stream.write(str)
        self.stream.write('-- written from --\n')
        for line in stack_lines:
            self.stream.write(line)
        self.stream.write('------------------\n')