o Zj[/ @sbUddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z ddl mZmZddlmZmZmZdd lmZd gZejd kr^e Ze ed <d ZdejdejfddZn dejdejfddZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ* e +de,dede,fddZ-e +de,de,dedede,f dd Z.Gd!d d 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_loopreturncsvtvrtStttdtdsStjttddddl m }|t<j fdd}|_ 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)AddThreadSelectorEventLoopcs_tddSN)closerpopr loop_close selector_looprH/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/curl_cffi/aio.py_close_selector_and_loop7s  z.get_selector.._close_selector_and_loop) r isinstancegetattrasynciotypewarningswarnPROACTOR_WARNINGr_asyncio_selectorrr)rrr!rrr get_selectors   r*loopcCs|Srr)r+rrr r*Csr timeout_msclientpcCs>t|}|jr|jd|_|j|d|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)curlmr/r0 async_curlrrr timer_functionus  r:sockfdwhatdatacCst|}|j}||jvr|||||t@r*|||j|t |j ||t @r=| ||j|t |j ||tkrG|j|dS)z[This callback is called when libcurl decides it's time to interact with certain socketsr)rr1r+_sockfds remove_reader remove_writer CURL_POLL_IN add_readerr5CURL_CSELECT_INadd CURL_POLL_OUT add_writerCURL_CSELECT_OUTCURL_POLL_REMOVEremove)curlr;r<r0r=r9r+rrr socket_functions       rKc@seZdZdZd%defddZddZd d Zd d Zd e fddZ de de de fddZ de de fddZ d e fddZd e fddZd e fddZd e fddZde d efd!d"Zd#d$ZdS)&rzhWrapper around curl_multi handle to provide asyncio support. It uses the libcurl socket_action APIs.NcacertcCsft|_|pt|_i|_i|_t|_t |dur|nt |_ |j ||_d|_|dS)z Parameters: cacert: CA cert path to use, by default, certs from ``certifi`` are used. loop: EventLoop to use. N)rcurl_multi_init_curlmr _cacert _curl2future _curl2curlsetr>r*r$get_running_loopr+ create_task_force_timeout_timeout_checkerr2_setup)selfrMr+rrr __init__s   zAsyncCurl.__init__cCsP|tjtj|tjtjt||_ |tj |j |tj |j dSr) setoptr TIMERFUNCTIONrr:SOCKETFUNCTIONrKr new_handle _self_handle SOCKETDATA TIMERDATArYrrr rXs  zAsyncCurl._setupcs|jttj|jIdHWdn1swY|jD]\}}t|j |j | s?| s?| dq&t|j d|_ |jD]}|j||j|qL|jre|jdSdS)z?Close and cleanup running timers, readers, writers and handles.N)rWr3rr$CancelledErrorrQitemsrcurl_multi_remove_handlerO_curldone cancelled set_resultcurl_multi_cleanupr>r+r?r@r2)rYrJfuturer;rrr rs$      zAsyncCurl.closecs, |jsdS|tttdIdHq)zpThis coroutine is used to safeguard from any missing signals from curl, and put everything back on trackTg?N)rO socket_actionr6r7r$sleeprbrrr rVs zAsyncCurl._force_timeoutrJcCsF|t|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_cacertrcurl_multi_add_handlerOrf _check_errorr+ create_futurerQrR)rYrJerrcoderkrrr add_handles    zAsyncCurl.add_handler; ev_bitmaskrcCs.td}t|j|||}|||dS)zYwrapper for curl_multi_socket_action, returns the number of running curl handles.int *r)rnewrcurl_multi_socket_actionrOrp)rYr;rtrunning_handlerrrrr rls   zAsyncCurl.socket_actioncCs|js tjdtdddS|||td} zHt|j|}|tj kr)WdS|j t kr\|j |j }|jj}|}|durG|||n|dkrQ||n||||dntd Wntyrtjd tddYnwq) z8Call curl_multi_info_read to read data for given socket.z0Curlm already closed! quitting from process_datarrNruTrperformzNOT DONEzLUnexpected curl multi state in process_data, please open an issue on GitHub )rOr&r'rrlrrvrcurl_multi_info_readNULLmsg CURLMSG_DONErR easy_handler=result_get_callback_exception set_exceptionri _get_errorprint Exception)rYr;rt msg_in_queuecurl_msgrJretcodecallback_exceptionrrr r5sB        zAsyncCurl.process_datacCs8t|j|j}|||j|jd|j|dSr)rrerOrfrprRrrQ)rYrJrrrrr _pop_future*s zAsyncCurl._pop_futurecCs6||}|r|s|s|dSdSdSdS)z&Cancel a future for given curl handle.N)rrgrhr3rYrJrkrrr remove_handle0s   zAsyncCurl.remove_handlecCs8||}|r|s|s|ddSdSdSdS)z,Mark a future as done for given curl handle.N)rrgrhrirrrr ri6  zAsyncCurl.set_resultcCs8||}|r|s|s||dSdSdSdS)z2Raise exception of a future for given curl handle.N)rrgrhr)rYrJ exceptionrkrrr r<rzAsyncCurl.set_exceptionrrargscGsH|tjkrdSt|}ddd|D}td|d|d|d)N cSsg|]}t|qSr)str).0arrr Fsz*AsyncCurl._check_error..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 OKrcurl_multi_strerrorjoinr )rYrrrerrmsgactionrrr rpBs  zAsyncCurl._check_errorcCsB|tjtjtjtjtjtjfvrtd|}n|}t |j ||S)z!Wrapper around curl_multi_setopt.zlong*) r PIPELINING MAXCONNECTSMAX_HOST_CONNECTIONSMAX_PIPELINE_LENGTHMAX_TOTAL_CONNECTIONSMAX_CONCURRENT_STREAMSrrvrcurl_multi_setoptrO)rYoptionvaluec_valuerrr r[MszAsyncCurl.setopt)rLN)__name__ __module__ __qualname____doc__rrZrXrrVr rsintrlr5rrrirrrpr[rrrr rs   ' )0r$sysr& contextlibrtypingrrweakrefr_wrapperrrconstr r rJr r r utilsr__all__platformr__annotations__r(AbstractEventLoopr*r7rArECURL_POLL_INOUTrHr6CURL_SOCKET_BADrCrGCURL_CSELECT_ERRr}CURLPIPE_NOTHINGCURLPIPE_HTTP1CURLPIPE_MULTIPLEX def_externrr:rKrrrrr sP      *