o gZh @sHddlZddlmZddlmZddlmZmZm Z GdddeZ dS)N)ObjectApiResponse)NamespacedClient) SKIP_IN_PATH_quote_rewrite_parametersc@sbeZdZeddddddedejedejejeej efdejedejede ej f d d Z edddddd dedejedejejeej efdejedejed ejede ej fd dZ edddddddejejeej efdejedejejeej efdejedejede ej f ddZedddddddddddedejedejejeej efdejejeej fdejedejejeej fdejedejejeej fde ej fddZeddddddejedejejeej efdejedejede ej f ddZdS) EnrichClientN) error_trace filter_pathhumanprettynamer r r r returnc Cs||tvrtddt|}i}|dur||d<|dur!||d<|dur)||d<|dur1||d<dd i}|jd |||d S) z Deletes an existing enrich policy and its enrich index. ``_ :param name: Enrich policy to delete. 'Empty value passed for parameter 'name'/_enrich/policy/Nr r r r acceptapplication/jsonDELETEparamsheadersr ValueErrorrperform_request selfr r r r r _EnrichClient__path_EnrichClient__query_EnrichClient__headersrX/var/www/html/lang_env/lib/python3.10/site-packages/elasticsearch/_sync/client/enrich.py delete_policys zEnrichClient.delete_policy)r r r r wait_for_completionr"c Cs|tvrtddt|d}i}|dur||d<|dur"||d<|dur*||d<|dur2||d<|dur:||d <d d i} |jd ||| d S)ah Creates the enrich index for an existing enrich policy. ``_ :param name: Enrich policy to execute. :param wait_for_completion: If `true`, the request blocks other enrich policy execution requests until complete. rrz /_executeNr r r r r"rrPUTrr) rr r r r r r"rrrrrr execute_policy=s$zEnrichClient.execute_policy)r r r r r c Csz|tvr dt|}nd}i}|dur||d<|dur ||d<|dur(||d<|dur0||d<dd i}|jd |||d S) aM Gets information about an enrich policy. ``_ :param name: Comma-separated list of enrich policy names used to limit the request. To return information for all enrich policies, omit this parameter. rz/_enrich/policyNr r r r rrGETr)rrrrrrr get_policyds zEnrichClient.get_policyT)Z body_fields)r r geo_matchr matchr ranger'r(r)c Cs|tvrtddt|} i} i} |dur|| d<|dur#|| d<|dur+|| d<|dur3|| d<|dur;|| d<|durC|| d <|durK|| d <d d d } |jd | | | | dS)a6 Creates a new enrich policy. ``_ :param name: Name of the enrich policy to create or update. :param geo_match: Matches enrich data to incoming documents based on a `geo_shape` query. :param match: Matches enrich data to incoming documents based on a `term` query. :param range: Matches a number, date, or IP address in incoming documents to a range in the enrich index based on a `term` query. rrNr r r'r r(r r)r)rz content-typer#)rrbodyr) rr r r r'r r(r r)rrZ_EnrichClient__bodyrrrr put_policys.  zEnrichClient.put_policycCsbd}i}|dur ||d<|dur||d<|dur||d<|dur$||d<ddi}|jd |||d S) z Gets enrich coordinator statistics and information about enrich policies that are currently executing. ``_ z/_enrich/_statsNr r r r rrr%r)r)rr r r r rrrrrr statsszEnrichClient.stats)__name__ __module__ __qualname__rstrtOptionalboolUnionSequencerAnyr!r$r&Mappingr+r,rrrr rs ! & #   0r) typingr1Zelastic_transportr_baserutilsrrrrrrrr s