o i@sdZddlZddlZddlZddlZddlZeeZ gdZ dZ ddZ ddZ dd d Zdd d Zd dZddZGdddejZGdddejZdS)z(Helper functions for documentation, etc.N)s3s3ns3us3agsz0///smart_open.utils.QUESTION_MARK_PLACEHOLDER///c Cszt|}Wn8ty?z t|\}}}}Wn ty&iYYSw|s-iYS|t| d}tt||YSwdd|j DS)NcSs&i|]\}}|jtjjkr||jqS)defaultinspect Parameterempty).0nameparamrrD/home/ubuntu/.local/lib/python3.10/site-packages/smart_open/utils.py /s z"inspect_kwargs..) r signatureAttributeError getargspec TypeErrorlendictzip parametersitems)kallablerargsvarargskeywordsdefaultssupported_keywordsrrrinspect_kwargss    r csLtt|fddt|D}fdd|D}|r$td||S)aCheck which keyword arguments the callable supports. Parameters ---------- kallable: callable A function or method to test kwargs: dict The keyword arguments to check. If the callable doesn't support any of these, a warning message will get printed. Returns ------- dict A dictionary of argument names and values supported by the callable. csg|]}|vr|qSrr)r krrr Gsz check_kwargs..csi|] \}}|vr||qSrr)r r!vr"rrrHsz check_kwargs..z*ignoring unsupported keyword arguments: %r)sortedr rloggerwarning)rkwargsunsupported_keywordssupported_kwargsrr"r check_kwargs6s  r+cCs |dur t||}t||}|S)aIClamp a numeric value to a specific range. Parameters ---------- value: numeric The value to clamp. minval: numeric The lower bound. maxval: numeric The upper bound. Returns ------- numeric The clamped value. It will be in the range ``[minval, maxval]``. N)minmax)valueminvalmaxvalrrrclampPs  r1cCsL|dur |dur td|durdnt|}|durdnt|}d||fS)akCreate a byte range specifier in accordance with RFC-2616. Parameters ---------- start: int, optional The start of the byte range. If unspecified, stop indicated offset from EOF. stop: int, optional The end of the byte range. If unspecified, indicates EOF. Returns ------- str A byte range specifier. Nz7make_range_string requires either a stop or start valuez bytes=%s-%s) ValueErrorstr)startstop start_strstop_strrrrmake_range_stringjs  r9cCsH|dd\}}|dd\}}|dd\}}|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. Parameters ---------- content_range: str The content-range header to parse. Returns ------- tuple (units: str, start: int, stop: int, length: int) The units and three integers from the content-range header.  /-)splitint) content_rangeunitsnumbersrangelengthr5r6rrrparse_content_rangesrEcCstjj|dd}d}|jtvr'd|vr't|vr't}|d|}tjj|dd}|dur-|S|j|d}tj|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?r2) urllibparseurlsplitschemeWORKAROUND_SCHEMESQUESTION_MARK_PLACEHOLDERreplacepath SplitResultnetloc)urlsr placeholderrOrrr safe_urlsplits rUc@seZdZddZdS) TextIOWrappercCs|dur |dSdS)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_tbrrr__exit__s zTextIOWrapper.__exit__N)__name__ __module__ __qualname__r\rrrrrVs rVcsDeZdZdZfddZddZfddZdd Zd d ZZ S) FileLikeProxy.cst|||_dSN)super__init___FileLikeProxy__inner)rXouterinner __class__rrrcs  zFileLikeProxy.__init__cC |jS)zAThis explicit proxy method is only required for pylance ref #916.) __wrapped__ __enter__rXrrrrks zFileLikeProxy.__enter__c s<ztj|i|W|jj|i|S|jj|i|w)zExit inner after exiting outer.)rbr\rd)rXrr(rgrrr\s(zFileLikeProxy.__exit__cCrira)rj__next__rlrrrrms zFileLikeProxy.__next__cCs@z|jW|j|jkr|jSS|j|jkr|jwwra)rjrWrdrlrrrrWs    zFileLikeProxy.close) r]r^r_rdrcrkr\rmrW __classcell__rrrgrr`s  r`)rN)NN)__doc__r iologging urllib.parserHwrapt getLoggerr]r&rLrMr r+r1r9rErUrV ObjectProxyr`rrrrs"   )