lcB(dZddlZddlZddlZddlZ ddlZn #e$rdZ YnwxYwddl m Z m Z ddl Z dZdZejeZddiZ d Zd Zdd ZGd dejZGddeZdS)z3Implements file-like objects for reading from http.NT) bytebuffer constantsi)httphttpszAccept-Encodingidentityctj|}|jtvsJ|j|jz}d|dz}t|j|S)N/)schemeuri_path) urllibparseurlsplitr SCHEMESnetlocpathlstripdict) uri_as_string split_urir s /lib/python3.11/site-packages/smart_open/http.py parse_urir&sf %%m44I  w &&&&).0HX__S)))H y'( ; ; ;;chtjt|}t||fi|SN) smart_openutils check_kwargsopen)urimodetransport_paramskwargss ropen_urir#/s3   * *41A B BF T $ $V $ $$rFc |tjkr^t|||||||}tjt j|j|_ |Std|z)aOImplement streamed reader from a web site. Supports Kerberos and Basic HTTP authentication. Parameters ---------- url: str The URL to open. mode: str The mode to open using. kerberos: boolean, optional If True, will attempt to use the local Kerberos credentials user: str, optional The username for authenticating over HTTP password: str, optional The password for authenticating over HTTP headers: dict, optional Any headers to send in the request. If ``None``, the default headers are sent: ``{'Accept-Encoding': 'identity'}``. To use no headers at all, set this variable to an empty dict, ``{}``. Note ---- If neither kerberos or (user, password) are set, will connect unauthenticated, unless set separately in headers. )kerberosuserpasswordheaderstimeoutz(http support for mode %r not implemented) r READ_BINARYSeekableBufferedInputBaseosrbasenamer r urlparsenameNotImplementedError)rr r%r&r'r(r)fobjs rrr4s~8 y$$U( '7   G$$V\%:%:3%?%?%DEE  !"Lt"STTTrcReZdZdedddddfdZdZdZdZdZd d Z d d Z d Z dS)BufferedInputBaserFNc |rddl} | } n ||||f} nd} ||_||_|t|_n||_||_tj || d|j|j|_ |j j s|j |j |j|_tj||_d|_d|_dS)NrTauthstreamr(r))requests_kerberosHTTPKerberosAuth buffer_sizer _HEADERScopyr(r)requestsgetresponseokraise_for_status iter_content _read_iterr ByteBuffer _read_buffer _current_posraw) selfurlr r;r%r&r'r(r)r9r7s r__init__zBufferedInputBase.__init__\s   $ $ $ $$5577DD  ( (#DDD&  ##==??DLL"DL  LL     } - M * * , , ,-44T5EFF&1+>> rcVtdd|_d|_dS)zFlush and close this stream.z close: calledN)loggerdebugr@rDrIs rclosezBufferedInputBase.closes& _%%% rcdS)z+Return True if the stream can be read from.TrOs rreadablezBufferedInputBase.readablestrcdS)NFrRrOs rseekablezBufferedInputBase.seekablesurctjz Unsupported.ioUnsupportedOperationrOs rdetachzBufferedInputBase.detach %%rc"td||jdS|dkrdS|dkr8t|jdkr |jj}n|dkr9|j|jjz}nt|j|krtd|j||j|j }|dkr8|j}|xjt|z c_|St|j|k|j|}|xjt|z c_|S)z> Mimics the read call to a filehandle object. zreading with size: %dNrrz:http reading more content at current_pos: %d with size: %d) rMrNr@lenrFrHreadrGfillrD)rIsizeretval bytes_reads rr`zBufferedInputBase.reads  ,d333 = 3 19 23 AX 2#d/00A5 2]&++--FF AX 2&++-- 0A0F0F0H0HHFFd'((4/ " P%t".33DODD ?"!.3355F%%V4%%!Md'((4/ "&++D11F S[[( rc.||S)zThis is the same as read().)rb)r`rIrbs rread1zBufferedInputBase.read1syydy###rc|t|}|sdS||dt|<t|S)zLRead up to len(b) bytes into b, and return the number of bytes read.rN)r`r_)rIbdatas rreadintozBufferedInputBase.readintosGyyQ   1*3t99* 4yyr)r]) __name__ __module__ __qualname__DEFAULT_BUFFER_SIZErKrPrSrUr[r`rgrkrRrrr3r3[s!$2EdT4QU&&&&V  &&&!!!!F$$$$rr3cLeZdZdZdedddddfdZd dZdZd Zd d Z d d Z dS)r+zr Implement seekable streamed reader from a web site. Supports Kerberos and Basic HTTP authentication. r4FNc :||_|rddl} | |_n| | ||f|_nd|_|t|_n||_||_||_||_ | |_ |j j s|j td|j |j jt#|j jdd|_|j jdddk|_|j |j|_t1j||_d|_d|_dS) z If Kerberos is True, will attempt to use the local Kerberos credentials. Otherwise, will try to use "basic" HTTP authentication via username/password. If none of those are set, will connect unauthenticated. rNzself.response: %r, raw: %rzContent-Lengthr]z Accept-Rangesnone)rJr9r:r7r<r=r(r)r;r _partial_requestr@rArBrMrNrHintr?content_lengthlower _seekablerCrDrrErFrG) rIrJr r;r%r&r'r(r)r9s rrKz"SeekableBufferedInputBase.__init__sx   $ $ $ $)::<> rrctd|||tjvrt dtjz|st d|tjkr|}n5|tjkr |j |z}n|tj kr |j |z}|j dkr"tj |d}n&tj ||j }|j |kr|j Std|j |||_ ||j kr(d|_d|_|jnh||}|jrE||_|j|j|_|jnd|_|j S)zSeek to the specified position. :param int offset: The offset in bytes. :param int whence: Where the offset is from. Returns the position after seeking.z seeking to offset: %r whence: %rz"invalid whence, expected one of %rzstream is not seekabler]N)maxvalz0http seeking from current_pos: %d to new_pos: %d)rMrNrWHENCE_CHOICES ValueErrorrUOSError WHENCE_STARTWHENCE_CURRENTrG WHENCE_ENDrvrrclampr@rDrFemptyrtrArCr;)rIoffsetwhencenew_posr@s rseekzSeekableBufferedInputBase.seeks  7HHH 1 1 ^AID\\]] ]}} 4233 3 Y+ + 3GG y/ / 3'&0GG y+ + 3)F2G  " $ R &,,WT,BBGG &,,WT=P,QQG   ' %$ $ GIZ\cddd# d) ) % DM"DO   # # % % % %,,W55H{ % ( "&-"<"r?rJr7r))rI start_posr@s rrtz*SeekableBufferedInputBase._partial_request@sk  Z L  **:*L*LY*W*W X Y Y Y< HLL    r)rr) rlrmrn__doc__rorKrrrUrrtrRrrr+r+s "%2EdT4QU2222h.!.!.!.!`!!!&&&&      rr+)FNNNN)rrYloggingos.pathr, urllib.parser r> ImportError MISSING_DEPSrrrsmart_open.utilsror getLoggerrlrMr<rr#rBufferedIOBaser3r+rRrrrsh:9 OOOOLLL-,,,,,,,    8 $ $ z *<<<%%% $U$U$U$UNnnnnn)nnnb 1s ##