fsed/utils.py: fix read/write '-' in py2

wroberts · Dec 19, 2015 · 2ed13e5 · 2ed13e5
1 parent 4aaf374
commit 2ed13e5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fsed/utils.py b/fsed/utils.py
@@ -28,9 +28,11 @@ def open_file(filename, mode='rb'):
         if filename == '-' and 'r' in mode:
             if PY3:
                 return sys.stdin.buffer
+            return sys.stdin
         elif filename == '-' and ('w' in mode or 'a' in mode):
             if PY3:
                 return sys.stdout.buffer
+            return sys.stdout
         if filename.lower().count('.zip:'):
             assert 'r' in mode
             assert filename.count(':') == 1