o i-@s UddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZejdkr7ddlmZnddlmZeeedd fZzzddlZWn eyYddlZYnwd ZWn eyhd ZYnwzejd krvdd lmZndd lmZd ZWn eyd ZYnwdZdZdZdZ Gddde Z!Gddde Z"Gddde Z#Gddde d dZ$GdddZ%e%eZ&e%e'd<de#ddfdd Z(  d/d!e e)d"e*de+fd#d$Z,Gd%d&d&eZ-Gd'd(d(Z.Gd)d*d*e-Z/Gd+d,d,e-Z0Gd-d.d.e-Z1dS)0N)ABCabstractmethod)Executor)AnyFinalOptionalProtocol TypedDictcast) )Buffer)Unionzmemoryview[int]zmemoryview[bytes]TF)r )ZstdDecompressoriic@s2eZdZdedefddZd dedefddZd S) ZLibCompressObjProtocoldatareturncCdSNselfrrrS/home/kim/smarthome/.venv/lib/python3.10/site-packages/aiohttp/compression_utils.pycompress-z ZLibCompressObjProtocol.compress.modecCrrrrrrrrflush.rzZLibCompressObjProtocol.flushN.)__name__ __module__ __qualname__r bytesrintrrrrrr,src@sJeZdZd dededefddZd dedefdd Zede fd d Z d S)ZLibDecompressObjProtocol.r max_lengthrcCrrrrrr'rrr decompress2rz$ZLibDecompressObjProtocol.decompresslengthcCrrrrr*rrrr3rzZLibDecompressObjProtocol.flushcCrrrrrrreof5zZLibDecompressObjProtocol.eofNr ) r!r"r#r r%r$r)rpropertyboolr-rrrrr&1s r&c@seZdZUeed<eed<eed<eed<eed<      ddeded ed ed ed eed efddZ dd ed ed e fddZ dded eded e fddZ dd ededed e fddZ dS)ZLibBackendProtocol MAX_WBITS Z_FULL_FLUSH Z_SYNC_FLUSH Z_BEST_SPEEDZ_FINISH.levelmethodwbitsmemLevelstrategyzdictrcCrrr)rr7r8r9r:r;r<rrr compressobj@szZLibBackendProtocol.compressobjcCrrr)rr9r<rrr decompressobjIz!ZLibBackendProtocol.decompressobjrcCrrr)rrr7r9rrrrMr?zZLibBackendProtocol.compressbufsizecCrrr)rrr9r@rrrr)Pr?zZLibBackendProtocol.decompressN)......)..)r!r"r#r%__annotations__rr rr=r&r>r$rr)rrrrr19sn     r1c@s&eZdZUeed<eed<eed<dS)CompressObjArgsr9r;r7N)r!r"r#r%rArrrrrBUs  rB)totalc@seZdZdefddZedefddZedefddZ edefd d Z edefd d Z edefd dZ edefddZ dededefddZdededefddZdedededefddZdedededefddZdedefddZdS) ZLibBackendWrapper _zlib_backendcCs ||_dSr)rE)rrErrr__init__\ zZLibBackendWrapper.__init__rcCst|jddS)Nr!Z undefinedgetattrrEr,rrrname_szZLibBackendWrapper.namecC|jjSr)rEr2r,rrrr2czZLibBackendWrapper.MAX_WBITScCrKr)rEr3r,rrrr3grLzZLibBackendWrapper.Z_FULL_FLUSHcCrKr)rEr4r,rrrr4krLzZLibBackendWrapper.Z_SYNC_FLUSHcCrKr)rEr5r,rrrr5orLzZLibBackendWrapper.Z_BEST_SPEEDcCrKr)rEr6r,rrrr6srLzZLibBackendWrapper.Z_FINISHargskwargscO|jj|i|Sr)rEr=rrMrNrrrr=wzZLibBackendWrapper.compressobjcOrOr)rEr>rPrrrr>zrQz ZLibBackendWrapper.decompressobjrcO|jj|g|Ri|Sr)rErrrrMrNrrrr}zZLibBackendWrapper.compresscOrRr)rEr)rSrrrr)rTzZLibBackendWrapper.decompressattrnamecCs t|j|SrrH)rrUrrr __getattr__ zZLibBackendWrapper.__getattr__N)r!r"r#r1rFr/strrJr%r2r3r4r5r6rrr=r&r>r r$rr)rVrrrrrD[s&rD ZLibBackendnew_zlib_backendrcCs |t_dSr)rYrE)rZrrrset_zlib_backendrGr[encodingsuppress_deflate_headercCs$|dkr dtjS|rtj StjS)Ngzip)rYr2)r\r]rrrencoding_to_modes r`c@sdeZdZdefdeedeefddZee fde dede fd d Z e fde dede fd d Z dS) DecompressionBaseHandlerNexecutormax_sync_chunk_sizecCs||_||_dS)z&Base class for decompression handlers.N) _executor_max_sync_chunk_sizerrbrcrrrrFs z!DecompressionBaseHandler.__init__rr'rcCdSDecompress the given data.Nrr(rrrdecompress_syncrz(DecompressionBaseHandler.decompress_synccsD|jdurt||jkrt|j|j||IdHS|||Srh)relenasyncioZget_event_looprun_in_executorrdrjr(rrrr)s   z#DecompressionBaseHandler.decompress) r!r"r#MAX_SYNC_CHUNK_SIZErrr%rFrZLIB_MAX_LENGTH_UNLIMITEDr$rjr)rrrrras2   rac@seZdZddddddefdeededeedeedeedeed eefd d Z d e d e fddZ d e d e fddZ ddeed e fddZ dS)ZLibCompressorNFr\r]r7r9r;rbrcc Csv||_||_|durt||n||_ttj|_i}|j|d<|dur'||d<|dur/||d<|jjdi||_dS)Nr9r;r7r) rdrer`_moderDrYrEr= _compressor) rr\r]r7r9r;rbrcrNrrrrFs    zZLibCompressor.__init__rrcCs |j|Sr)rrrrrrr compress_syncrWzZLibCompressor.compress_synccsF|jduo t||jk}|rt|j|jj|IdHS||S)aCompress the data and returned the compressed bytes. Note that flush() must be called after the last call to compress() If the data size is large than the max_sync_chunk_size, the compression will be done in the executor. Otherwise, the compression will be done in the event loop. **WARNING: This method is NOT cancellation-safe when used with flush().** If this operation is cancelled, the compressor state may be corrupted. The connection MUST be closed after cancellation to avoid data corruption in subsequent compress operations. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N) rerkrlZget_running_looprmrdrrrrs)rrZshould_use_executorrrrrs    zZLibCompressor.compressrcCs|j|dur |S|jjS)a Flush the compressor synchronously. **WARNING: This method is NOT cancellation-safe when called after compress().** The flush() operation accesses shared compressor state. If compress() was cancelled, calling flush() may result in corrupted data. The connection MUST be closed after compress() cancellation. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N)rrrrEr6rrrrrs  zZLibCompressor.flushr)r!r"r#rnrrXr0r%rrFr$rsrrrrrrrps4 rpc seZdZdddefdeededeedeeffdd Z e fd e d ed e fd d Z dded e fddZed efddZZS)ZLibDecompressorNFr\r]rbrccs>tj||dt|||_ttj|_|jj|jd|_dS)Nrbrc)r9) superrFr`rqrDrYrEr> _decompressor)rr\r]rbrc __class__rrrFs  zZLibDecompressor.__init__rr'rcCs|j||Sr)rwr)r(rrrrjsz ZLibDecompressor.decompress_syncrr*cCs|dkr |j|S|jS)Nr)rwrr+rrrrs  zZLibDecompressor.flushcCrKr)rwr-r,rrrr-rLzZLibDecompressor.eof)r)r!r"r#rnrrXr0rr%rFror r$rjrr/r- __classcell__rrrxrrts2  rtcs`eZdZdefdeedeeddffdd Zefde dede fd d Z de fd d Z Z S) BrotliDecompressorNrbrcrcs*tstdt|_tj||ddS)z)Decompress data using the Brotli library.zIThe brotli decompression is not available. Please install `Brotli` moduleruN) HAS_BROTLI RuntimeErrorbrotliZ Decompressor_objrvrFrfrxrrrF$s  zBrotliDecompressor.__init__rr'cCs4t|jdrtt|j||Stt|j||S)rir))hasattrrr r$r)processr(rrrrj2s z"BrotliDecompressor.decompress_synccCs t|jdrtt|jSdS)zFlush the decompressor.r)rrr r$rr,rrrr:s zBrotliDecompressor.flush)r!r"r#rnrrr%rFror r$rjrrzrrrxrr{ s& r{cs`eZdZdefdeedeeddffdd Zefde dede fd d Z de fd d Z Z S) ZSTDDecompressorNrbrcrcs(tstdt|_tj||ddS)NzOThe zstd decompression is not available. Please install `backports.zstd` moduleru)HAS_ZSTDr}rrrvrFrfrxrrrFBs zZSTDDecompressor.__init__rr'cCs|tkrtn|}|j||Sr)roZSTD_MAX_LENGTH_UNLIMITEDrr))rrr'Zzstd_max_lengthrrrrjOs z ZSTDDecompressor.decompress_synccCrg)Nrrr,rrrr[r.zZSTDDecompressor.flush) r!r"r#rnrrr%rFror$rjrrzrrrxrrAs&  r)NF)2rlsyszlibabcrrconcurrent.futuresrtypingrrrrr r version_infocollections.abcr rr$ bytearrayZ brotlicffir~ ImportErrorr|Zcompression.zstdrZbackports.zstdrrnZDEFAULT_MAX_DECOMPRESS_SIZErorrr&r1rBrDrYrAr[rXr0r%r`rarprtr{rrrrrsj            -  K!