Skip to content

Commit 31e0fad

Browse files
committed
Update client to reflect elasticsearch-specification#2276
1 parent 4d501d4 commit 31e0fad

File tree

2 files changed

+20
-12
lines changed

2 files changed

+20
-12
lines changed

elasticsearch/_async/client/query_ruleset.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -139,13 +139,15 @@ async def list(
139139
)
140140

141141
@_rewrite_parameters(
142-
body_name="query_ruleset",
142+
body_fields=True,
143143
)
144144
async def put(
145145
self,
146146
*,
147147
ruleset_id: str,
148-
query_ruleset: t.Mapping[str, t.Any],
148+
rules: t.Union[
149+
t.List[t.Mapping[str, t.Any]], t.Tuple[t.Mapping[str, t.Any], ...]
150+
],
149151
error_trace: t.Optional[bool] = None,
150152
filter_path: t.Optional[
151153
t.Union[str, t.Union[t.List[str], t.Tuple[str, ...]]]
@@ -160,14 +162,17 @@ async def put(
160162
161163
:param ruleset_id: The unique identifier of the query ruleset to be created or
162164
updated
163-
:param query_ruleset:
165+
:param rules:
164166
"""
165167
if ruleset_id in SKIP_IN_PATH:
166168
raise ValueError("Empty value passed for parameter 'ruleset_id'")
167-
if query_ruleset is None:
168-
raise ValueError("Empty value passed for parameter 'query_ruleset'")
169+
if rules is None:
170+
raise ValueError("Empty value passed for parameter 'rules'")
169171
__path = f"/_query_rules/{_quote(ruleset_id)}"
172+
__body: t.Dict[str, t.Any] = {}
170173
__query: t.Dict[str, t.Any] = {}
174+
if rules is not None:
175+
__body["rules"] = rules
171176
if error_trace is not None:
172177
__query["error_trace"] = error_trace
173178
if filter_path is not None:
@@ -176,7 +181,6 @@ async def put(
176181
__query["human"] = human
177182
if pretty is not None:
178183
__query["pretty"] = pretty
179-
__body = query_ruleset
180184
__headers = {"accept": "application/json", "content-type": "application/json"}
181185
return await self.perform_request( # type: ignore[return-value]
182186
"PUT", __path, params=__query, headers=__headers, body=__body

elasticsearch/_sync/client/query_ruleset.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -139,13 +139,15 @@ def list(
139139
)
140140

141141
@_rewrite_parameters(
142-
body_name="query_ruleset",
142+
body_fields=True,
143143
)
144144
def put(
145145
self,
146146
*,
147147
ruleset_id: str,
148-
query_ruleset: t.Mapping[str, t.Any],
148+
rules: t.Union[
149+
t.List[t.Mapping[str, t.Any]], t.Tuple[t.Mapping[str, t.Any], ...]
150+
],
149151
error_trace: t.Optional[bool] = None,
150152
filter_path: t.Optional[
151153
t.Union[str, t.Union[t.List[str], t.Tuple[str, ...]]]
@@ -160,14 +162,17 @@ def put(
160162
161163
:param ruleset_id: The unique identifier of the query ruleset to be created or
162164
updated
163-
:param query_ruleset:
165+
:param rules:
164166
"""
165167
if ruleset_id in SKIP_IN_PATH:
166168
raise ValueError("Empty value passed for parameter 'ruleset_id'")
167-
if query_ruleset is None:
168-
raise ValueError("Empty value passed for parameter 'query_ruleset'")
169+
if rules is None:
170+
raise ValueError("Empty value passed for parameter 'rules'")
169171
__path = f"/_query_rules/{_quote(ruleset_id)}"
172+
__body: t.Dict[str, t.Any] = {}
170173
__query: t.Dict[str, t.Any] = {}
174+
if rules is not None:
175+
__body["rules"] = rules
171176
if error_trace is not None:
172177
__query["error_trace"] = error_trace
173178
if filter_path is not None:
@@ -176,7 +181,6 @@ def put(
176181
__query["human"] = human
177182
if pretty is not None:
178183
__query["pretty"] = pretty
179-
__body = query_ruleset
180184
__headers = {"accept": "application/json", "content-type": "application/json"}
181185
return self.perform_request( # type: ignore[return-value]
182186
"PUT", __path, params=__query, headers=__headers, body=__body

0 commit comments

Comments
 (0)