a mNX=@sddlZddlZddlmZmZGdddeZeZGdddejZ e ej Z e j Z e jZe jZdd Ze ejejdd fd d Zd dZddZddZddZddZGdddeZGdddeZdS)N)ffilibc@seZdZdZdS)Errorz Raised whenever an error is encountered with compressing or decompressing data using brotlipy. .. versionadded:: 0.5.1 N)__name__ __module__ __qualname____doc__r r ,lib/python3.9/site-packages/brotli/brotli.pyrsrc@s"eZdZdZejZejZej Z dS)BrotliEncoderModezP Compression modes for the Brotli encoder. .. versionadded:: 0.5.0 N) rrrr rZBROTLI_MODE_GENERICGENERICZBROTLI_MODE_TEXTTEXTZBROTLI_MODE_FONTFONTr r r r r sr cCst}||}||S)z{ Decompress a complete Brotli-compressed string. :param data: A bytestring containing Brotli-compressed data. ) Decompressor decompressfinish)datadr r r rRs rcCsPt|||||d}||tj}t|jtjks6Jt|jtjksLJ|S)a Compress a string using Brotli. .. versionchanged:: 0.5.0 Added ``mode``, ``quality``, `lgwin``, ``lgblock``, and ``dictionary`` parameters. :param data: A bytestring containing the data to compress. :type data: ``bytes`` :param mode: The encoder mode. :type mode: :class:`BrotliEncoderMode` or ``int`` :param quality: Controls the compression-speed vs compression-density tradeoffs. The higher the quality, the slower the compression. The range of this value is 0 to 11. :type quality: ``int`` :param lgwin: The base-2 logarithm of the sliding window size. The range of this value is 10 to 24. :type lgwin: ``int`` :param lgblock: The base-2 logarithm of the maximum input block size. The range of this value is 16 to 24. If set to 0, the value will be set based on ``quality``. :type lgblock: ``int`` :param dictionary: A pre-set dictionary for LZ77. Please use this with caution: if a dictionary is used for compression, the same dictionary **must** be used for decompression! :type dictionary: ``bytes`` :returns: The compressed bytestring. :rtype: ``bytes`` )modequalitylgwinlgblock dictionary) Compressor _compressrBROTLI_OPERATION_FINISHBrotliEncoderIsFinished_encoder BROTLI_TRUEBrotliEncoderHasMoreOutput BROTLI_FALSE)rrrrrrZ compressorZcompressed_datar r r compress^s.r#cCs0z t|}Wnty*td|Yn0dS)z* Validate that the mode is valid. z%s is not a valid encoder modeN)r ValueErrorrvalr r r _validate_modes  r'cCs&d|krdks"ntd|dS)z5 Validate that the quality setting is valid. r z3%d is not a valid quality, must be between 0 and 11Nrr%r r r _validate_qualitysr*cCs&d|krdks"ntd|dS)z3 Validate that the lgwin setting is valid. z2%d is not a valid lgwin, must be between 10 and 24Nr)r%r r r _validate_lgwinsr-cCs.|dkr*d|krdks*ntd|dS)z5 Validate that the lgblock setting is valid. rr,z@%d is not a valid lgblock, must be either 0 or between 16 and 24Nr)r%r r r _validate_lgblocks r/cCst|||}|tjkr"t|nD|tjkr6t|n0|tjkrJt|n|tjkr^t |nt d|tj krt d||fdS)z This helper function sets a specific Brotli encoder parameter, checking the return code and raising :class:`Error ` if it is invalid. zUnexpected parameter!zError setting parameter %s: %dN) rZBrotliEncoderSetParameterBROTLI_PARAM_MODEr'BROTLI_PARAM_QUALITYr*BROTLI_PARAM_LGWINr-BROTLI_PARAM_LGBLOCKr/ RuntimeErrorr r)encoderZ parameterZparameter_namer&rcr r r _set_parameters          r7c@sPeZdZdZdZdZeejej ddfddZ ddZ d d Z d d Z d dZdS)ra An object that allows for streaming compression of data using the Brotli compression algorithm. .. versionadded:: 0.5.0 :param mode: The encoder mode. :type mode: :class:`BrotliEncoderMode` or ``int`` :param quality: Controls the compression-speed vs compression-density tradeoffs. The higher the quality, the slower the compression. The range of this value is 0 to 11. :type quality: ``int`` :param lgwin: The base-2 logarithm of the sliding window size. The range of this value is 10 to 24. :type lgwin: ``int`` :param lgblock: The base-2 logarithm of the maximum input block size. The range of this value is 16 to 24. If set to 0, the value will be set based on ``quality``. :type lgblock: ``int`` :param dictionary: A pre-set dictionary for LZ77. Please use this with caution: if a dictionary is used for compression, the same dictionary **must** be used for decompression! :type dictionary: ``bytes`` NrrcCsttjtjtj}|s tdt|tj}t|tjd|t|tj d|t|tj d|t|tj d||rt d||_ t||_t||j|j ||_dS)Nz"Unable to allocate Brotli encoder!rrrr uint8_t [])rZBrotliEncoderCreateInstancerNULLr4gcZBrotliEncoderDestroyInstancer7r0r1r2r3new _dictionarylen_dictionary_sizeZ BrotliEncoderSetCustomDictionaryr)selfrrrrrencr r r __init__s"   zCompressor.__init__c Csttt|t|d?d}td}||d<td|d}td|}tdt|}td|}td|} t|j||| ||tj } | tj krt d|drJ||d} t || ddS) z This private method compresses some data in a given mode. This is used because almost all of the code uses the exact same setup. It wouldn't have to, but it doesn't hurt at all. i(size_t *rr8 uint8_t **z#Error encountered compressing data.N) intmathZceilr=rr;rZBrotliEncoderCompressStreamrr9r rbuffer) r?rZ operationZoriginal_output_size available_outZ output_bufferZptr_to_output_bufferZ input_sizeZ input_bufferZptr_to_input_bufferr6Zsize_of_outputr r r rs0       zCompressor._compresscCs||tjS)a> Incrementally compress more data. :param data: A bytestring containing data to compress. :returns: A bytestring containing some compressed data. May return the empty bytestring if not enough data has been inserted into the compressor to create the output yet. )rrZBROTLI_OPERATION_PROCESS)r?rr r r r#?s zCompressor.compresscCsJg}||dtjt|jtjkr@||dtjqd|S)z Flush the compressor. This will emit the remaining output data, but will not destroy the compressor. It can be used, for example, to ensure that given chunks of content will decompress immediately. r)appendrrZBROTLI_OPERATION_FLUSHr!rr joinr?chunksr r r flushJs zCompressor.flushcCs6g}t|jtjkr,||dtjqd|S)z Finish the compressor. This will emit the remaining output data and transition the compressor to a completed state. The compressor cannot be used again after this point, and must be replaced. r)rrrr"rIrrrJrKr r r rXszCompressor.finish)rrrr r<r> DEFAULT_MODErBROTLI_DEFAULT_QUALITYBROTLI_DEFAULT_WINDOWrArr#rMrr r r r rs $ rc@s:eZdZdZdZdZd ddZddZdd Zd d Z dS) ra An object that allows for streaming decompression of Brotli-compressed data. .. versionchanged:: 0.5.0 Added ``dictionary`` parameter. :param dictionary: A pre-set dictionary for LZ77. Please use this with caution: if a dictionary is used for compression, the same dictionary **must** be used for decompression! :type dictionary: ``bytes`` NrcCsXttjtjtj}t|tj|_|rTtd||_t ||_ t |j|j |jdS)Nr8) rZBrotliDecoderCreateInstancerr9r:ZBrotliDecoderDestroyInstance_decoderr;r<r=r>Z BrotliDecoderSetCustomDictionary)r?rZdecr r r rAus zDecompressor.__init__cCsg}tdt|}td|}td|}dt|}td|}td|}td|} t|j|||| tj} | tjkrt|j} t | } t dt | t |||ddd} | | | tjkr|ddksJq q,| tjkrq q,| tjks,Jq,d|S) z Decompress part of a complete Brotli-compressed string. :param data: A bytestring containing Brotli-compressed data. :returns: A bytestring containing the decompressed data. rCz uint8_t[]rDzDecompression error: %srNr)rr;r=rZBrotliDecoderDecompressStreamrQr9ZBROTLI_DECODER_RESULT_ERRORZBrotliDecoderGetErrorCodeZBrotliDecoderErrorStringrstringrGrIZ&BROTLI_DECODER_RESULT_NEEDS_MORE_INPUTZBROTLI_DECODER_RESULT_SUCCESSZ'BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUTrJ)r?rrLZ available_inZ in_bufferZnext_inZ buffer_sizerHZ out_bufferZnext_outr6Z error_codeZ error_messagechunkr r r rs<             zDecompressor.decompresscCsdS)a? Complete the decompression, return whatever data is remaining to be decompressed. .. deprecated:: 0.4.0 This method is no longer required, as decompress() will now decompress eagerly. :returns: A bytestring containing the remaining decompressed data. rr r?r r r rMs zDecompressor.flushcCs4t|jtjksJt|jtjkr0tddS)ao Finish the decompressor. As the decompressor decompresses eagerly, this will never actually emit any data. However, it will potentially throw errors if a truncated or damaged data stream has been used. Note that, once this method is called, the decompressor is no longer safe for further use and must be thrown away. z2Decompression error: incomplete compressed stream.r)rZBrotliDecoderHasMoreOutputrQr"ZBrotliDecoderIsFinishedrrUr r r rs zDecompressor.finish)r) rrrr r<r>rArrMrr r r r res  3r)rFenumZ_brotlirr ExceptionrerrorIntEnumr ZBROTLI_DEFAULT_MODErNr Z MODE_GENERICrZ MODE_TEXTrZ MODE_FONTrrOrPr#r'r*r-r/r7objectrrr r r r s0     =