diff options
| author | Georg Brandl <georg@python.org> | 2010-10-30 08:29:28 +0000 |
|---|---|---|
| committer | Georg Brandl <georg@python.org> | 2010-10-30 08:29:28 +0000 |
| commit | 31ca1e43367dd11a61de23ec1d4bd36f5889f098 (patch) | |
| tree | f28991ad27356b0c0beb9037bc05a6f951a0444a /Lib/wave.py | |
| parent | 35f786a30a4ceeb1c905e98dbe3b763be604f418 (diff) | |
| download | cpython-31ca1e43367dd11a61de23ec1d4bd36f5889f098.tar.gz | |
#10198: fix duplicate header when writeframes() is called with an empty string.
Diffstat (limited to 'Lib/wave.py')
| -rw-r--r-- | Lib/wave.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/Lib/wave.py b/Lib/wave.py index 57f9d17163..b5bb105d61 100644 --- a/Lib/wave.py +++ b/Lib/wave.py @@ -319,6 +319,7 @@ class Wave_write: self._nframeswritten = 0 self._datawritten = 0 self._datalength = 0 + self._headerwritten = False def __del__(self): self.close() @@ -449,7 +450,7 @@ class Wave_write: # def _ensure_header_written(self, datasize): - if not self._datawritten: + if not self._headerwritten: if not self._nchannels: raise Error('# channels not specified') if not self._sampwidth: @@ -459,6 +460,7 @@ class Wave_write: self._write_header(datasize) def _write_header(self, initlength): + assert not self._headerwritten self._file.write(b'RIFF') if not self._nframes: self._nframes = initlength // (self._nchannels * self._sampwidth) @@ -472,8 +474,10 @@ class Wave_write: self._sampwidth * 8, 'data')) self._data_length_pos = self._file.tell() self._file.write(struct.pack('<l', self._datalength)) + self._headerwritten = True def _patchheader(self): + assert self._headerwritten if self._datawritten == self._datalength: return curpos = self._file.tell() |