X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/03188ad6eb14ee3dcd6bdf74198624c9358936c5..fcf3ca5baf89bdd944e3a7dcdc1b65f8ff4945ca:/sdk/python/arvados/commands/get.py diff --git a/sdk/python/arvados/commands/get.py b/sdk/python/arvados/commands/get.py index e54fd87a19..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 @@ -87,7 +90,7 @@ skipped. """) group.add_argument('--strip-manifest', action='store_true', default=False, help=""" -When getting a collection manifest, strip its access tokens before writing +When getting a collection manifest, strip its access tokens before writing it. """) @@ -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())):