lc,dZddlZddlZejeZiZdZ dZ dZ dZ dZ dZ d Zd Zd Ze d ee d edS)z?Implements the compression layer of the ``smart_open`` library.Ndisableinfer_from_extensionc<ttgtzS)z}Return the list of supported compression types available to open. See compression paratemeter to smart_open.open(). )NO_COMPRESSIONINFER_FROM_EXTENSIONget_supported_extensions6lib/python3.11/site-packages/smart_open/compression.pyget_supported_compression_typesr s 0 14L4N4N NNr cNttS)zLReturn the list of file extensions for which we have registered compressors.)sorted_COMPRESSOR_REGISTRYkeysr r r rr!s &++-- . ..r c|r |ddkstd|z|tvrtd||t|<dS)a)Register 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, format=lzma.FORMAT_XZ) >>> >>> register_compressor('.xz', _handle_xz) r.z,ext must be a string starting with ., not %rz.overriding existing compression handler for %rN) ValueErrorrloggerwarning)extcallbacks r register_compressorr&sf2 OCFcMOG#MNNN ""NGMMM (r c2|jfd}||_dS)aHEnsure that closing the `outer` stream closes the `inner` stream as well. Use this when your compression library's `close` method does not automatically close the underlying filestream. See https://github.com/RaRe-Technologies/smart_open/issues/630 for an explanation why that is a problem for smart_open. c rdc}|dSdS#rdc}|wwxYw)Nclose)argsfpinner outer_closes r close_bothztweak_close..close_bothPsk  KMMM  % r   u  % r  s ,A Nr)outerrr!r s ` @r tweak_closer#Fs8+KEKKKr cJddlm}|||}t|||S)Nr)BZ2File)bz2r%r#)file_objmoder%results r _handle_bz2r*\s: WXt $ $F!!! Mr c\ddl}|||}t|||S)Nr)fileobjr()gzipGzipFiler#)r'r(r-r)s r _handle_gzipr/cs5KKK ]]8$] 7 7F!!! Mr c|tkr|S|tkr_ |j}n4#ttf$r t d||cYSwxYwtj |\}}|tvr'| drtd|z t|}|||S#t$r|cYSwxYw)a[ This function will wrap the file_obj with an appropriate [de]compression mechanism based on the specified extension. file_obj must either be a filehandle object, or a class which behaves like one. It must have a .name attribute. If the filename extension isn't recognized, will simply return the original file_obj. zRunable to transparently decompress %r because it seems to lack a string-like .name+z3transparent (de)compression unsupported for mode %r)rrnameAttributeError TypeErrorrrospathsplitextrendswithrKeyError)r'r( compressionfilename_rs r compression_wrapperr=jsn$ 4 , , 4 }HH *    NN45=   OOO   ))(33;**Wt}}S/A/AWNQUUVVV(' 4x$''' s!".AA) C CCz.bz2z.gz)__doc__loggingos.pathr5 getLogger__name__rrrrr rrr#r*r/r=r r r rCsFE  8 $ $4- OOO/// )))@,!(!(!(NFK(((E<(((((r