o gZh/=@sHddlZddlmZddlmZddlmZmZm Z GdddeZ dS)N)ObjectApiResponse)NamespacedClient) SKIP_IN_PATH_quote_rewrite_parametersc0@seZdZeddddddddedejedejejeej efdejed ejed e ej f d d Z edddddd edejedejejeej efdejed ejed e ej f ddZ edddddddddd edejedejedejejeej efdejedejedejejddefd ejedejejddefd e ej fddZedddddd edejedejejeej efdejed ejed e ej f ddZedddhddddddddddddddddddddddddejed ejedejedejed!ejed"ejed#ejejeej fdejejeej efdejedejed$ejedejejddefd%ejed&ejejddefdejejeej fd ejed'ejedejejddefd(ejejeejeej ffd)ejedejejddefd e ej f,d*d+Zedddddddddd,d'edejed!ejed#ejejeej fdejejeej efdejed ejed)ejed e ej fd-d.ZdS)/ SqlClientT) body_fieldsN) error_trace filter_pathhumanprettycursorr r r r returnc Cs|durtdd}i}i}|dur||d<|dur||d<|dur&||d<|dur.||d<|dur6||d<d d d } |jd ||| |d S) z Clears the SQL cursor ``_ :param cursor: Cursor to clear. Nz)Empty value passed for parameter 'cursor'z /_sql/closerr r r r application/jsonacceptz content-typePOSTparamsheadersbody ValueErrorperform_request) selfrr r r r _SqlClient__path_SqlClient__body_SqlClient__query_SqlClient__headersr U/var/www/html/lang_env/lib/python3.10/site-packages/elasticsearch/_sync/client/sql.py clear_cursors&  zSqlClient.clear_cursoridc C||tvrtddt|}i}|dur||d<|dur!||d<|dur)||d<|dur1||d<dd i}|jd |||d S) a( Deletes an async SQL search or a stored synchronous SQL search. If the search is still running, the API cancels it. ``_ :param id: Identifier for the search. %Empty value passed for parameter 'id'z/_sql/async/delete/Nr r r r rrDELETErrrrrr rr#r r r r rrrr r r! delete_asyncB zSqlClient.delete_async) delimiterr r formatr keep_aliver wait_for_completion_timeoutr,r-r.z t.Literal[-1]z t.Literal[0]r/c Cs|tvrtddt|} i} |dur|| d<|dur!|| d<|dur)|| d<|dur1|| d<|dur9|| d<|durA|| d <|durI|| d <| durQ| | d <d d i} |jd| | | dS)a Returns the current status and available results for an async SQL search or stored synchronous SQL search ``_ :param id: Identifier for the search. :param delimiter: Separator for CSV results. The API only supports this parameter for CSV responses. :param format: Format for the response. You must specify a format using this parameter or the Accept HTTP header. If you specify both, the API uses this parameter. :param keep_alive: Retention period for the search and its results. Defaults to the `keep_alive` period for the original SQL search. :param wait_for_completion_timeout: Period to wait for complete results. Defaults to no timeout, meaning the request waits for complete search results. r%z /_sql/async/Nr,r r r-r r.r r/rrGETr'r() rr#r,r r r-r r.r r/rrrr r r! get_asynces0!zSqlClient.get_asyncc Cr$) a  Returns the current status of an async SQL search or a stored synchronous SQL search ``_ :param id: Identifier for the search. r%z/_sql/async/status/Nr r r r rrr0r'r(r)r r r!get_async_statusr+zSqlClient.get_async_statusrrequest_timeout)r Zignore_deprecated_options)catalogcolumnarrr fetch_sizefield_multi_value_leniencyfilterr r-r index_using_frozenr.keep_on_completion page_timeoutrr queryr3runtime_mappings time_zoner/r4r5r6r7r8r9r:r;r<r=r>cCszd}i}i}|dur||d<|dur||d<|dur||d<|dur&||d<|dur.||d<|dur6||d<|dur>||d <|durF||d <| durN| |d <| durV| |d <| dur^| |d <| durf| |d<| durn| |d<|durv||d<|dur~||d<|dur||d<|dur||d<|dur||d<|dur||d<|dur||d<|dur||d<ddd}|jd||||dS)u Executes a SQL request ``_ :param catalog: Default catalog (cluster) for queries. If unspecified, the queries execute on the data in the local cluster only. :param columnar: If true, the results in a columnar fashion: one row represents all the values of a certain column from the current page of results. :param cursor: Cursor used to retrieve a set of paginated results. If you specify a cursor, the API only uses the `columnar` and `time_zone` request body parameters. It ignores other request body parameters. :param fetch_size: The maximum number of rows (or entries) to return in one response :param field_multi_value_leniency: Throw an exception when encountering multiple values for a field (default) or be lenient and return the first value from the list (without any guarantees of what that will be - typically the first in natural ascending order). :param filter: Elasticsearch query DSL for additional filtering. :param format: Format for the response. :param index_using_frozen: If true, the search can run on frozen indices. Defaults to false. :param keep_alive: Retention period for an async or saved synchronous search. :param keep_on_completion: If true, Elasticsearch stores synchronous searches if you also specify the wait_for_completion_timeout parameter. If false, Elasticsearch only stores async searches that don’t finish before the wait_for_completion_timeout. :param page_timeout: The timeout before a pagination request fails. :param params: Values for parameters in the query. :param query: SQL query to run. :param request_timeout: The timeout before the request fails. :param runtime_mappings: Defines one or more runtime fields in the search request. These fields take precedence over mapped fields with the same name. :param time_zone: ISO-8601 time zone ID for the search. :param wait_for_completion_timeout: Period to wait for complete results. Defaults to no timeout, meaning the request waits for complete search results. If the search doesn’t finish within this period, the search becomes async. z/_sqlNr4r5rr r6r7r8r r-r r9r.r:r;rr r<r3r=r>r/rrrr)r)rr4r5rr r6r7r8r r-r r9r.r:r;rr r<r3r=r>r/rrrrr r r!r<sbE  zSqlClient.query)r r6r8r r r r>c Cs|durtdd} i} i} |dur|| d<|dur|| d<|dur&|| d<|dur.|| d<|dur6|| d<|dur>|| d <|durF|| d <|durN|| d <d d d } |jd| | | | dS)a Translates SQL into Elasticsearch queries ``_ :param query: SQL query to run. :param fetch_size: The maximum number of rows (or entries) to return in one response. :param filter: Elasticsearch query DSL for additional filtering. :param time_zone: ISO-8601 time zone ID for the search. Nz(Empty value passed for parameter 'query'z/_sql/translater<r r6r8r r r r>rrrrr) rr<r r6r8r r r r>rrrrr r r! translate9s2  zSqlClient.translate)__name__ __module__ __qualname__rstrtOptionalboolUnionSequencerAnyr"r*r1r2intMappingr<r?r r r r!rs $ "    9 "      s   r) typingrDZelastic_transportr_baserutilsrrrrr r r r!s