From 8525a815c11af8f87f2beef7987650738dbc620c Mon Sep 17 00:00:00 2001 From: Helmut Grohne Date: Thu, 16 Apr 2015 17:43:48 +0200 Subject: use binary stdin on py3k --- importpkg.py | 8 ++++++-- 1 file 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__": -- cgit v1.2.3