暾j$dZddlmZddlZddlZddlZddlZddlm Z m Z m Z ddl Z e r ddl mZddlmZej"eZgdZdZdd Zdd Zddd Zddd Zdd ZddZGddej8ZGdde j:Zy)z(Helper functions for documentation, etc.) annotationsN)IO TYPE_CHECKINGAny)Callable) TracebackType)s3s3ns3agcsgsz0///smart_open.utils.QUESTION_MARK_PLACEHOLDER///ctj|}|jjDcic]9\}}|jtj j k7r ||j;c}}Scc}}w)zRReturn a ``{name: default}`` mapping for every default-valued kwarg of `kallable`.)inspect signature parametersitemsdefault Parameterempty)kallablernameparams Z/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/smart_open/utils.pyinspect_kwargsrsh!!(+I%//557  D% ==G--33 3 emm  s>A5ctt|}t|Dcgc] }||vs| }}|jDcic] \}}||vs ||}}}|rtj d||Scc}wcc}}w)a^Check which keyword arguments the callable supports. Args: kallable: A function or method to test. kwargs: The keyword arguments to check. If the callable doesn't support any of these, a warning message will get printed. Returns: A dictionary of argument names and values supported by the callable. z*ignoring unsupported keyword arguments: %r)sortedrrloggerwarning)rkwargssupported_keywordskunsupported_keywordsvsupported_kwargss r check_kwargsr%(s x 89'-f~U!BT9TAUU+1<<>U!QQBT=T1UUCEYZ  VUs A4A4 A9A9c6| t||}t||S)zClamp a numeric value to a specific range. Args: value: The value to clamp. minval: The lower bound. maxval: The upper bound. Returns: The clamped value. It will be in the range ``[minval, maxval]``. )minmax)valueminvalmaxvals rclampr,=s#E6" uf cp|| d}t||dn t|}|dn t|}d|d|S)apCreate a byte range specifier in accordance with RFC-2616. Args: start: The start of the byte range. If unspecified, stop indicated offset from EOF. stop: The end of the byte range. If unspecified, indicates EOF. Returns: A byte range specifier. Raises: ValueError: If neither ``start`` nor ``stop`` are specified. z7make_range_string requires either a stop or start valuezbytes=-) ValueErrorstr)startstopmsg start_strstop_strs rmake_range_stringr8MsM  }GomUI\rs4yH I;az **r-c|jdd\}}|jdd\}}|jdd\}}|t|t|t|fS)aExtract units, start, stop, and length from a content range header like "bytes 0-846981/846982". Assumes a properly formatted content-range header from S3. See werkzeug.http.parse_content_range_header for a more robust version. Args: content_range: The content-range header to parse. Returns: A tuple ``(units, start, stop, length)`` of one string and three integers from the content-range header.  /r0)splitint) content_rangeunitsnumbersrangelengthr3r4s rparse_content_rangerDes]#((a0NE7MM#q)ME6++c1%KE4 #e*c$iV 44r-ctjj|d}d}|jtvrEd|vrAt |vr9t }|j d|}tjj|d}||S|jj |d}tjj|j|j|ddS)aThis is a hack to prevent the regular urlsplit from splitting around question marks. A question mark (?) in a URL typically indicates the start of a querystring, and the standard library's urlparse function handles the querystring separately. Unfortunately, question marks can also appear _inside_ the actual URL for some schemas like S3, GS. Replaces question marks with a special placeholder substring prior to splitting. This work-around behavior is disabled in the unlikely event the placeholder is already part of the URL. If this affects you, consider changing the value of QUESTION_MARK_PLACEHOLDER to something more suitable. See Also: - https://bugs.python.org/issue43882 - https://github.com/python/cpython/blob/3.14/Lib/urllib/parse.py - https://github.com/piskvorky/smart_open/issues/285 - https://github.com/piskvorky/smart_open/issues/458 - ``smart_open/utils.py:QUESTION_MARK_PLACEHOLDER`` F)allow_fragmentsN?r/) urllibparseurlsplitschemeWORKAROUND_SCHEMESQUESTION_MARK_PLACEHOLDERreplacepath SplitResultnetloc)urlsr placeholderrOs r safe_urlsplitrUxs(   sE  :BK yy&&3#::S[^:^ 0 kk#{+ \\ " "3 " > 77??; ,D << # #BIIryy$B GGr-c(eZdZdZ ddZy) TextIOWrapperzI`io.TextIOWrapper` subclass that does not close the buffer on exceptions.c*||jyy)atCall close on underlying buffer only when there was no exception. Without this patch, TextIOWrapper would call self.buffer.close() during exception handling, which is unwanted for e.g. s3 and azure. They only call self.close() when there was no exception (self.terminate() otherwise) to avoid committing unfinished/failed uploads. N)close)selfexc_typeexc_valexc_tbs r__exit__zTextIOWrapper.__exit__s   JJL r-N)r[type[BaseException] | Noner\BaseException | Noner]TracebackType | NonereturnNone)__name__ __module__ __qualname____doc__r^r-rrWrWs0S,&%   r-rWcheZdZUdZdZded<d fd Zd dZ d fd Zd dZ d d Z xZ S) FileLikeProxyzHWrap an `outer` file-like object so that closing it also closes `inner`..r_FileLikeProxy__innerc2t||||_y)N)super__init__rk)rZouterinner __class__s rrnzFileLikeProxy.__init__s  r-c6|jjS)zAThis explicit proxy method is only required for pylance ref #916.) __wrapped__ __enter__rZs rrtzFileLikeProxy.__enter__s))++r-c t|||||jj|||S#|jj|||wxYw)zExit inner after exiting outer.)rmr^rk)rZr[ exc_value tracebackrqs rr^zFileLikeProxy.__exit__sG B7#HiC LL ! !(Iy ADLL ! !(Iy As 1Ac6|jjS)z3Delegate iteration to the wrapped file-like object.)rs__next__rus rrzzFileLikeProxy.__next__s((**r-c |jj|j|jk7r|jjSS#|j|jk7r|jjwwxYw)z3Close both the wrapped object and the inner object.)rsrYrkrus rrYzFileLikeProxy.closesh %##))+||t/// ""$0t||t/// ""$0s A6B)roIO[Any]rpr|rbrc)rbr)r[r_rwr`rxrarbr)rbrc) rdrerfrgrk__annotations__rnrtr^rzrY __classcell__)rqs@rrjrjsTRGS, B, B( B( B B+%r-rj)rCallable[..., Any]rbdict[str, Any])rrrrrbr)rN)r)r>r*r>r+ int | Nonerbr>)NN)r3rr4rrbr2)r?r2rbztuple[str, int, int, int])rRr2rbzurllib.parse.SplitResult)rg __future__rriologging urllib.parserHtypingrrrwraptcollections.abcrtypesr getLoggerrdrrLrMrr%r,r8rDrUrW ObjectProxyrjrhr-rrs/" )) (#   8 $6N*  +05&%HPB$$(#%E%%#%r-