CODEWITHSUNDEEP
pythonfile-iostringio
gauteh
gauteh

Reputation: 17195

What is the best way to write the contents of a StringIO to a file?

What is the best way to write the contents of a StringIO buffer to a file ?

I currently do something like:

buf = StringIO()
fd = open('file.xml', 'w')
# populate buf
fd.write(buf.getvalue ())

But then buf.getvalue() would make a copy of the contents?

Upvotes: 62

Views: 69990

Answers (2)

Steven
Steven

Reputation: 28686

Use shutil.copyfileobj:

with open('file.xml', 'w') as fd:
  buf.seek(0)
  shutil.copyfileobj(buf, fd)

or shutil.copyfileobj(buf, fd, -1) to copy from a file object without using chunks of limited size (used to avoid uncontrolled memory consumption).

Upvotes: 98

Demitri
Demitri

Reputation: 14039

Python 3:

from io import StringIO
...
with open('file.xml', mode='w') as f:
    print(buf.getvalue(), file=f)

Python 2.x:

from StringIO import StringIO
...
with open('file.xml', mode='w') as f:
    f.write(buf.getvalue())

Upvotes: 19

Related Questions