暾j'`UdZddlmZddlZddlZddlmZddlmZm Z m Z e rddl m Z m Z ejeZiZded<d Z d Z dd Zdd Zdd ZddZddZddZddZddZddZeddf ddZedeedeedeedeedey)z=Implements the compression layer of the `smart_open` library.) annotationsN)Path)IO TYPE_CHECKINGAny)CompressionKwargs Compressorzdict[str, Compressor]_COMPRESSOR_REGISTRYdisableinfer_from_extensionc.ttgtS)z}Return the list of supported compression types available to open. See compression paratemeter to smart_open.open(). )NO_COMPRESSIONINFER_FROM_EXTENSIONget_supported_extensions`/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/smart_open/compression.pyget_supported_compression_typesr s 0 N3K3M NNrc<ttjS)zLReturn the list of file extensions for which we have registered compressors.)sortedr keysrrrrr(s &++- ..rc|r|ddk(sd|}t||j}|tvrtj d||t|<y)aRegister a callback for transparently decompressing files with a specific extension. Args: ext: The extension. Must include the leading period, e.g. `.gz`. callback: The callback. It must accept two positional arguments, file_obj and mode, and is recommended to also accept **kwargs so that whatever the caller passes via smart_open.open(..., compression_kwargs={...}) reaches the underlying library unchanged. Callbacks with the legacy (file_obj, mode) signature still work, but will raise TypeError if the caller supplies compression_kwargs that the callback doesn't declare. Raises: ValueError: If `ext` does not start with a period. Example: Instruct smart_open to use the `lzma` module whenever opening a file with a .xz extension (see README.md for the complete example showing I/O): >>> def _handle_xz(file_obj, mode, **kwargs): ... import lzma ... ... return lzma.open(filename=file_obj, mode=mode, **kwargs) >>> >>> register_compressor(".xz", _handle_xz) This is just an example: `lzma` is in the standard library and is registered by default. r.z*ext must be a string starting with ., not z.overriding existing compression handler for %rN) ValueErrorlowerr loggerwarning)extcallbackmsgs rregister_compressorr!-sU8 CFcM:3'Bo ))+C ""GM (rc|}d|vrd|vrtj|}|Sd|vrd|vrtj|}|S)Nbwr)ioBufferedWriterBufferedReader)file_objmoderesults r_maybe_wrap_bufferedr,RsO F d{sd{""6* M ""6* Mrc Lddl}|jd||d|}t||SNr)filenamer*r)bz2openr,)r)r*kwargsr0r+s r _handle_bz2r3\s, SXX =xd =f =F  --rc Lddl}|jd||d|}t||Sr.)gzipr1r,)r)r*r2r5r+s r _handle_gzipr6c, TYY >t >v >F  --rc ddl}|jdk\rddlm}nddlm}|j }||fd|i|}t ||S)Nr))zstdr*)sys version_info compressionr; backportsr1r,)r)r*r2r<r; zstd_openr+s r _handle_zstdrAjsC 7"$"YYI x 5d 5f 5F  --rc Lddl}|jd||d|}t||Sr.)lzmar1r,)r)r*r2rCr+s r _handle_xzrDwr7rc `ddl}|jj|fd|i|}t||S)Nrr*) lz4.frameframer1r,)r)r*r2lz4r+s r _handle_lz4rI~s0 SYY^^H :4 :6 :F  --rc|tk(r|S|tk(r4 |xs |jj}t|j}|tvr!|jdrd|}t| t|}|||fi|xsiS#tt f$rt jd||cYSwxYw#t$r|cYSwxYw)a,Wrap `file_obj` with an appropriate [de]compression mechanism based on its file extension. If the filename extension isn't recognized, simply return the original `file_obj` unchanged. `file_obj` must either be a filehandle object, or a class which behaves like one. If `filename` is specified, it will be used to extract the extension. If not, the `file_obj.name` attribute is used as the filename. If `compression_kwargs` is specified, its contents are forwarded as keyword arguments to the registered compressor callback. zRunable to transparently decompress %r because it seems to lack a string-like .name+z1transparent (de)compression unsupported for mode )rrnamerAttributeError TypeErrorrrrsuffixr endswithrKeyError)r)r*r>r/compression_kwargs inferred_namer rs rcompression_wrapperrTs&n$** %6==?M =)00 **t}}S/AA$Jo' 4 Hd A'9'?R AA! *  NNdfn O   s#B 3 B7 'B43B47 CCz.bz2z.gzz.zstz.xzz.lz4)returnz list[str])rstrrr rUNone)r)rr*rVrU IO[bytes])r)rXr*rVr2rrUIO[Any]) r)rYr*rVr>rVr/z str | NonerRzCompressionKwargs | NonerUrY)__doc__ __future__rr&loggingpathlibrtypingrrrsmart_open._typingrr getLogger__name__rr __annotations__rrrrr!r,r3r6rArDrIrTrrrrcsD" ))@   8 $.0+04- O/ ")J.. ...,37 (B(B (B(B (B 1 (B  (B\FK(E<(FL)E:&FK(r