Store standard output on a variable in Python

storeOutputToString.py

#from StringIO import StringIO  # Python2
from io import StringIO  # Python3
 
import sys
 
# Store the reference, in case you want to show things again in standard output
old_stdout = sys.stdout

# This variable will store everything that is sent to the standard output
result = StringIO()
sys.stdout = result

print("noooo")

# Here we can call anything we like, like external modules, and everything that they will send to standard output will be stored on "result"
# do_fancy_stuff()

sys.stdout = old_stdout

print(result.getvalue())

old_stdout = sys.stdout
result = StringIO()
sys.stdout = result

print("yes")

sys.stdout = old_stdout

print(result.getvalue())