using a global logger for all `print` about autogen HOT 3 OPEN

My solution is

class Transcript(object):

    def __init__(self, filename):
        self.terminal = sys.stdout
        self.logfile = open(filename, "w")

    def write(self, message):
        self.terminal.write(message)
        self.terminal.flush()
        self.logfile.write(message)
        self.logfile.flush()

    def flush(self):
        # this flush method is needed for python 3 compatibility.
        # this handles the flush command by doing nothing.
        # you might want to specify some extra behavior here.
        pass


def start(filename):
    """Start transcript, appending print output to given filename"""
    sys.stdout = Transcript(filename)


def stop():
    """Stop transcript and return print functionality to normal"""
    sys.stdout.logfile.close()
    sys.stdout = sys.stdout.terminal


start('filename')

main()

stop()

from autogen.

speaking of this, could add a feature to also save all print to a file at the same time?

from autogen.

`
import sys
import logging

class LoggerWriter:
def init(self, level):
self.logger = logging.getLogger()
self.level = level
self.message = ''
def write(self, message):
if '\n' in message:
self.message += message.rstrip('\n')
if self.message:
self.logger.log(self.level, self.message)
self.message = ''
else:
self.message += message
def flush(self):
pass
def isatty(self):
return False

console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(logging.Formatter('%(message)s'))

file_handler = logging.FileHandler('myapp.log', 'w')
file_handler.setFormatter(logging.Formatter('%(message)s'))

logger = logging.getLogger()
logger.setLevel(logging.INFO)
logger.addHandler(console_handler)
logger.addHandler(file_handler)

fmt = 'Thinking .... - %(message)s'
formatter = logging.Formatter(fmt)
file_handler.setFormatter(formatter)

sys.stdout = LoggerWriter(logging.INFO)
sys.stderr = LoggerWriter(logging.ERROR)

if name == 'main':
print("This will go", end=" ")
print("to the log file")
`

from autogen.