1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
# Licensed to Elasticsearch B.V. under one or more contributor
# license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright
# ownership. Elasticsearch B.V. licenses this file to you under
# the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
from typing import Any, Dict, Literal, Mapping, Sequence, Union
from elastic_transport.client_utils import DEFAULT, DefaultType
from . import Query
from .document_base import InstrumentedField
from .utils import AttrDict
PipeSeparatedFlags = str
{% for k in classes %}
class {{ k.name }}({{ k.parent if k.parent else "AttrDict[Any]" }}):
{% if k.docstring or k.args %}
"""
{% for line in k.docstring %}
{{ line }}
{% endfor %}
{% if k.args %}
{% if k.docstring %}
{% endif %}
{% endif %}
{% for arg in k.args %}
{% for line in arg.doc %}
{{ line }}
{% endfor %}
{% endfor %}
"""
{% for arg in k.args %}
{% if arg.name not in ["keys", "items"] %}
{{ arg.name }}: {{ arg.type }}
{% else %}
{{ arg.name }}: {{ arg.type }} # type: ignore[assignment]
{% endif %}
{% endfor %}
{% if not k.for_response %}
def __init__(
self,
{% for arg in k.args %}
{% if arg.positional %}
{{ arg.name }}: {{ arg.type }} = DEFAULT,
{% endif %}
{% endfor %}
{% if k.args and not k.args[-1].positional %}
*,
{% endif %}
{% for arg in k.args %}
{% if not arg.positional %}
{{ arg.name }}: {{ arg.type }} = DEFAULT,
{% endif %}
{% endfor %}
**kwargs: Any
):
{% if k.is_single_field %}
if _field is not DEFAULT:
kwargs[str(_field)] = _value
{% elif k.is_multi_field %}
if _fields is not DEFAULT:
for field, value in _fields.items():
kwargs[str(field)] = value
{% endif %}
{% for arg in k.args %}
{% if not arg.positional %}
if {{ arg.name }} is not DEFAULT:
{% if "InstrumentedField" in arg.type %}
kwargs["{{ arg.name }}"] = str({{ arg.name }})
{% else %}
kwargs["{{ arg.name }}"] = {{ arg.name }}
{% endif %}
{% endif %}
{% endfor %}
{% if k.parent %}
super().__init__(**kwargs)
{% else %}
super().__init__(kwargs)
{% endif %}
{% endif %}
{% if k.buckets_as_dict %}
@property
def buckets_as_dict(self) -> Mapping[str, {{ k.buckets_as_dict }}]:
return self.buckets # type: ignore[return-value]
{% endif %}
{% else %}
pass
{% endif %}
{% endfor %}
|
