Hj0iddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZddlZddlZddlmZmZddlmZmZmZmZmZdd lmZd3d ZeZej d Z!ej dZ"e r GddZ#Gdde$Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0ej1d4dZ2d5d6d$Z3d7d'Z4ej1d(Z5d)Z6ej1d*Z7ej1d+Z8ej1d,Z9d8d.Z:Gd/d0Z;Gd1d2Z>(/:: =??sHTTP/\d\.\d [0-9]{3} (.*)sHTTP/(\d\.\d) ([0-9]{3}) (.*)cBeZdZUded<ded<ded<ded<ded<dS) CurlWsFrameintageflagsoffset bytesleftlenN)__name__ __module__ __qualname____annotations__r'r%r)r),s=  r'r)c(eZdZdZ d d fd ZxZS) CurlErrorz$Base exception for curl_cffi packagermsgrcodeint | CurlECode | Literal[0]rNonecPtj|g|Ri|||_dSN)super__init__r8)selfr7r8argskwargs __class__s r%r>zCurlError.__init__7s7 .t...v...26 r')r)r7rr8r9rr:)r0r1r2__doc__r> __classcell__)rBs@r%r6r64sM..>?77777777777r'r6ilitype_r*sizectj|}tj||dd}|||dS)z ffi callback for curl debug infoNr)r from_handlebuffer)curlrJdatarKclientpcallbacktexts r%debug_functionrTLsDw''H :dD ! !!!! $D HUD 1r'Fbbytes uppercaseboolcR|rdnddfd|DS)u Convert a bytes object to a space-separated hex string, e.g. "0a ff 3c". If uppercase=True, letters will be A–F instead of a–f. z{:02X}z{:02x} c3BK|]}|VdSr<)format).0bytefmts r% zbytes_to_hex..[s/33CJJt$$333333r')join)rUrWr_s @r% bytes_to_hexrbUs:  -((XC 883333333 3 33r'rPr:cXtdtdtdtdtdt dt di}d}||d}|t t fvrgt|d|}t||krd nd }tj |d t|d ||d dS | d}tj |d||tttfvr!tj d dSdS#t$rit|d|}t||krd nd }tj |d t|d ||d YdSwxYw)N*<>z< DATAz> DATAz< SSLz> SSL(z...z [z bytes]:  zutf-8rZ) CURLINFO_TEXTCURLINFO_HEADER_INCURLINFO_HEADER_OUTCURLINFO_DATA_INCURLINFO_DATA_OUTCURLINFO_SSL_DATA_INCURLINFO_SSL_DATA_OUTrrbr/sysstderrwritedecodeUnicodeDecodeError)rJrPPREFIXESMAX_SHOW_BYTESprefixhex_strpostfixrSs r%debug_function_defaultr{^ssCS(8gwHN \\% % %F %'<===tO^O455D ^33"" FNNc$iiNN'N7NNNOOOOO T;;w''D J  //// 0 0 0],>@STTT   &&&&&UT! T T T"4#899GII77bbUG J  RR#d))RRgRwRRR S S S S S S  TsA.D66A/F)(F)ctj|}|tj||dd||zS)zAffi callback for curl write function, directly writes to a bufferN)r rMrsrN)ptrrKnmembuserdatarNs r%buffer_callbackr}sD_X & &F LLC''*+++ 4<r'c(|sdSt|SNr)r*)ss r% ensure_intrs q q66Mr'ctj|}|tj||dd}t|}|tks |t kr|S|||zkrt jdtd||zS)zHffi callback for curl write function, calls the callback python functionNzWrote bytes != received bytes.rE stacklevel) r rMrNrCURL_WRITEFUNC_PAUSECURL_WRITEFUNC_ERRORwarningswarnr)r}rKr~rrRwrotes r%write_callbackrsx((H HSZU++AAA. / /E u  E $$$1E(E(E    6TUVVVV 4<r'ctj|}||z}||}|dSt|tr|St|t r|}|sdSt||kr#tdt|d|dtj ||t|t|S)zIffi callback for curl read function, reads from a buffer/file-like objectNrRead callback returned  bytes, but only  bytes are allowed.) r rMread isinstancer*rencoder/r6memmove)r}rKr~rrNmax_lenrPs r%read_buffer_callbackrs_X & &FUlG ;;w  D |q$ ${{}} q 4yy7 ^c$ii ^ ^' ^ ^ ^   KT3t99%%% t99r'ctj|}||z}||}|dSt|tr|St|tr|}|sdSt ||kr#tdt |d|dtj||t |t |S)zGffi callback for curl read function, calls the callback python functionNrrrr) r rMrr*rrr/r6r)r}rKr~rrRrrPs r% read_callbackrsx((HUlG 8G  D |q$ ${{}} q 4yy7 ^c$ii ^ ^' ^ ^ ^   KT3t99%%% t99r' list[bytes]cg}|}|r5|tj|j|j}|5t j||S)z%Converts curl slist to a python list.)appendr stringrPnextr curl_slist_free_all)headresultr}s r% slist_to_listrs[ F C  cj**+++h D!!! Mr'ceZdZdZdZd@dAd ZdBd ZdBdZdBdZdCdZ dDdZ dEdZ dFdZ dGdZ dHdId"ZdBd#ZdJdKd&ZdLd'Z dJdKd(ZdMd)ZdBd*ZdNd.ZedOd0ZedPd2ZdBd3ZdQd5ZejfdRd:ZdSdTd?ZdS)UCurlz; Wrapper for ``curl_easy_*`` functions of libcurl. irhFNcacertrdebugrXrr:c`|r|ntj|_tj|_tj|_tj|_|pt|_ d|_ d|_ d|_ d|_ d|_d|_d|_tjdd|_||_|tjd|j|_tjd|_tjd|_tjd|_dS)z Parameters: cacert: CA cert path to use, by default, certs from ``certifi`` are used. debug: whether to show curl debug messages. handle: a curl handle instance from ``curl_easy_init``. FNzchar[]zsize_t *zstruct curl_ws_frame **)r curl_easy_init_curlr NULL_headers_proxy_headers_resolveDEFAULT_CACERT_cacert _is_cert_set _skip_cacert _write_handle_header_handle _debug_handle _body_handle _read_handlenew _error_buffer_debug_set_error_buffer_WS_RECV_BUFFER_SIZE_ws_recv_buffer_ws_recv_n_recv_ws_recv_p_frame_ws_send_n_sent)r?rrhandles r%r>z Curl.__init__s &?VV3+=+?+?  !h / !!"&#'"&!%!% WXs33      #wx1JKK"wz22 #(A B B"wz22r'ctj|jtj|j}|dkrt jdtd|j r| dSdS)NrzFailed to set error bufferrEr) r _curl_easy_setoptrr ERRORBUFFERrrrrrrr?rets r%rzCurl._set_error_bufferse#DJ0CTEWXX !88 M6TU V V V V ;  JJLLLLL  r'c|tjd|tjddS)zSet debug to Truer TN)setoptrVERBOSE DEBUGFUNCTIONr?s r%rz Curl.debugs5 GOQ''' G)400000r'c.|dSr<closers r%__del__z Curl.__del__ r'errcoder*r@rc:|dkrdS|j|g|R}||dSr) _get_error)r?rr@errors r% _check_errorzCurl._check_errors< a<< F/$///  K  r'c |dkrztj|jd}dd|D}t d|d|d|d t t| SdS) NrbackslashreplaceerrorsrZc,g|]}t|Sr4)r)r]as r% z#Curl._get_error..s444!s1vv444r'z Failed to z , curl: (z) zK. See https://curl.se/libcurl/c/libcurl-errors.html first for more details.)r8)r rrrtrar6r r )r?rr@errmsgactions r%rzCurl._get_errors a<<Z 233::BT:UUFXX44t44455FVg)W--   > >,TZ88  DK K K Kr'cd|_d|_|j-tj|j|t j|_dS)z8Reset all curl options, wrapper for ``curl_easy_reset``.FN) rrrr curl_easy_resetrr rrrs r%resetz Curl.reset'sL!! : !   + + +  " " $ $ $ r'headersrrct}|D]X}|dr/||ddY|S)zExtract ``cookies.SimpleCookie`` from header lines. Parameters: headers: list of headers in bytes. Returns: A parsed cookies.SimpleCookie instance. s set-cookie: N)rlowerrloadrt)r?r5cookier s r%parse_cookie_headerszCurl.parse_cookie_headers0sf ,~~ 2 2F||~~((99 2 F233K..00111 r' status_linecht|}|r|dndS)zTExtract reason phrase, like ``OK``, ``Not Found`` from response status line.r r')REASON_PHRASE_REmatchgroup)r<ms r%get_reason_phrasezCurl.get_reason_phrase?s0  " "; / /'qwwqzzzC'r'"tuple[CurlHttpVersion, int, bytes]ct|}|stjddfS|ddkr tj}nX|ddkr tj}n2|ddkr tj}n tj}t|d}|d}|||fS) zfParse status line. Returns: http_version, status_code, and reason phrase rr'r z2.0z1.1z1.0rErF) STATUS_LINE_REr?rV1_0r@V2_0V1_1NONEr*)r<rA http_version status_codereasons r%parse_status_linezCurl.parse_status_lineFs   - - 0"'C/ / 771::  */LL WWQZZ5 */LL WWQZZ5 */LL*/L!''!**oo [&00r'c||jr tj|jd|_t j|j|j"t j|jd|_dSdS)zAClose and cleanup curl handle, wrapper for ``curl_easy_cleanup``.N)r)rr curl_easy_cleanupr releaserrrs r%rz Curl.close]s} &&((( :   !$* - - -DJ D&'''   + K, - - -#'D  , +r'tuple[bytes, CurlWsFrame]c2|jtdtj|j|j|j|j|jx}r||dtj |jd|jd|jdfS)zReceive a frame from a websocket connection. Returns: a tuple of frame content and curl frame meta struct. Raises: CurlError: if failed. Nz/Cannot receive websocket data on closed handle.WS_RECVr) rr6r curl_ws_recvrrrrrr rNrs r%ws_recvz Curl.ws_recvjs : MNN N" J   %   !    3 .   c9 - - - Jt+ , ,-Ft/CA/F-F G  !! $  r'payloadbytes | memoryviewr,CurlWsFlag | intc|jtdtj|jt j|t ||jd|x}r||d|jdS)aSend data to a websocket connection. Args: payload: content to send. flags: websocket flag to set for the frame, default: binary. Returns: The number of bytes sent. Raises: CurlError: if failed. Nz,Cannot send websocket data on closed handle.rWS_SEND) rr6r curl_ws_sendr from_bufferr/rr)r?rVr,rs r%ws_sendz Curl.ws_sends : JKK K" J OG $ $ LL      3 .   c9 - - -#A&&r'r'r8messagecttjd||z}||tjS)aClose a websocket connection. Shorthand for :meth:`ws_send` with close code and message. Note that to completely close the connection, you must close the curl handle after this call with :meth:`close`. Args: code: close code. message: close message. Returns: 0 if no error. Raises: CurlError: if failed. z!H)r,)structpackr]rCLOSE)r?r8r_rVs r%ws_closez Curl.ws_closes3+dD))G3||G:+;|<<rrrrrrrrr!r#r'r,r)r1r4r; staticmethodrBrMrrUrBINARYr]rdr4r'r%rrsD&33333<1111     B+8+8+8+8Z....     7777IIIII20000 AE+++++2    !!!!    (((\( 111\1, ( ( ( (    8FPEV''''':=======r'rcbeZdZdZdddZdddddddZeddZdddZddZ ddZ dS)CurlMimez#Wrapper for the ``curl_mime_`` API.NrO Curl | Nonecz|r|n t|_tj|jj|_dS)z? Args: curl: Curl instance to use. N)rrr curl_mime_init_form)r?rOs r%r>zCurlMime.__init__s3 "-TTtvv ' (899 r') content_typefilename local_pathrPnamerrn str | Nonerorpstr | bytes | Path | NonerP bytes | Nonerr:c tj|j}tj||}|dkrt d|+>+@+@AACaxx 3444  (x/@/@AACaxx 3444  Q$ QOPP P  !*d++ ,!$ZJ.. ,!+!2!2!4!4!+''..00 O'(M^(M(MNNN(~/D/D/F/FGGCaxx 3444  dE** *4yy''))$T4T;;CCC  r'files list[dict]c>|}|D]}|jdi||S)zJCreate a multipart instance from a list of dict, for keys, see ``addpart``r4)r)clsrformfiles r% from_listzCurlMime.from_lists>suu ! !D DL 4  r'cf|r|n|j}|tj|jdS)z,Attach the mime instance to a curl instance.N)rrrMIMEPOSTrm)r?rOr0s r%attachzCurlMime.attachs1 (DDdj !4:.....r'cZtj|jtj|_dS)zrClose the mime instance and underlying files. This method must be called after ``perform`` or ``request``.N)r curl_mime_freermr rrs r%rzCurlMime.close s$ 4:&&&X r'c.|dSr<rrs r%rzCurlMime.__del__rr'r<)rOrj) rqrrnrrrorrrprsrPrtrr:)rr)rOrjrr:re) r0r1r2rCr>r classmethodrrrrr4r'r%riris--:::::$(#04!;<;<;<;<;<;rEr) Exceptionr6rjrkrlrmrnrorprrCURL_READFUNC_ABORTCURL_READFUNC_PAUSE def_externrTrbr{rrrrrrrrir4r'r%rsd"""""" %%%%%%444444444444 LLLLLLLLLLLLLL""""""     !""2:;<<=>>77777 777 !!      44444TTTT>    *,_=_=_=_=_=_=_=_=D\\\\\\\\\\r'