#HJjG dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z mZmZmZmZmZmZmZddlmZddlmZddlmZddlmZm Z erddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'm(Z(ejRe*Z+ejXd Z-dd Z.dd Z/e/Z0e1Z2e d d!dZ1e d"d d d d d d d d d d#dZ1e d d$dZ1dddddddejfddf d$dZ1d%dZ4 d& d'dZ5d(dZ6 d) d*dZ7GddZ8d+dZ9 ejte1ejve/y#e<$re+j{dYywxYw),z6Implements the majority of smart_open's top-level API.) annotationsN)IO TYPE_CHECKINGAnyBinaryIOLiteralTextIOcastoverload)doctools transport)Callable)Self)CompressionKwargsTransportParamsUriF) do_setlocalectjdk(r d|vrd|z}tjj |j S)z0Returns the scheme of the URL only, as a string.ntz://zfile://)osnameurllibparseurlsplitscheme) uri_as_strings c/Users/ahmed/devFolder/claude-voice/.venv/lib/python3.12/site-packages/smart_open/smart_open_lib.py _sniff_schemer,s;  ww$5 5!M1 <<  / 6 66ct|}tj|}|j|}t j dt |j}|di|S)zParse the given URI from a string. Args: uri_as_string: The URI to parse. Returns: The parsed URI as a ``collections.namedtuple``. smart_open/doctools.py magic goes here r)rr get_transport parse_uri collections namedtuplesortedkeys)rr submoduleas_dictrs rr#r#8sY= )F''/I!!-0G  w||~(> ?C >>r.c yNr! urimode bufferingencodingerrorsnewlineclosefdopener compressioncompression_kwargstransport_paramss ropenr8Wsr)r0r1r2r3r4r5r6r7cyr+r!r,s rr8r8gsrc yr+r!r,s rr8r8xsrrTc tjdtt|ts d} t | |t jvrd|} t| | i} t|||||||} | | S|d|vr|jdd}t|tjr t |}|} | xst} t|}t#||| }t%|d d}t|tr|nt|tr|nd}t j&||||| }d|vs| t)||||| }n|}||k7rCd g}|D];}t+j,t.5t1||t%||ddd=t3d t5j6||S#t$r}t|j d|d}~wwxYw#1swYxYw)a Open the URI object, returning a file-like object. The URI is usually a string in a variety of formats. For a full list of examples, see the :func:`parse_uri` function. The URI may also be one of: - an instance of the pathlib.Path class - a stream (anything that implements io.IOBase-like functionality) Args: uri: The object to open. mode: Mimicks built-in open parameter of the same name. buffering: Mimicks built-in open parameter of the same name. encoding: Mimicks built-in open parameter of the same name. errors: Mimicks built-in open parameter of the same name. newline: Mimicks built-in open parameter of the same name. closefd: Mimicks built-in open parameter of the same name. Ignored. opener: Mimicks built-in open parameter of the same name. Ignored. compression: Explicitly specify the compression/decompression behavior. See ``smart_open.compression.get_supported_compression_types``. compression_kwargs: Keyword arguments forwarded to the registered compressor callback. Examples of each library's max-compression option: ``{'compresslevel': 9}`` for .gz/.bz2, ``{'preset': 9}`` for .xz, ``{'level': 22}`` for .zst, ``{'compression_level': 12}`` for .lz4. Ignored when compression is 'disable' or the URI's extension doesn't match a registered compressor. transport_params: Additional parameters for the transport layer (see notes below). Returns: A file-like object. Raises: TypeError: If ``mode`` is not a string or if the URI type is not recognized. ValueError: If ``compression`` is not a supported value. NotImplementedError: If ``mode`` cannot be parsed into a valid binary mode. Note: smart_open has several implementations for its transport layer (e.g. S3, HTTP). Each transport layer has a different set of keyword arguments for overriding default behavior. If you specify a keyword argument that is *not* supported by the transport layer being used, smart_open will ignore that argument and log a warning message. smart_open/doctools.py magic goes here See Also: - `Standard library reference `__ - `smart_open README.md `__ z%rzmode should be a stringzinvalid compression type: N)r5r/r0r1r2brr)filenamer6)r0r1r2to_boto3IO[Any])loggerdebuglocals isinstancestr TypeErrorso_compressionget_supported_compression_types ValueError_shortcut_openreplacepathlibPathDEFAULT_ENCODING_get_binary_modeNotImplementedErrorargs_open_binary_streamgetattrcompression_wrapper_encoding_wrapper contextlibsuppressAttributeErrorsetattrr so_utils FileLikeProxy)r-r.r/r0r1r2r3r4r5r6r7msgfobjexplicit_encoding binary_modevebinaryrr@ decompresseddecodedpromoted_attrsattrs rr8r8sF LLvx dC 'n.HHJJ*;-8o    D  t ||C$#w||$#h  4$4H6&t, !k3C DF 664 (D!$,tC9M#SWH!55- L $+7#    &$"D$$^4wvt'<=54#  811'6B CCK 6!"''!*-256D54s$ F,*G, G5GGG cBt|gdvrdvr d}t|dDcgc]}j|}}t|dkDr d}t|dfd }dvr |dn'd vr |d nd vr |d n d }t|dvr |dnGdvr2j j dj dnj dd vr|d rd |}t|djScc}w)Ntr>z'can't have text and binary mode at oncerwaz6must have exactly one of create/read/write/append modecdjjj|yr+)appendpopindex)charbinmoder.s rtransferz"_get_binary_mode..transfer?s!txx 4 012rawr;zKMust have exactly one of create/read/write/append mode and at most one plus+zinvalid mode: r?)rprGreturnNone)listrKcountsumrnrormjoin)mode_strr^xcountsrrrqr.s @@rrQrQ,s$ >DG d{sd{7o%* +UdjjmUF + 6{QFo3 d{   [o d{  C!ss d{  xl+o 777 M,sDct|tsyt|}|tjt j fvryt j|}|tjk(r6tj|j} | tjvry|tjk7ryi} ||| d<|jdd}||| d<|r d|vr|| d<t!||fd|i| S)aTry to open the URI using the standard library io.open function. This can be much faster than the alternative of opening in binary mode and then decoding. This is only possible under the following conditions: 1. Opening a local file; and 2. Compression is disabled If it is not possible to use the built-in open for the specified URI, returns None. Args: uri: A string indicating what to open. mode: The mode to pass to the open function. compression: The compression type selected. buffering: Mimicks built-in open parameter of the same name. encoding: Mimicks built-in open parameter of the same name. errors: Mimicks built-in open parameter of the same name. newline: Mimicks built-in open parameter of the same name. Returns: The opened file, or None if no shortcut is possible. Nr0r>r?r2r1r/)rFrGrr NO_SCHEMEso_fileSCHEMEextract_local_pathrIINFER_FROM_EXTENSIONrNrOsuffixget_supported_extensionsNO_COMPRESSIONrM _builtin_open) r-r.r5r/r0r1r2r local_path extension open_kwargss rrLrLcsD c3  3 F i))7>>::++C0Jn999LL,33 ??A A 55 5"$K"* J||C$!( I #T/ & H T NY N+ NNrc6|dvrd|}t|t|trt||dSt|tsd|}t |t |}tj|}|j|||}t|ds||_ |S)a Open an arbitrary URI in the specified binary mode. Not all modes are supported for all protocols. Args: uri: The URI to open. May be a string, or something else. mode: The mode to open with. Must be rb, wb or ab. transport_params: Keyword arguments for the transport layer. Returns: A file-like object with a ``.name`` attribute. Raises: NotImplementedError: If ``mode`` is not a supported binary mode. TypeError: If ``uri`` is not a string or integer file descriptor. )rbzrb+wbzwb+abzab+zunsupported mode: F)r3zdon't know how to handle uri r) rRrFintrrGrHrr r"open_urihasattrr)r-r.r7r^rr(r_s rrTrTs" :: #4(+!#&&#sS$66 c3 -cW5n 3 F''/I   c4)9 :D 4  Krctjdtd|vr||S|t}t j ||||dS)aDecode bytes into text, if necessary. If mode specifies binary access, does nothing, unless the encoding is specified. A non-null encoding implies text mode. Args: fileobj: Must quack like a filehandle object. mode: The mode which was originally requested by the user. encoding: The text encoding to use. If mode is binary, overrides mode. errors: The method to use when handling encoding/decoding errors. newline: Forwarded to the text wrapper. Returns: A file object. zencoding_wrapper: %rr>T)r0r1r2 write_through)rCrDrErPr\ TextIOWrapper)fileobjr.r0r1r2s rrWrWsU, LL'2 d{x'#  ! !  rc(eZdZdZddZddZddZy) patch_pathlibz+Replace `Path.open` with `smart_open.open`.c,tt|_yr+)_patch_pathlibr8old_implselfs r__init__zpatch_pathlib.__init__s&t, rc|Sr+r!rs r __enter__zpatch_pathlib.__enter__ s rc.t|jyr+)rr)rexc_typeexc_valexc_tbs r__exit__zpatch_pathlib.__exit__ st}}%rN)rvrw)rvr)robjectrrrrrvrw)__name__ __module__ __qualname____doc__rrrr!rrrrs5-&rrcdtjj}|tj_|S)z Replace `Path.open` with `func`.)rNrOr8)funcrs rrrs"||  HGLL OrzEncountered a non-fatal error while building docstrings (see below). help(smart_open) will provide incomplete information as a result. For full help text, see .)rrGrvrG)rrGrvztuple[Any, ...]) ..........)r-rr.zELiteral['r', 'w', 'a', 'x', 'r+', 'w+', 'a+', 'rt', 'wt', 'at', 'xt']r/rr0 str | Noner1rr2rr3boolr4 Callable[[str, int], int] | Noner5rGr6CompressionKwargs | Noner7TransportParams | Nonervr ).)r-rr.zFLiteral['rb', 'wb', 'ab', 'xb', 'rb+', 'wb+', 'ab+', 'br', 'bw', 'ba']r/rr0rwr1rr2rr3rr4rr5rGr6rr7rrvr)r-rr.rGr/rr0rr1rr2rr3rr4rr5rGr6rr7rrvrB)r|rGrvrG)r<NNN)r-rr.rGr5rGr/rr0rr1rr2rrvzIO[Any] | None)r-rr.rGr7rrvz IO[bytes])NNN) rrBr.rGr0rr1rr2rrvrB)rCallable[..., Any]rvr)>r __future__rr$rXlocaleloggingros.pathrN urllib.parsertypingrrrrrr r r smart_open.compressionr5rIsmart_open.local_file local_filersmart_open.utilsutilsr\ smart_openr r collections.abcrtyping_extensionsrsmart_open._typingrrr getLoggerrrCgetpreferredencodingrPrr# _parse_urir8rrrQrLrTrWrrtweak_open_docstringtweak_parse_uri_docstring Exception exceptionr!rrrsS="  TTT/ (#*(&JJ   8 $.6..EB 74   SV/236/2    O         -  1 -        /236/2    P         -  1 -      /236/2             -  1 -     "/3%::37/3aD aD aDaD aD  aD  aDaD -aDaD1aD-aD aDH4v>O >O >O>O >O  >O  >O>O>OB-f  - - --  -  -  -` & & !H!!$'&H&&y1  Js$F99GG