use binary stdin on py3k

1 files changed, 6 insertions, 2 deletions

diff --git a/importpkg.py b/importpkg.py
index 6dced21..99e9850 100755
--- a/importpkg.py
+++ b/importpkg.py
@@ -110,10 +110,14 @@ def main():
                       help="verify that stdin hash given sha256 hash")
     options, args = parser.parse_args()
     hash_functions = [sha512_nontrivial, gziphash, pnghash, gifhash]
+    try:
+        stdin = sys.stdin.buffer
+    except AttributeError: # python2
+        stdin = sys.stdin
     if options.hash:
-        gen = process_package_with_hash(sys.stdin, hash_functions, options.hash)
+        gen = process_package_with_hash(stdin, hash_functions, options.hash)
     else:
-        gen = process_package(sys.stdin, hash_functions)
+        gen = process_package(stdin, hash_functions)
     yaml.safe_dump_all(gen, sys.stdout)
 
 if __name__ == "__main__":