Ej=rUddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z ejdk\rddlmZnddlmZeeedd fZ ddlZd Z ejd k\rdd lmZndd lmZd ZdZdZdZGdde ZGdde Z Gdde Z!Gdde d Z"GddZ#e#eZ$e#e%d<de!ddfdZ& d,de'dzd e(de)fd!Z*Gd"d#eZ+Gd$d%Z,Gd&d'e+Z-Gd(d)e+Z.Gd*d+e+Z/y#e$rddlZYwxYw#e$rd ZYwxYw#e$rd ZYwxYw)-N)ABCabstractmethod)Executor)AnyFinalProtocol TypedDictcast) )Buffer)Unionzmemoryview[int]zmemoryview[bytes]TF)r )ZstdDecompressoric.eZdZdedefdZddedefdZy)ZLibCompressObjProtocoldatareturncyNselfrs W/root/workspace/ytshorts/venv/lib/python3.12/site-packages/aiohttp/compression_utils.pycompressz ZLibCompressObjProtocol.compress,modecyrrrrs rflushzZLibCompressObjProtocol.flush-rrN.)__name__ __module__ __qualname__r bytesrintr"rrrrr+s2V225#555rrcveZdZd dededefdZd dedefdZede fdZ edefdZ edefd Z y ) ZLibDecompressObjProtocolr max_lengthrcyrrrrr+s r decompressz$ZLibDecompressObjProtocol.decompress1rrlengthcyrrrr/s rr"zZLibDecompressObjProtocol.flush2rrcyrrrs reofzZLibDecompressObjProtocol.eof4srcyrrr3s runconsumed_tailz)ZLibDecompressObjProtocol.unconsumed_tail7s(+rcyrrr3s r unused_dataz%ZLibDecompressObjProtocol.unused_data:s$'rNr#) r$r%r&r r(r'r.r"propertyboolr4r6r8rrrr*r*0sZKvK3KK7C7U7 T +++ 'U''rr*ceZdZUeed<eed<eed<eed<eed< ddededed ed ed ed zd efdZ dded ed efdZ ddededed e fdZ ddededed e fdZ y )ZLibBackendProtocol MAX_WBITS Z_FULL_FLUSH Z_SYNC_FLUSH Z_BEST_SPEEDZ_FINISHlevelmethodwbitsmemLevelstrategyzdictNrcyrr)rrBrCrDrErFrGs r compressobjzZLibBackendProtocol.compressobjEs#&rcyrr)rrDrGs r decompressobjz!ZLibBackendProtocol.decompressobjNs$'rrcyrr)rrrBrDs rrzZLibBackendProtocol.compressRrbufsizecyrr)rrrDrNs rr.zZLibBackendProtocol.decompressUrMr)......)..) r$r%r&r(__annotations__r rrIr*rKr'rr.rrrr<r<>sNM"&&& &  &  &}& !&14(('-( "( ?B&)8; AD&):= rr<c,eZdZUeed<eed<eed<y)CompressObjArgsrDrFrBN)r$r%r&r(rPrrrrRrRZs JM JrrR)totalceZdZdefdZedefdZedefdZ edefdZ edefdZ edefdZ edefd Z d ed edefd Zd ed edefd Zded ed edefdZded ed edefdZdedefdZy)ZLibBackendWrapper _zlib_backendc||_yr)rV)rrVs r__init__zZLibBackendWrapper.__init__as 2?rrc0t|jddS)Nr$ undefinedgetattrrVr3s rnamezZLibBackendWrapper.namedst)):{CCrc.|jjSr)rVr=r3s rr=zZLibBackendWrapper.MAX_WBITShs!!+++rc.|jjSr)rVr>r3s rr>zZLibBackendWrapper.Z_FULL_FLUSHl!!...rc.|jjSr)rVr?r3s rr?zZLibBackendWrapper.Z_SYNC_FLUSHpr`rc.|jjSr)rVr@r3s rr@zZLibBackendWrapper.Z_BEST_SPEEDtr`rc.|jjSr)rVrAr3s rrAzZLibBackendWrapper.Z_FINISHxs!!***rargskwargsc:|jj|i|Sr)rVrIrrdres rrIzZLibBackendWrapper.compressobj|s -t!!--t>v>>rc:|jj|i|Sr)rVrKrgs rrKz ZLibBackendWrapper.decompressobjs /t!!//@@@rrcB|jj|g|i|Sr)rVrrrrdres rrzZLibBackendWrapper.compresss%*t!!**4A$A&AArcB|jj|g|i|Sr)rVr.rjs rr.zZLibBackendWrapper.decompresss%,t!!,,TCDCFCCrattrnamec.t|j|Srr[)rrls r __getattr__zZLibBackendWrapper.__getattr__st))844rN)r$r%r&r<rXr9strr]r(r=r>r?r@rArrrIr*rKr r'rr.rnrrrrUrU`sO@&9@DcDD,3,,/c///c///c//+#++???8O?A3A#A:SABVBCB3B5BDvDcDSDUD5C5C5rrU ZLibBackendnew_zlib_backendrc|t_yr)rprV)rqs rset_zlib_backendrss 0Krencodingsuppress_deflate_headercx|dk(rdtjzS|rtj StjS)Ngzip)rpr=)rtrus rencoding_to_moderys96K))))% ? %7{7P7P$Q"$**w  !)F:   #F7O94--99CFCrrrc8|jj|Sr)rrrs r compress_synczZLibCompressor.compress_syncs((..rcK|jduxrt||jkD}|rKtjj |j |j j|d{S|j|S7w)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) rrrget_running_looprrrrr)rrshould_use_executors rrzZLibCompressor.compresss&  % %T 1 6D D555   113CC 0 0 9 94 !!$''sA1B 3B 4B rcp|jj||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. )rr"rVrAr!s rr"zZLibCompressor.flushs>%%$D  *.*<*<*E*E  rr)r$r%r&rror:r(rrXr r'rrr"rrrrrs $(-  #$(*=D*D"&DTz D Tz D * DT/D!4ZD6/&/U/(6(e(8  #*    rrc eZdZdddefdedzdededzdedzffd Ze fde d ed e fd Z dd ed e fd Z ed efdZed efdZxZS)ZLibDecompressorNFrtrur|r}ct|||t|||_t t j |_|j j|j|_d|_ d|_ y)Nr|r}rDF) superrXryrrUrprVrK _decompressor _last_empty_pending_unused_data)rrtrur|r} __class__s rrXzZLibDecompressor.__init__ sj (@ST%h0GH $6{7P7P$Q!//==DJJ=O 26!rrr+rcp|j|jt|z}d|_|jj|jj|z|}|dk(|_|jj r|jjr|jj}|jj|j|_|tk7r|t|z}|dkr||_nX|jj||}|dk(|_||z }|jj r|jjr|jj rN|j|jjkDr+|jj|j|_|S)Nrrr)rr'rr.r6rr4r8rVrKrrrr=)rrr+resultunusedchunks rrz ZLibDecompressor.decompress_syncsx  $ $ 0,,uT{:D(,D %##..    . . 5z "S=   $$););)G)G''33F!%!3!3!A!A !A!SD 66c&k) ?06D-&&11&*EE$|D  eOF  $$););)G)G    ! !djj43E3E3O3O&O!%!3!3!A!A !A!SD  rr/cv|dkDr|jj|S|jjSNr)rr"r1s rr"zZLibDecompressor.flush?s?z    $ $V , ##))+ rc~t|jjxs|j xs|jduSr)r:rr6rrr3s rrzZLibDecompressor.data_availableFsA ##33 4 5### 5((4 rc.|jjSr)rr4r3s rr4zZLibDecompressor.eofNs!!%%%r)r)r$r%r&rror:rr(rXrr r'rr"r9rr4 __classcell__rs@rrr s $(-$(*= 7* 7"& 7T/ 7 !4Z 7/H""(+" "H C     &T&&rrcxeZdZdefdedzdedzddffd Zefdedede fdZ de fd Z e de fd ZxZS) BrotliDecompressorNr|r}rcts tdtj|_d|_t |||y)z)Decompress data using the Brotli library.zIThe brotli decompression is not available. Please install `Brotli` moduleFrN) HAS_BROTLI RuntimeErrorbrotli Decompressor_objrrrXrr|r}rs rrXzBrotliDecompressor.__init__WsG 1 '')   (@STrrr+ct|jdr^|tk(r*tt|jj |}ntt|jj ||}n]|tk(r*tt|jj |}n*tt|jj ||}|dk(|_|S)rr.r)hasattrrrr r'r.processr)rrr+rs rrz"BrotliDecompressor.decompress_syncfs 499l +66eTYY%9%9$%?@eTYY%9%9$ %KL66eTYY%6%6t%<=eTYY%6%6tZ%HI!S= rct|jdr(tt|jj Sy)zFlush the decompressor.r"r)rrr r'r"r3s rr"zBrotliDecompressor.flushxs+ 499g &tyy01 1rcV|jj xr |j Sr)r is_finishedrr3s rrz!BrotliDecompressor.data_available~s&99((**C43C3C/CCrr$r%r&rrr(rXrr r'rr"r9r:rrrs@rrrSs %)*= UT/ U!4Z U  U /H(+ $u DDDrrcxeZdZdefdedzdedzddffd Zefdedede fdZ de fd Z e de fd ZxZS) ZSTDDecompressorNr|r}rcvts tdt|_d|_t |||y)NzOThe zstd decompression is not available. Please install `backports.zstd` moduler)HAS_ZSTDrrrrrrXrs rrXzZSTDDecompressor.__init__sA 9 %& 26! (@STrrr+cp|tk(rtn|}|j|j|z}d|_|jj ||}|jj r|jj r|jj }t|_|tk7r|t|z}|dkr||_nL||jj ||z }|jj r|jj r|jj rt|_|Sr) rZSTD_MAX_LENGTH_UNLIMITEDrrr.r4r8rr)rrr+zstd_max_lengthrr8s rrz ZSTDDecompressor.decompress_syncs66 &   $ $ 0,,t3D(,D %%%dO<iimm 5 5))//K(*DI";;3v;."a'0;D- dii**;H HFiimm 5 5 99==(*DI rcy)Nrrr3s rr"zZSTDDecompressor.flushsrc|jj xr|jj xs|jduSr)r needs_inputr4rr3s rrzZSTDDecompressor.data_availables< %% % ;diimm*;3  & &d 2 3rrrs@rrrs%)*= UT/ U!4Z U  U/H$$(+$ $Lu333rr)NF)0rsyszlibabcrrconcurrent.futuresrtypingrrrr r version_infocollections.abcr rr' bytearray brotlicffir ImportErrorrcompression.zstdrbackports.zstdrrrrrr*r<rRrUrprPrsror:r(ryr{rrrrrrrrs #'88w& 5)%68KK LF#J 7"53H 6h6 ( ((8iu *5*5Z#5T": :1':1t1  $)XDjX!X X Fs FFH H VD&/D&N-D1-D`<3/<3c JHsBDDD, DDDDD)(D),D65D6