o 2.a@s\ddlZddlZddlmZeZddZddZdd Zdd d Z dd dfddZ dS)N)StreamParseErrorccs,|D]}t|ts|dd}|VqdS)z Given a stream of bytes or text, if any of the items in the stream are bytes convert them to text. This function can be removed once we return text streams instead of byte streams. zutf-8replaceN) isinstancestrdecode)streamdatar :/usr/lib/python3/dist-packages/docker/utils/json_stream.pystream_as_text s   r cCsR|}zt|\}}|tjj||d}||fWSty(YdSw)zAttempt to parse a json object from a buffer. If there is at least one object, return it and the rest of the buffer, otherwise return None. N) strip json_decoderZ raw_decodejsondecoder WHITESPACEmatchend ValueError)bufferobjindexrestr r r json_splitters  rcCst|ttjS)zGiven a stream of text, return a stream of json objects. This handles streams which are inconsistently buffered (some entries may be newline delimited, and others are not). ) split_bufferrrr)rr r r json_stream$sr cCs:|t|}|dkr dS|d|d||ddfS)N)findr)rZ separatorrr r r line_splitter,s r cCs|S)Nr )ar r r 3sr"c cs|pt}d}t|D]}||7} ||}|durn|\}}|Vqq |r?z||VWdSty>}zt|d}~wwdS)aGiven a generator which yields strings and a splitter function, joins all input, splits on the separator and yields each chunk. Unlike string.split(), each chunk includes the trailing separator, except for the last one if none was found on the end of the input. TN)r r Exceptionr)rZsplitterrZbufferedr Z buffer_splititemer r r r3s* r)r) rZ json.decodererrorsrZ JSONDecoderrr rrr rr r r r s