diff options
Diffstat (limited to 'Lib/sunau.py')
| -rw-r--r-- | Lib/sunau.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/Lib/sunau.py b/Lib/sunau.py index b49a5da..0ba7dc9 100644 --- a/Lib/sunau.py +++ b/Lib/sunau.py @@ -456,12 +456,17 @@ class Au_write: self._datalength = self._datawritten self._file.seek(0, 2) -def open(f, mode): - if mode == 'r': +def open(f, mode=None): + if mode is None: + if hasattr(f, 'mode'): + mode = f.mode + else: + mode = 'rb' + if mode in ('r', 'rb'): return Au_read(f) - elif mode == 'w': + elif mode in ('w', 'wb'): return Au_write(f) else: - raise Error, "mode must be 'r' or 'w'" + raise Error, "mode must be 'r', 'rb', 'w', or 'wb'" openfp = open |