Use make_stream_binary to turn stdin into a binary stream.

[third_party/subunit] / python / subunit / _output.py

diff --git a/python/subunit/_output.py b/python/subunit/_output.py
index af9dcada53ff9b24d564034436d16245e6a16ff3..6df64045b04c1362490b84f8cd12a9abf5f5e7c5 100644 (file)
--- a/python/subunit/_output.py
+++ b/python/subunit/_output.py
@@ -22,6 +22,7 @@ from optparse import (
 )
 import sys
 
+from subunit import make_stream_binary
 from subunit.iso8601 import UTC
 from subunit.v2 import StreamResultToBytes
 
@@ -131,10 +132,7 @@ def parse_arguments(args=None, ParserClass=OptionParser):
         if options.attach_file == '-':
             if not options.file_name:
                 options.file_name = 'stdin'
-            if sys.version[0] >= '3':
-                options.attach_file = getattr(sys.stdin, 'buffer', sys.stdin)
-            else:
-                options.attach_file = sys.stdin
+                options.attach_file = make_stream_binary(sys.stdin)
         else:
             try:
                 options.attach_file = open(options.attach_file, 'rb')