Issue #9957: SpooledTemporaryFile.truncate() now accepts an optional size parameter, as other file-like objects.

Patch by Ryan Kelly.
This commit is contained in:
Antoine Pitrou 2011-11-25 18:03:09 +01:00
parent e333d00d3a
commit 0e86a5842d
3 changed files with 31 additions and 3 deletions

View file

@ -578,8 +578,13 @@ class SpooledTemporaryFile:
def tell(self):
return self._file.tell()
def truncate(self):
self._file.truncate()
def truncate(self, size=None):
if size is None:
self._file.truncate()
else:
if size > self._max_size:
self.rollover()
self._file.truncate(size)
def write(self, s):
file = self._file