X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/e61df48303291900581ef4a64dcf97864598c5f4..701150f33acb6c07d35b3802cc02964321d0417c:/sdk/python/arvados/stream.py diff --git a/sdk/python/arvados/stream.py b/sdk/python/arvados/stream.py index 59558162b4..37cd5d7db8 100644 --- a/sdk/python/arvados/stream.py +++ b/sdk/python/arvados/stream.py @@ -1,5 +1,10 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 + from __future__ import print_function from __future__ import absolute_import +from future.utils import listvalues from builtins import object import collections import hashlib @@ -12,14 +17,16 @@ import copy from ._ranges import locators_and_ranges, Range from .arvfile import StreamFileReader from arvados.retry import retry_method -from .keep import * +from arvados.keep import * from . import config from . import errors +from . import util from ._normalize_stream import normalize_stream class StreamReader(object): + @util._deprecated('3.0', 'arvados.collection.Collecttion') def __init__(self, tokens, keep=None, debug=False, _empty=False, - num_retries=0): + num_retries=10): self._stream_name = None self._data_locators = [] self._files = collections.OrderedDict() @@ -63,7 +70,7 @@ class StreamReader(object): return self._files def all_files(self): - return list(self._files.values()) + return listvalues(self._files) def size(self): n = self._data_locators[-1] @@ -80,13 +87,13 @@ class StreamReader(object): def readfrom(self, start, size, num_retries=None): """Read up to 'size' bytes from the stream, starting at 'start'""" if size == 0: - return '' + return b'' if self._keep is None: self._keep = KeepClient(num_retries=self.num_retries) data = [] for lr in locators_and_ranges(self._data_locators, start, size): data.append(self._keepget(lr.locator, num_retries=num_retries)[lr.segment_offset:lr.segment_offset+lr.segment_size]) - return ''.join(data) + return b''.join(data) def manifest_text(self, strip=False): manifest_text = [self.name().replace(' ', '\\040')] @@ -98,5 +105,5 @@ class StreamReader(object): manifest_text.extend([d.locator for d in self._data_locators]) manifest_text.extend([' '.join(["{}:{}:{}".format(seg.locator, seg.range_size, f.name.replace(' ', '\\040')) for seg in f.segments]) - for f in list(self._files.values())]) + for f in listvalues(self._files)]) return ' '.join(manifest_text) + '\n'