Get output of a function as string

Question

I have a function that prints. How to get its output to string?
Note: I cant change the function code.

def f():
    print('hello world')

# do something

assert x == 'hello world\n'

I need this because spark dataframe has method explain that prints info about execution plan. But I need this info as string in my program.

>>> df.explain()
== Physical Plan ==
Scan ExistingRDD[age#0,name#1]

You can redirect stdout. This should help https://stackoverflow.com/questions/4675728/redirect-stdout-to-a-file-in-python https://stackoverflow.com/questions/1218933/can-i-redirect-the-stdout-in-python-into-some-sort-of-string-buffer — icemtel, Feb 22 '19 at 09:46

score 0 · Accepted Answer · answered Feb 22 '19 at 09:48

I found solution here https://stackoverflow.com/a/22434594/4204843

def f():
    print('hello world')

import io
from contextlib import redirect_stdout

with io.StringIO() as buf, redirect_stdout(buf):
    f()
    x = buf.getvalue()

assert x == 'hello world\n'

Get output of a function as string

1 Answers1