o gZh: @s UddlZddlZddlZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZddlmZm Z m!Z!m"Z"m#Z#m$Z$ddl%m&Z&dd l'm(Z(m)Z)m*Z*erud d l+m,Z,dd d gdfZ-e e e.d<de e&fZ/e!de&Z0ee1eee1ee1ee1e2ffeffZ3e eege eefZ4e eegeefZ5hdZ6ede de fdZ7 d?dee3dee1dee de deef ddZ8de3deefddZ9dee1ee1e2ffdefd d!Z:de1deefd"d#Z;d$ee1ee1ee1e1ffde1fd%d&Zd,ee1e fde1fd-d.Z?de e1e fd/e e1e fddfd0d1Z@  2  d@d3ee1d4eAd5ee e1e1fd6eee1de e7ge7ff d7d8ZBd9e deAfd:d;ZCdZDdS)AN)datedatetime)wraps) TYPE_CHECKINGAny AwaitableCallable CollectionDictListMappingOptionalSequenceSetTupleTypeTypeVarUnion)AsyncTransport HttpHeaders NodeConfigRequestsHttpNode SniffOptions Transport)DEFAULTclient_meta_versioncreate_user_agentparse_cloud_idpercent_encodeurl_to_node_config)__versionstr__)to_bytesto_strwarn_stacklevel)NamespacedClient SKIP_IN_PATHeszelasticsearch-py>Z opaque_idZapi_keyrequest_timeoutignoreheaders http_authF.)boundhostscloud_idrequests_session_authkwargsreturnc s|dur|dur tdt|}n |dusJt|}dd|Dtdd}|dt|d<|durB|did<d td tffd d fd d|DS)Nzosz'client_node_configs..r.r) user-agent_extraszrequests.session.auth node_configr6csP|j}dd}|r|||dt||d<|jdiS)zMNeeds special handling of headers since .replace() wipes out existing headersr.r)r=)r.copypopupdate setdefault USER_AGENTfreezereplace)r?r.Zheaders_to_add) node_optionsr)r;apply_node_options|s    z/client_node_configs..apply_node_optionscsg|]}|qSr)r))r8r?)rHr)r; z'client_node_configs..) ValueErrorcloud_id_to_node_configshosts_to_node_configsitemsrrArCrDr)r2r3r4r5 node_configsr.r))rHrGr;client_node_configs^s&    rPcCsvt|tr t|gSg}|D]*}t|tr||qt|tr(|t|qt|tr5|t|qtd|S)z7Transforms the many formats of 'hosts' into NodeConfigsz<'hosts' must be a list of URLs, NodeConfigs, or dictionaries) isinstancestrrMrappendrr host_mapping_to_node_configrK)r2rOhostr)r)r;rMs      rMrUcCshd}t||}|r(dt|}dt|}td|d|dt|}d|vrj|d}t|t s>t dd|vrW|dd k|krWtd |dd |d t j d t td|d|rgd ndd|vrd|vrvtdt j dt td|d|d<tdi|S)zCConverts an old-style dictionary host specification to a NodeConfig>use_sslschemeportrU url_prefix path_prefixz', 'zCan't specify the options 'z%' via a dictionary in 'hosts', only 'z' options are allowedrVz 'use_ssl' must be of type 'bool'rWhttpsz+Cannot specify conflicting options 'scheme=z' and 'use_ssl=z'. Use 'scheme' only insteadzQThe 'use_ssl' option is no longer needed as specifying a 'scheme' is now requiredcategory stacklevelhttprYrZzaCannot specify conflicting options 'url_prefix' and 'path_prefix'. Use 'path_prefix' only insteadz?The 'url_prefix' option is deprecated in favor of 'path_prefix'Nr))setkeys differencejoinsortedrKdictrArQbool TypeErrorwarningswarnDeprecationWarningr$rCr)rUZallow_hosts_keysZdisallowed_keysZ bad_keys_usedZ allowed_keysoptionsrVr)r)r;rTsN   rTcCs<t|j}|dus t|std|\}}td||ddgS)z0Transforms an Elastic Cloud ID into a NodeConfigNzt|S|S)z Escape a single value of a URL string or a query parameter. If it is a list or tuple, turn it into a comma-separated string first. ,cSsg|]}t|qSr))_escape)r8itemr)r)r;rIrJz_escape..zutf-8 surrogatepass) rQrprqrcrr isoformatrfrRlowerbytesrtrvr)r)r;rxs     rxcCstt|dS)Nz,*)rrxr~r)r)r;_quotesrquerycCsddd|DS)N&cSs"g|] \}}|dt|qS)=)rr7r)r)r;rIs"z _quote_query..)rcrN)rr)r)r; _quote_querysrvaluescCs6|D]\}}||vrtd|d|||<qdS)NzReceived multiple values for 'zB', specify parameters directly instead of using 'body' or 'params')rNrK)r5rkeyvalr)r)r;_merge_kwargs_no_duplicatess  rF body_name body_fieldsparameter_aliasesignore_deprecated_optionscs dtdtffdd }|S)Napir6cs.tdtdtdtffdd }|S)Nargsr5r6c st|dkr tdd|vr2rdvr2|d}|r2t|ds$tdtjdttdt ||t |}|ri}|D]#}rF|vrFq=z|}|dkrOd }||||<Wq=t y`Yq=w|rtjd ttd|d }d}t|d r~t |}|j}|jdi|}|dur||}|f|d d}d|vrӈrdvr|d} | durӈrÈ|vrtdddd| |<nrt| dstdt || rD]\} } z || || <Wqt yYqw|i|S)NzfPositional arguments can't be used with Elasticsearch API methods. Instead only use keyword arguments.paramsrNz}Couldn't merge 'params' with other parameters as it wasn't a mapping. Instead of using 'params' use individual API parameterszpThe 'params' parameter is deprecated and will be removed in a future version. Instead use individual parameters.r\r-Z ignore_statuszaPassing transport options in the API method is deprecated. Use 'Elasticsearch.options()' instead.r_clientr%bodyz Can't use 'z*' and 'body' parameters together because 'z;' is an alias for 'body'. Instead you should only use the 'z]' parameter. See https://github.com/elastic/elasticsearch-py/issues/1698 for more informationzyCouldn't merge 'body' with other parameters as it wasn't a mapping. Instead of using 'body' use individual API parametersr))lenrgrAhasattrrKrhrirjr$r_TRANSPORT_OPTIONS intersectionKeyErrortyperrkrN) rr5rZmaybe_transport_optionsZtransport_optionsoptionZ option_renameclientZnamespaced_clientraliasZ rename_to)rrrrrr)r;wrapped0s            z5_rewrite_parameters..wrapper..wrapped)rr)rrrrrr)rr;wrapper/s$cz$_rewrite_parameters..wrapper)r0)rrrrrr)rr;_rewrite_parameters)sgrr/cCs.z ddlm}t||WStyYdSw)z=Detect if an http_auth value is a custom Requests auth objectr)AuthBaseF)Z requests.authrrQ ImportError)r/rr)r)r;is_requests_http_auths   r node_classcCs,|duo|tuo|dkpt|ot|tS)zLDetect if 'RequestsHttpNode' would be used given the setting of 'node_class'Nrequests)rinspectisclass issubclassr)rr)r)r;is_requests_node_classs r)N)NFNN)Errrrhrr functoolsrtypingrrrrr r r r r rrrrrrZelastic_transportrrrrrrZelastic_transport.client_utilsrrrrrr_versionr!compatr"r#r$_baser&r*__annotations__ZCLIENT_META_SERVICErDrRintZ _TYPE_HOSTSZ_TYPE_ASYNC_SNIFF_CALLBACKZ_TYPE_SYNC_SNIFF_CALLBACKrr0rPrMrTrLrurxrrrrfrrrr)r)r)r;sz  D      $  /"=( &    p