X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2333472a4f517a227278f028bbbc4e72687c0e71..0eb72b526bf8bbb011551ecf019f604e17a534f1:/sdk/python/arvados/commands/get.py diff --git a/sdk/python/arvados/commands/get.py b/sdk/python/arvados/commands/get.py index 3bf929584e..56d471888d 100755 --- a/sdk/python/arvados/commands/get.py +++ b/sdk/python/arvados/commands/get.py @@ -1,4 +1,7 @@ #!/usr/bin/env python +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 import argparse import hashlib @@ -131,7 +134,11 @@ def parse_arguments(arguments, stdout, stderr): def main(arguments=None, stdout=sys.stdout, stderr=sys.stderr): global api_client - + + if stdout is sys.stdout and hasattr(stdout, 'buffer'): + # in Python 3, write to stdout as binary + stdout = stdout.buffer + args = parse_arguments(arguments, stdout, stderr) if api_client is None: api_client = arvados.api('v1') @@ -155,11 +162,11 @@ def main(arguments=None, stdout=sys.stdout, stderr=sys.stderr): open_flags |= os.O_EXCL try: if args.destination == "-": - stdout.write(reader.manifest_text(strip=args.strip_manifest)) + stdout.write(reader.manifest_text(strip=args.strip_manifest).encode()) else: out_fd = os.open(args.destination, open_flags) with os.fdopen(out_fd, 'wb') as out_file: - out_file.write(reader.manifest_text(strip=args.strip_manifest)) + out_file.write(reader.manifest_text(strip=args.strip_manifest).encode()) except (IOError, OSError) as error: logger.error("can't write to '{}': {}".format(args.destination, error)) return 1 @@ -236,7 +243,7 @@ def main(arguments=None, stdout=sys.stdout, stderr=sys.stderr): if args.hash: digestor = hashlib.new(args.hash) try: - with s.open(f.name, 'r') as file_reader: + with s.open(f.name, 'rb') as file_reader: for data in file_reader.readall(): if outfile: outfile.write(data) @@ -271,7 +278,7 @@ def main(arguments=None, stdout=sys.stdout, stderr=sys.stderr): def files_in_collection(c): # Sort first by file type, then alphabetically by file path. - for i in sorted(c.keys(), + for i in sorted(list(c.keys()), key=lambda k: ( isinstance(c[k], arvados.collection.Subcollection), k.upper())):