Hj. UddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z ddl mZmZddlmZmZmZdd lmZd gZejd kr$e Ze ed <d ZdejdejfdZndejdejfdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ* e j+de,dede,fdZ-e j+de,de,dedede,f dZ.Gdd Z/dS)N)suppress)AnyOptional)WeakKeyDictionary)ffilib) CurlECodeCurlMOpt)DEFAULT_CACERTCurl CurlError)CurlCffiWarning AsyncCurlwin32 _selectorsa Proactor event loop does not implement add_reader family of methods required. Registering an additional selector thread for add_reader support. To avoid this warning use: asyncio.set_event_loop_policy(WindowsSelectorEventLoopPolicy()) asyncio_loopreturnc @tvr tStttdt dsSt jttdddl m }|xt<j fd}|_ S)zGet selector-compatible loop Returns an object with ``add_reader`` family of methods, either the loop itself or a SelectorThread instance. Workaround Windows proactor removal of *reader methods. ProactorEventLoopN stacklevelr)AddThreadSelectorEventLoopct_tddSN)closerpop)r loop_close selector_loopsL/home/lighthouse/health/venv/lib64/python3.11/site-packages/curl_cffi/aio.py_close_selector_and_loopz.get_selector.._close_selector_and_loop7s9",L  NN< . . .    ! ! ! ! !) r isinstancegetattrasynciotypewarningswarnPROACTOR_WARNINGr_asyncio_selectorrr)rrr"rr s` @@r! get_selectorr,s : % %l+ + ''+>T KK     &AFFFFAAAAAA3M3M 4 4  <0 "'  " " " " " " "6 r#loopc|Sr)r-s r!r,r,Cs r#r timeout_msclientpctj|}|jr |jd|_|j|dz |jtt|_dS)zD see: https://curl.se/libcurl/c/CURLMOPT_TIMERFUNCTION.html Nir) r from_handle_timercancelr- call_later process_dataCURL_SOCKET_TIMEOUTCURL_POLL_NONE)curlmr3r4 async_curls r!timer_functionr?ust ))J!  """ #22T J 1r#sockfdwhatdatactj|}|j}||jvr*|||||t zr<|||j|t|j ||tzr<| ||j|t|j ||tkr|j|dS)z[This callback is called when libcurl decides it's time to interact with certain socketsr)rr6r-_sockfds remove_reader remove_writer CURL_POLL_IN add_readerr:CURL_CSELECT_INadd CURL_POLL_OUT add_writerCURL_CSELECT_OUTCURL_POLL_REMOVEremove)curlr@rAr4rBr>r-s r!socket_functionrQs ))J ?D$$$ 6""" 6""" l(  7QQQ''' m(  7AQRRR''' ""6*** 1r#ceZdZdZddefdZdZdZdZd e fd Z d e d e d e fdZ d e d e fdZ d e fdZd e fdZd e fdZd e fdZde defdZdZdS)rzhWrapper around curl_multi handle to provide asyncio support. It uses the libcurl socket_action APIs.Ncacertc~tj|_|pt|_i|_i|_t|_t||ntj |_ |j ||_d|_|dS)z Parameters: cacert: CA cert path to use, by default, certs from ``certifi`` are used. loop: EventLoop to use. N)r curl_multi_init_curlmr _cacert _curl2future _curl2curlsetrDr,r&get_running_loopr- create_task_force_timeout_timeout_checkerr7_setup)selfrTr-s r!__init__zAsyncCurl.__init__s )++ / 8:13"%%%  $DD'*B*D*D   !% 5 5d6I6I6K6K L L59  r#ct|tjtj|tjtjtj||_ |tj |j |tj |j dSr) setoptr TIMERFUNCTIONr r?SOCKETFUNCTIONrQr new_handle _self_handle SOCKETDATA TIMERDATAras r!r`zAsyncCurl._setups H*C,>??? H+S-@AAAN400 H'):;;; H&(9:::::r#cK|jttj5|jd{Vdddn #1swxYwY|jD]a\}}tj|j |j | s)| s| dbtj|j d|_ |jD]6}|j||j|7|jr|jdSdS)z?Close and cleanup running timers, readers, writers and handles.N)r_r8rr&CancelledErrorrYitemsr curl_multi_remove_handlerW_curldone cancelled set_resultcurl_multi_cleanuprDr-rErFr7)rarPfuturer@s r!rzAsyncCurl.closes $$&&& g, - - ( (' ' ' ' ' ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (!-3355 ( (LD&  (dj A A A;;== ()9)9);); (!!$''' t{+++ m , ,F I # #F + + + I # #F + + + + ; ! K    ! !sAAAcK |jsdS|ttt jdd{VD)zpThis coroutine is used to safeguard from any missing signals from curl, and put everything back on trackTg?N)rW socket_actionr;r<r&sleeprks r!r^zAsyncCurl._force_timeoutsX %;    2N C C C-$$ $ $ $ $ $ $ $  %r#rPc|tj|j|j}|||j}||j|<||j |j<|S)znAdd a curl handle to be managed by curl_multi. This is the equivalent of `perform` in the async world.) _ensure_cacertr curl_multi_add_handlerWrp _check_errorr- create_futurerYrZ)rarPerrcoderus r! add_handlezAsyncCurl.add_handless +DKDD '"""((**"($&* # r#r@ ev_bitmaskrctjd}tj|j|||}|||dS)zYwrapper for curl_multi_socket_action, returns the number of running curl handles.int *r)rnewr curl_multi_socket_actionrWr|)rar@rrunning_handler~s r!rwzAsyncCurl.socket_actionsQ)). K^   '"""a  r#c`|jstjdtddS|||t jd} tj|j|}|t j krdS|j tkre|j |j }|jj}|dkr||n:||||dnt'd n,#t($rtjd tdYnwxYw) z8Call curl_multi_info_read to read data for given socket.z0Curlm already closed! quitting from process_datarrNrTrperformzNOT DONEzLUnexpected curl multi state in process_data, please open an issue on GitHub )rWr(r)rrwrrr curl_multi_info_readNULLmsg CURLMSG_DONErZ easy_handlerBresultrs set_exception _get_errorprint Exception)rar@r msg_in_queuecurl_msgrPretcodes r!r:zAsyncCurl.process_datasR{  MB     F 6:...ww''   3DKNNsx''E<<//?8+?@D&m2G!||----**4)1T1TUUUU*%%%    7#   s*D>BD&D,+D,ctj|j|j}|||j|jd|j|dSr)r rorWrpr|rZrrY)rarPr~s r! _pop_futurezAsyncCurl._pop_future's_.t{DJGG '""" DJ--- $$T4000r#c||}|r>|s,|s|dSdSdSdS)z&Cancel a future for given curl handle.N)rrqrrr8rarPrus r! remove_handlezAsyncCurl.remove_handle-sn!!$''  &++-- 0@0@0B0B  MMOOOOO      r#c||}|r?|s-|s|ddSdSdSdS)z,Mark a future as done for given curl handle.N)rrqrrrsrs r!rszAsyncCurl.set_result3sw!!$''  $&++-- $0@0@0B0B $   d # # # # # $ $ $ $ $ $r#c||}|r?|s-|s||dSdSdSdS)z2Raise exception of a future for given curl handle.N)rrqrrr)rarP exceptionrus r!rzAsyncCurl.set_exception9sw!!$''  ,&++-- ,0@0@0B0B ,   + + + + + , , , , , ,r#r~argsc |tjkrdStj|}dd|D}t d|d|d|d)N c,g|]}t|Sr/)str).0as r! z*AsyncCurl._check_error..Cs000a3q66000r#z Failed in z , multi: (z) z. See https://curl.se/libcurl/c/libcurl-errors.html first for more details. Please open an issue on GitHub to help debug this error.)r OKr curl_multi_strerrorjoinr)rar~rerrmsgactions r!r|zAsyncCurl._check_error?s il " " F(1100400011 P P P7 P Pf P P P   r#c|tjtjtjtjtjtjfvrtjd|}n|}tj |j ||S)z!Wrapper around curl_multi_setopt.zlong*) r PIPELINING MAXCONNECTSMAX_HOST_CONNECTIONSMAX_PIPELINE_LENGTHMAX_TOTAL_CONNECTIONSMAX_CONCURRENT_STREAMSrrr curl_multi_setoptrW)raoptionvaluec_values r!rdzAsyncCurl.setoptJsi      )  (  *  +    ggu--GGG$T[&'BBBr#)rSN)__name__ __module__ __qualname____doc__rrbr`rr^r rintrwr:rrrsrrr|rdr/r#r!rrsts$;;;!!!6%%% t    !C!S!S!!!!"3"C""""H11111 $ $t$$$$ ,$,,,,  C        C C C C Cr#)0r&sysr( contextlibrtypingrrweakrefr_wrapperrr constr r rPr r rutilsr__all__platformr__annotations__r*AbstractEventLoopr,r<rGrKCURL_POLL_INOUTrNr;CURL_SOCKET_BADrIrMCURL_CSELECT_ERRrCURLPIPE_NOTHINGCURLPIPE_HTTP1CURLPIPE_MULTIPLEX def_externrr?rQrr/r#r!rss  %%%%%%&&&&&&&&1111111111"""""" -<7$5$5$7$7J!777&/&  "&&&&&T749R   4 c C C    . # S 3 c c    :lClClClClClClClClClCr#