Open wave files in binary mode.

Accept 'rb' and 'wb' as well as 'r' and 'w' as the mode parameter to open().

diff --git a/Lib/wave.py b/Lib/wave.py
index bba4e87dcb5a9196c0bc0c9035e58ac621287899..365bbf9c77ea7776bef0bba090e4ba9be6d03d0a 100644 (file)
--- a/Lib/wave.py
+++ b/Lib/wave.py
@@ -230,7 +230,7 @@ class Wave_read:
 
        def __init__(self, f):
                if type(f) == type(''):
-                       f = __builtin__.open(f, 'r')
+                       f = __builtin__.open(f, 'rb')
                # else, assume it is an open file object already
                self.initfp(f)
 
@@ -368,7 +368,7 @@ class Wave_write:
 
        def __init__(self, f):
                if type(f) == type(''):
-                       f = __builtin__.open(f, 'w')
+                       f = __builtin__.open(f, 'wb')
                self.initfp(f)
 
        def initfp(self, file):
@@ -549,11 +549,11 @@ class Wave_write:
                self._datalength = self._datawritten
 
 def open(f, mode):
-       if mode == 'r':
+       if mode in ('r', 'rb'):
                return Wave_read(f)
-       elif mode == 'w':
+       elif mode in ('w', 'wb'):
                return Wave_write(f)
        else:
-               raise Error, "mode must be 'r' or 'w'"
+               raise Error, "mode must be 'r', 'rb', 'w', or 'wb'"
 
 openfp = open # B/W compatibility