I'm trying to store the print output from a function of another module imported, as a string and write it into a file. However, that function does not return a string, it only prints the output. so I need something like:
import someModule ...... f.open('test.v','w') out = storetheprintoutputasstring(someModule.main()) f.write(out) f.close How do I do this? Please help me out and thank you in advance
I think what you're asking to do is a bit of a hack, so I assume you have to do it this way.
Here is how you could redirect stdout to a file using the with statement:
import sys from contextlib import contextmanager @contextmanager def redirected(stdout): saved_stdout = sys.stdout sys.stdout = open(stdout, 'w') yield sys.stdout = saved_stdout with redirected(stdout='file.txt'): print 'Hello' print 'Hello again' 
try: yield; finally:....mod1.py:
def main(): print "BOHOO" mod2.py:
import sys from StringIO import StringIO import mod1 def storetheprintoutputasstring(func): saved_stdout = sys.stdout sys.stdout = mystdout = StringIO() func() # Call function sys.stdout = saved_stdout return mystdout.getvalue() f = open('test.v','w') out = storetheprintoutputasstring(mod1.main) f.write(out) f.close() Run python mod2.py.
test.v contains:
BOHOO 
$python someModule.py > output_file.txtIf not, then you'll want to read in from stdout and write to your file. I'm operating under the assumption that you don't need to have this done dynamically.