\jdZddlZddlZddlZej eZiZdZ dZ dZ dZ dZ dZd Zd Zd Zd Zd Ze dfdZe dee dee dee dey)z=Implements the compression layer of the `smart_open` library.Ndisableinfer_from_extensionc0ttgtzS)z}Return the list of supported compression types available to open. See compression paratemeter to smart_open.open(). )NO_COMPRESSIONINFER_FROM_EXTENSIONget_supported_extensions@/root/env/lib/python3.12/site-packages/smart_open/compression.pyget_supported_compression_typesr s 0 14L4N NNr c<ttjS)zLReturn the list of file extensions for which we have registered compressors.)sorted_COMPRESSOR_REGISTRYkeysr r r rr"s &++- ..r c|r|ddk(std|z|j}|tvrtj d||t|<y)alRegister a callback for transparently decompressing files with a specific extension. Parameters ---------- ext: str The extension. Must include the leading period, e.g. `.gz`. callback: callable The callback. It must accept two position arguments, file_obj and mode. This function will be called when `smart_open` is opening a file with the specified extension. Examples -------- Instruct smart_open to use the `lzma` module whenever opening a file with a .xz extension (see README.rst for the complete example showing I/O): >>> def _handle_xz(file_obj, mode): ... import lzma ... return lzma.LZMAFile(filename=file_obj, mode=mode) >>> >>> 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 %rz.overriding existing compression handler for %rN) ValueErrorlowerrloggerwarning)extcallbacks r register_compressorr'sP6 CFcMG#MNN ))+C ""GM (r c:|jfd}||_y)aEnsure that closing the `outer` stream closes the `inner` stream as well. Deprecated: `smart_open.open().__exit__` now always calls `__exit__` on the underlying filestream. Use this when your compression library's `close` method does not automatically close the underlying filestream. See https://github.com/piskvorky/smart_open/issues/630 for an explanation why that is a problem for smart_open. c rdc}|jyy#rdc}|jwwxYw)Nclose)argsfpinner outer_closes r close_bothztweak_close..close_bothWsF  M % r u % r s#=Nr)outerr r"r!s ` @r tweak_closer$Js++KEKr c|}d|vrd|vrtj|}|Sd|vrd|vrtj|}|S)Nbwr)ioBufferedWriterBufferedReader)file_objmoderesults r _maybe_wrap_bufferedr/csO F d{sd{""6* M ""6* Mr cHddl}|j||}t||SNr)filenamer-)bz2openr/)r,r-r3r.s r _handle_bz2r5ms$ XXxdX 3F  --r cHddl}|j||}t||Sr1)gzipr4r/)r,r-r7r.s r _handle_gzipr8s$ YYtY 4F  --r cddl}|jdk\rddlm}nddlm}|j ||}t ||S)Nr))zstd)r-)sys version_info compressionr= backportsr4r/)r,r-r>r=r.s r _handle_zstdrBys8 7"$" YYxdY +F  --r cHddl}|j||}t||Sr1)lzmar4r/)r,r-rDr.s r _handle_xzrEr9r c|tk(r|S|tk(rA |xs |jj}tjj|\}}|tvr|jdrtd|z t|}|||S#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. zRunable to transparently decompress %r because it seems to lack a string-like .name+z3transparent (de)compression unsupported for mode %r)rrnamerAttributeError TypeErrorrrospathsplitextrendswithrKeyError)r,r-r@r2_rs r compression_wrapperrQsn$ , ,  1HMM88:H))(3;**t}}S/ANQUUVV(' 4$''! *  NN45= O   s#B> B='B:9B:= C  C z.bz2z.gzz.zstz.xz)__doc__r)loggingos.pathrK getLogger__name__rrrrr rrr$r/r5r8rBrErQr r r rWsD    8 $4- O/ )F2. . .. 5ISW!(NFK(E<(FL)E:&r