From 5275f25d14d833ce2a25c0b97607713439bdc3db Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Tue, 20 Jan 2015 10:19:45 +0200
Subject: [PATCH] cli: simplify stdin/stdout logic

---
 amodem-cli | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/amodem-cli b/amodem-cli
index 8f1887d..b4dcd6b 100755
--- a/amodem-cli
+++ b/amodem-cli
@@ -7,12 +7,10 @@ import zlib
 import logging
 import argparse
 
-if sys.version_info.major == 2:
-    _stdin = sys.stdin
-    _stdout = sys.stdout
-else:
-    _stdin = sys.stdin.buffer
-    _stdout = sys.stdout.buffer
+# Python 3 has `buffer` attribute for byte-based I/O
+_stdin = getattr(sys.stdin, 'buffer', sys.stdin)
+_stdout = getattr(sys.stdout, 'buffer', sys.stdout)
+
 
 try:
     import argcomplete