jdZddlmZddlmZddlmZddlZddlmZGddejZ y) z.Utilities for representing a part of a stream.)absolute_import)division)unicode_literalsN) upload_streamc`eZdZdZ dfd ZfdZdfd Zejffd Z xZ S)ComponentStreamaNImplements a subset of the io.IOBase API exposing part of a stream. This class behaves as a contiguous subset of the underlying stream. This is helpful for composite uploads since even when total_size is specified, apitools.transfer.Upload looks at the size of the source file to ensure that all bytes have been uploaded. ct|||||||_|j|jz|_|j j |jy)aInitializes a ComponentStream instance. Args: stream (io.IOBase): See super class. offset (int|None): The position (in bytes) in the wrapped stream that corresponds to the first byte of the ComponentStream. length (int|None): The total number of bytes readable from the ComponentStream. digesters (dict[util.HashAlgorithm, hashlib hash object]|None): See super class. progress_callback (func[int]|None): See super class. )streamlength digestersprogress_callbackN)super__init__ _start_byte_length _end_byte_streamseek)selfr offsetr r r __class__s :lib/googlecloudsdk/command_lib/storage/component_stream.pyrzComponentStream.__init__$s_$ )T#+ $- D%% 4DN LLd&&'c:t||jz S)z?Returns the current position relative to the part's start byte.)rtellr)rrs rrzComponentStream.tellBs D & (4+;+; ;;rc|dkr |j}t||jt|z }t|t d|S)z?Reads `size` bytes from a stream, or all bytes when `size` < 0.r)rminrrrreadmax)rsizers rrzComponentStream.readFsJ ax \\d tT^^eIt&A&CC DD D &s1d| 44rc|tjk(r||jz}n4|tjk(rt||z}n||j z}t|||j z S)aGoes to a specific point in the stream. Args: offset (int): The number of bytes to move. whence: Specifies the position offset is added to. os.SEEK_END: offset is added to the last byte in the FilePart. os.SEEK_CUR: offset is added to the current position. os.SEEK_SET: offset is added to the first byte in the FilePart. Returns: The new relative position in the stream (int). )osSEEK_ENDrSEEK_CURrrrr)rrwhencenew_absolute_indexrs rrzComponentStream.seekMsp!DNN2 2;;  D686A!D$4$44 D &'9 :T=M=M MMr)NN)) __name__ __module__ __qualname____doc__rrrr"SEEK_SETr __classcell__)rs@rrrs4!% (<<5!# NNrr) r+ __future__rrrr""googlecloudsdk.command_lib.storager UploadStreamrrrr2s/5&' <GNm00GNr