o Zjx@sFddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZddlZddlZddlmZmZddlmZmZmZmZmZdd lmZdGd d ZeZe dZ!e dZ"e ruGdddZ#Gddde$Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7e4e6BZ8e5e7BZ9GdddZ:dHd d!Z;ejdLd0d1Z?ejd?ZEdNdAdBZFGdCdDdDZGGdEdFdFZHdS)O) annotationsN) SimpleCookie)Path) TYPE_CHECKINGAnyLiteralcast)ffilib) CurlECodeCurlHttpVersionCurlInfoCurlOpt CurlWsFlag)CurlCffiWarningreturnstrcCsXdD]}tj|}|rtj|r|Sqt}|jr(tj|jr(|jSt S)N) SSL_CERT_FILECURL_CA_BUNDLEREQUESTS_CA_BUNDLE) osenvirongetpathexistssslget_default_verify_pathscafilecertifiwhere)env_varrdefaultsr#I/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/curl_cffi/curl.py_default_cacerts r%sHTTP/\d\.\d [0-9]{3} (.*)sHTTP/(\d\.\d) ([0-9]{3}) (.*)c@s6eZdZUded<ded<ded<ded<ded<dS) CurlWsFrameintageflagsoffset bytesleftlenN)__name__ __module__ __qualname____annotations__r#r#r#r$r&,s  r&cs&eZdZdZ d d fd d ZZS) CurlErrorz$Base exception for curl_cffi packagermsgrcodeint | CurlECode | Literal[0]rNonecs$tj|g|Ri|||_dSN)super__init__r3)selfr2r3argskwargs __class__r#r$r87s zCurlError.__init__)r)r2rr3r4rr5)r-r.r/__doc__r8 __classcell__r#r#r<r$r14sr1ilic@seZdZdddZdS) _CallbackContextcallbackrrr5cCs||_d|_dSr6)rF exception)r9rFr#r#r$r8Ys z_CallbackContext.__init__N)rFrrr5)r-r.r/r8r#r#r#r$rEXsrE return_valuer'csfdd}|S)Ncs>|dur|jj}|d|d}|durt|}||_S)Nuserdataclientp)tb_framef_localsrr from_handlerG)exc_type exc_value traceback callback_argsrIcontextrHr#r$onerror_s z&_store_callback_error..onerrorr#)rHrTr#rSr$_store_callback_error^s  rU)rTtype_sizecCs@t|}|jdur dS|j}t||dd}|||dS)z ffi callback for curl debug infoNr)r rMrGrFbuffer)curlrVdatarWrJrRrFtextr#r#r$debug_functionns   r\Fbbytes uppercaseboolcs$|rdnddfdd|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} c3s|]}|VqdSr6)format).0bytefmtr#r$ szbytes_to_hex..)join)r]r_r#rer$ bytes_to_hexzs rirZr5cCs0tdtdtdtdtdtdtdi}d}||d}|ttfvrEt|d|}t ||kr.d nd }t j |d t |d ||d dSz#| d}t j |d||tttfvrft j d WdSWdStyt|d|}t ||krd nd }t j |d t |d ||d YdSw)N*<>z< DATAz> DATAz< SSLz> SSL(z...z [z bytes]:  zutf-8ra) CURLINFO_TEXTCURLINFO_HEADER_INCURLINFO_HEADER_OUTCURLINFO_DATA_INCURLINFO_DATA_OUTCURLINFO_SSL_DATA_INCURLINFO_SSL_DATA_OUTrrir,sysstderrwritedecodeUnicodeDecodeError)rVrZPREFIXESMAX_SHOW_BYTESprefixhex_strpostfixr[r#r#r$debug_function_defaults2  *  ,rcCs.t|j}|t||dd||S)zAffi callback for curl write function, directly writes to a bufferN)r rMrFryrX)ptrrWnmembrIrXr#r#r$buffer_callbacks rcCs|sdSt|SNr)r')sr#r#r$ ensure_intsrcCsdt|j}|t||dd}t|}|tks|tkr |S|||kr.tjdt dd||S)zHffi callback for curl write function, calls the callback python functionNzWrote bytes != received bytes.r@ stacklevel) r rMrFrXrCURL_WRITEFUNC_PAUSECURL_WRITEFUNC_ERRORwarningswarnr)rrWrrIrFwroter#r#r$write_callbacks  rcCst|j}||}||}|durdSt|tr|St|tr%|}|s)dSt||krZd_d?d@ZdmdDdEZedndGdHZedodJdKZd_dLdMZdpdOdPZejfdqdUdVZdrdsd[d\Z dS)tCurlz; Wrapper for ``curl_easy_*`` functions of libcurl. irnFNcacertrdebugr`rr5cCs|r|nt|_tj|_tj|_tj|_tj|_tj|_ |p t |_ d|_ d|_ d|_d|_d|_d|_d|_d|_tdd|_||_|td|j|_td|_td|_td|_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_http3_headers _ws_headers_proxy_headers_resolveDEFAULT_CACERT_cacert _is_cert_set _skip_cacert _write_handle_header_handle _debug_handle _body_handle _read_handle _seek_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)r9rrhandler#r#r$r8 s,   z Curl.__init__cCsBt|jtj|j}|dkrtjdtdd|j r| dSdS)NrzFailed to set error bufferr@r) r _curl_easy_setoptrr ERRORBUFFERrrrrrrr9retr#r#r$r+s  zCurl._set_error_buffercCs |tjd|tjddS)zSet debug to Truer TN)setoptrVERBOSE DEBUGFUNCTIONr9r#r#r$r2sz Curl.debugcC |dSr6closerr#r#r$__del__7 z Curl.__del__errcoder'r:rcGs.|dkrdS|j|g|R}|dur|dSr) _get_error)r9rr:errorr#r#r$ _check_error:s zCurl._check_errorcGsZ|dkr+t|jjdd}ddd|D}td|d|d |d tt|d SdS) NrbackslashreplaceerrorsracSsg|]}t|qSr#)r)rcar#r#r$ Esz#Curl._get_error..z Failed to z , curl: (z) zK. See https://curl.se/libcurl/c/libcurl-errors.html first for more details.)r3)r rrrzrhr1rr )r9rr:errmsgactionr#r#r$rBszCurl._get_errorBaseException | NonecCsF|j|j|j|j|jfD]}|dur t|j}|dur |Sq dSr6)rrrrrr rMrG)r9rrGr#r#r$_get_callback_exceptionMs zCurl._get_callback_exceptionoptionrvaluecCs|jdurdSdddddd}||dd}|dks |dkr(t||}n?|tjkrCtt|}||_t |jtj t j n$|tj kr^tt|}||_t |jtjt j n |tjkrxtt|}||_t |jtjt jn|tjkrtt|}||_t |jtjt jn|tj krtt|}||_t |jtj t jtj}n|tjkrtt|}||_t |jtjt jtj }n|tjkrtt|}||_t |jtjt jtj}n~|tjkr |d urt}tt|}||_t |jtjt jtj}nZ|dkr`t|t rTtj!tj"tj#tj$tj%tj&tj'tj(tj)tj*tj+tj,tj-tj.h}t/j01d rO||vrOt23d }|j4|d d }n|4}n|}|tj5kr_||_6nt7d|tj8dtj9dtj:di}||vr||} t;|| } |D] } t <| | } qt=|| | t |j|| } nN|tj>kr|D] } t <|j?| |_?qt |j||j?} n0|tj@kr|D]}t|t r|4}t <|jA||_Aqt |j||jA} nt |j||} |B| d|||tj!krd |_C| S)aWrapper for ``curl_easy_setopt``. Args: option: option to set, using constants from CurlOpt enum value: value to set, strings will be handled automatically Returns: 0 if no error, see ``CurlECode``. Nrlong*zchar*zvoid*int64_t*)r'i Ni0ui@rTwinFstrictrzOption unsupported: rrrr)Drrr rr WRITEDATA new_handlerErr r WRITEFUNCTIONr HEADERDATArHEADERFUNCTIONREADDATAr READFUNCTIONrSEEKDATAr SEEKFUNCTIONrrrrrrr\ DEBUGDATArrCAINFOCAPATH PROXY_CAINFO PROXY_CAPATHSSLCERTSSLKEYCRLFILE ISSUERCERTSSH_PUBLIC_KEYFILESSH_PRIVATE_KEYFILE COOKIEFILE COOKIEJAR NETRC_FILEUNIX_SOCKET_PATHrwplatform startswithlocalegetpreferredencodingr POSTFIELDSrNotImplementedError HTTPHEADERHTTP3_HTTPHEADER WS_HTTPHEADERgetattrcurl_slist_appendsetattr PROXYHEADERrRESOLVErrr)r9rr input_option value_typec_value filepath_optsencheader_options headers_attrheadersheaderr proxy_headerresolver#r#r$r[s                            z Curl.setoptr%bytes | int | float | list[str | int]c Csddddddd}tjtttttd}|d@}|jd ur'|d kr"d S||St||}|d kr;t|j||}nt |j||}| |d ||dkrTt |dS|d krnt|d}tt dd}||krldS|S|dtj krwd S|||dS)a Wrapper for ``curl_easy_getinfo``. Gets information in response after curl.perform. Parameters: option: option to get info of, using constants from ``CurlInfo`` enum Returns: value retrieved from last perform. zchar**rzdouble*zstruct curl_slist **z uintptr_t*r)i i0@Pi`iNrrgetinforr uintptr_t)r rr'floatlistrrr _curl_easy_getinfo_socketcurl_easy_getinforrrr) r9r ret_optionret_cast_option option_typerrsocket socket_badr#r#r$r!sB      z Curl.getinfor^cCsttS)z#Get the underlying libcurl version.)r rr curl_versionrr#r#r$versionsz Curl.versionTtargetdefault_headerscCs&|jdurdSt|j|t|S)zSet the browser type to impersonate. Parameters: target: browser to impersonate. default_headers: whether to add default headers, like User-Agent. Returns: 0 if no error. Nr)rr curl_easy_impersonaterr')r9r/r0r#r#r$ impersonate"s zCurl.impersonatecCsP|jrdS|js&|tj|j}||d|tj|j}||ddSdS)Nz set cacertzset proxy cacert)rrrrrrrrrr#r#r$_ensure_cacert2s zCurl._ensure_cacert clear_headers clear_resolvec Csh|jdur td|t|j}z|}|dur|||dW|||dS|||w)a+Wrapper for ``curl_easy_perform``, performs a curl request. Parameters: clear_headers: clear header slist used in this perform clear_resolve: clear resolve slist used in this perform Raises: CurlError: if the perform was not successful. Nz(Cannot perform request on closed handle.perform)rr1r3r curl_easy_performrrclean_handles_and_buffers)r9r4r5rcallback_exceptionr#r#r$r6;s  z Curl.performcCs|jdurdSt|jSr)rr curl_easy_upkeeprr#r#r$upkeepWs  z Curl.upkeeprcCs,|jdurdSt|j|}||d|S)z-Pause or resume data transfer on this handle.Nrpause)rr curl_easy_pauser)r9rrr#r#r$r<\s  z Curl.pausecCsd|_d|_d|_d|_d|_d|_|r$|jtjkr t |jtj|_|rCd}|D]}t ||}|tjkr;t |t ||tjq*dSdS)zuClean up handles and buffers after ``perform`` and ``close``, called at the end of ``perform`` and ``close``.N)rrrr) rrrrrrrr rr rr r)r9r4r5 header_attrs header_attrrr#r#r$r8ds&     zCurl.clean_handles_and_bufferscCs4|jdur tdt|j}t|j|j|d}|S)zWrapper for ``curl_easy_duphandle``. This is not a full copy of entire curl object in python. For example, headers handle is not copied, you have to set them again.NzCannot duplicate closed handle.)rrr)rr1r curl_easy_duphandlerrr)r9rcr#r#r$ duphandles  zCurl.duphandlecCs6d|_d|_|jdurt|j|tj|_dS)z8Reset all curl options, wrapper for ``curl_easy_reset``.FN) rrrr curl_easy_resetrr rrrr#r#r$resets    z Curl.resetrrrcCs8t}|D]}|dr||ddq|S)zExtract ``cookies.SimpleCookie`` from header lines. Parameters: headers: list of headers in bytes. Returns: A parsed cookies.SimpleCookie instance. s set-cookie: N)rlowerrloadrz)r9rcookierr#r#r$parse_cookie_headerss  zCurl.parse_cookie_headers status_linecCst|}|r |dSdS)zTExtract reason phrase, like ``OK``, ``Not Found`` from response status line.r r )REASON_PHRASE_REmatchgroup)rJmr#r#r$get_reason_phrases zCurl.get_reason_phrase"tuple[CurlHttpVersion, int, bytes]cCst|}|s tjddfS|ddkrtj}n|ddkr#tj}n|ddkr.tj}ntj}t|d}|d}|||fS) zfParse status line. Returns: http_version, status_code, and reason phrase rr r z2.0z1.1z1.0r@rA) STATUS_LINE_RErLr V1_0rMV2_0V1_1NONEr')rJrN http_version status_codereasonr#r#r$parse_status_lines    zCurl.parse_status_linecCsP||jrt|jd|_t|j|jdur&t|jd|_dSdS)zAClose and cleanup curl handle, wrapper for ``curl_easy_cleanup``.N)r8rr curl_easy_cleanupr releaserrrr#r#r$rs     z Curl.closetuple[bytes, CurlWsFrame]cCsb|jdur tdt|j|j|j|j|j}r||dt |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) rr1r curl_ws_recvrrrrrr rXrr#r#r$ws_recvs  z Curl.ws_recvpayloadbytes | memoryviewr)CurlWsFlag | intcCsN|jdur tdt|jt|t||jd|}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) rr1r curl_ws_sendr from_bufferr,rr)r9r`r)rr#r#r$ws_sends   z Curl.ws_sendr r3messagecCs td||}|j|tjdS)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))structpackrfrCLOSE)r9r3rhr`r#r#r$ws_closesz Curl.ws_close)rnFN)rrrr`rr5rr5)rr'r:rrr5)rr'r:r)rr)rrrrrr')rrrr)rr^)T)r/rr0r`rr')TT)r4r`r5r`rr5)rr')rr'rr')rr)rrrr)rJr^rr^)rJr^rrP)rr\)r`rar)rbrr')rgr )r3r'rhr^rr')!r-r.r/r>rr8rrrrrrrr!r.r2r3r6r;r<r8rBrDrI staticmethodrOrYrr_rBINARYrfrlr#r#r#r$rsB  !       4           rc@s`eZdZdZdd ddZdddddd!ddZed"ddZdd#ddZd$ddZ d$ddZ dS)%CurlMimez#Wrapper for the ``curl_mime_`` API.NrY Curl | NonecCs$|r|nt|_t|jj|_dS)z? Args: curl: Curl instance to use. N)rrr curl_mime_init_form)r9rYr#r#r$r8szCurlMime.__init__) content_typefilename local_pathrZnamerrt str | Nonerurvstr | bytes | Path | NonerZ bytes | Nonerr5c Cs2t|j}t||}|dkrtd|dur*t||}|dkr*td|r2|r2td|durit|tr@t |}n t|t rJ| }n|}t| sYt d|t||}|dkritd|dur}t||}|dkr}td|durt|t st |}t||t|}dSdS)aAdd a mime part for a mutlipart html form. Note: You can only use either local_path or data, not both. Args: name: name of the field. content_type: content_type for the field. for example: ``image/png``. filename: filename for the server. local_path: file to upload on local disk. data: file content to upload. rzAdd field failed.Nz1Can not use local_path and data at the same time.zFile not found at )r curl_mime_addpartrscurl_mime_namerr1curl_mime_typerrrr^rzrFileNotFoundErrorcurl_mime_filedatacurl_mime_filenamecurl_mime_datar,) r9rwrtrurvrZpartrlocal_path_strr#r#r$addpart&s<        zCurlMime.addpartfiles list[dict]cCs$|}|D] }|jdi|q|S)zJCreate a multipart instance from a list of dict, for keys, see ``addpart``Nr#)r)clsrformfiler#r#r$ from_listcszCurlMime.from_listcCs"|r|n|j}|tj|jdS)z,Attach the mime instance to a curl instance.N)rrrMIMEPOSTrs)r9rYrAr#r#r$attachkszCurlMime.attachcCst|jtj|_dS)zrClose the mime instance and underlying files. This method must be called after ``perform`` or ``request``.N)r curl_mime_freersr rrr#r#r$rps  zCurlMime.closecCrr6rrr#r#r$rvrzCurlMime.__del__r6)rYrq) rwrrtrxrurxrvryrZrzrr5)rr)rYrqrr5rm) r-r.r/r>r8r classmethodrrrrr#r#r#r$rps  =   rp)rr)rHr')rVr'rWr'rr')F)r]r^r_r`rr)rVr'rZr^rr5)r*r'rr'rr')rr)I __future__rrrerirrwr http.cookiesrpathlibrtypingrrrrrr_wrapperr r constr r rrrutilsrr%rcompilerKrQr& Exceptionr1rprqrrrsrtrurvrrCURL_READFUNC_ABORTCURL_READFUNC_PAUSErCURL_SEEKFUNC_FAILrCURLPAUSE_RECVCURLPAUSE_RECV_CONTCURLPAUSE_SENDCURLPAUSE_SEND_CONT CURLPAUSE_ALLCURLPAUSE_CONTrErU def_externr\rirrrrrrrrrrpr#r#r#r$s