o im&@sdZddlZddlZddlZddlZzddlZWn ey#dZ Ynwddl m Z m Z ddl Z dZdZeeZddiZ d d Zd d Zd ddddddefddZGdddejZGdddeZdS)z3Implements file-like objects for reading from http.NT) bytebuffer constantsi)httphttpszAccept-EncodingidentitycCsBtj|}|jtvs J|j|j}d|d}t|j|dS)N/)schemeuri_path) urllibparseurlsplitrSCHEMESnetlocpathlstripdict) uri_as_string split_urir rC/home/ubuntu/.local/lib/python3.10/site-packages/smart_open/http.py parse_uri&s  rcCs tjt|}t||fi|SN) smart_openutils check_kwargsopen)urimodetransport_paramskwargsrrropen_uri/sr Fc CsN|tjkr!t||| |||||||d } tjtj|j| _ | St d|)uImplement 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 cert: str/tuple, optional if String, path to ssl client cert file (.pem). If Tuple, (‘cert’, ‘key’) 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, ``{}``. session: object, optional The requests Session object to use with http get requests. Can be used for OAuth2 clients. buffer_size: int, optional The buffer size to use when performing I/O. Note ---- If neither kerberos or (user, password) are set, will connect unauthenticated, unless set separately in headers. ) buffer_sizekerberosuserpasswordcertheaderssessiontimeoutz(http support for mode %r not implemented) r READ_BINARYSeekableBufferedInputBaseosrbasenamer r urlparsenameNotImplementedError) rrr"r#r$r%r&r(r'r!fobjrrrr4s $ rc @steZdZdZdedddddddf ddZddZedd Zd d Z d d Z ddZ dddZ dddZ ddZdS)BufferedInputBaseNrFc Cs||_||_| p t|_|rddl} | |_n|dur%|dur%||f|_nd|_||_||_|dur8t |_ n||_ | |_ |jj |j|j|jd|j |j d|_|jjsY|jt||_d|_dS)NrT)authr%streamr&r()urlr%requestsr'requests_kerberosHTTPKerberosAuthr3r!r_HEADERScopyr&r(getresponseokraise_for_statusr ByteBuffer _read_buffer _current_pos) selfr5rr!r"r#r$r%r&r'r(r7rrr__init__gs6       zBufferedInputBase.__init__cCs$td|jsd|_d|_dSdS)zFlush and close this stream.z close: calledN)loggerdebugclosedr<r@rBrrrcloses  zBufferedInputBase.closecCs |jduSr)r<rGrrrrFs zBufferedInputBase.closedcCdS)z+Return True if the stream can be read from.TrrGrrrreadableszBufferedInputBase.readablecCrI)NFrrGrrrseekableszBufferedInputBase.seekablecCtjz Unsupported.ioUnsupportedOperationrGrrrdetachzBufferedInputBase.detachcCs|dkr td|td||js|dkrdS|dkr6t|jr/|j|jj}n&|jj}nt|j|krO|j |jjdkrHnt|j|ks=|j|}|j t|7_ |S)z> Mimics the read call to a filehandle object. rSzsize must be >= -1, got zreading with size: %dr) ValueErrorrDrErFlenr@readr<rawfillrA)rBsizeretvalrrrrWs    zBufferedInputBase.readcCs |j|dS)zThis is the same as read().)rZ)rWrBrZrrrread1s zBufferedInputBase.read1cCs.|t|}|s dS||dt|<t|S)zLRead up to len(b) bytes into b, and return the number of bytes read.rN)rWrV)rBbdatarrrreadintos zBufferedInputBase.readinto)rS)__name__ __module__ __qualname__r<DEFAULT_BUFFER_SIZErCrHpropertyrFrJrKrQrWr]r`rrrrr1ds ,    r1c sbeZdZdZdedddddddf fdd Zddd Zd d Zd d ZdddZ dddZ Z S)r*z Implement seekable streamed reader from a web site. Supports Kerberos, client certificate and Basic HTTP authentication. r2FNc sRt||||||||| | t|jjdd|_|jjdddk|_dS)a. If Kerberos is True, will attempt to use the local Kerberos credentials. If cert is set, will try to use a client certificate Otherwise, will try to use "basic" HTTP authentication via username/password. If none of those are set, will connect unauthenticated. zContent-LengthrSz Accept-RangesnoneN) superrCintr<r&r;content_lengthlower _seekable) rBr5rr!r"r#r$r%r&r'r( __class__rrrCs z"SeekableBufferedInputBase.__init__rcCstd|||tjvrtdtj|std|tjkr#|}n|tjkr.|j |}n |tj kr8|j |}|j dkrFt j j|dd}n t j j||j d}|j |krW|j Std|j |||_ ||j krrd|_|j|j S||}|jr||_|j|j Sd|_|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 seekablerSN)maxvalz0http seeking from current_pos: %d to new_pos: %d)rDrErWHENCE_CHOICESrUrKOSError WHENCE_STARTWHENCE_CURRENTrA WHENCE_ENDrjrrclampr<r@empty_partial_requestr=)rBoffsetwhencenew_posr<rrrseeks:            zSeekableBufferedInputBase.seekcC|jSr)rArGrrrtellzSeekableBufferedInputBase.tellcOr|r)rl)rBargsrrrrrKr~z"SeekableBufferedInputBase.seekablecCrLrMrNr\rrrtruncate"rRz"SeekableBufferedInputBase.truncatecCsF|j}|durtj||d<|jj|j|jd|j ||j d}|S)NrangeT)r3r4r%r&r() r&r:rrmake_range_stringr'r;r5r3r%r()rB start_posr&r<rrrrw&s z*SeekableBufferedInputBase._partial_request)rr) rarbrc__doc__rdrCr{r}rKrrw __classcell__rrrmrr*s . r*)rrOloggingos.pathr+ urllib.parser r6 ImportError MISSING_DEPSrrrsmart_open.utilsrdr getLoggerrarDr9rr rBufferedIOBaser1r*rrrrs0     0p