13
13
# See the License for the specific language governing permissions and
14
14
# limitations under the License.
15
15
#
16
+ from google .api_core import gapic_v1
17
+ from google .api_core import retry as retries
18
+ from google .api_core import retry_async as retries_async
16
19
from typing import (
17
20
Any ,
18
21
AsyncIterator ,
22
25
Tuple ,
23
26
Optional ,
24
27
Iterator ,
28
+ Union ,
25
29
)
26
30
31
+ try :
32
+ OptionalRetry = Union [retries .Retry , gapic_v1 .method ._MethodDefault , None ]
33
+ OptionalAsyncRetry = Union [
34
+ retries_async .AsyncRetry , gapic_v1 .method ._MethodDefault , None
35
+ ]
36
+ except AttributeError : # pragma: NO COVER
37
+ OptionalRetry = Union [retries .Retry , object , None ] # type: ignore
38
+ OptionalAsyncRetry = Union [retries_async .AsyncRetry , object , None ] # type: ignore
39
+
27
40
from google .cloud .firestore_admin_v1 .types import field
28
41
from google .cloud .firestore_admin_v1 .types import firestore_admin
29
42
from google .cloud .firestore_admin_v1 .types import index
@@ -53,6 +66,8 @@ def __init__(
53
66
request : firestore_admin .ListIndexesRequest ,
54
67
response : firestore_admin .ListIndexesResponse ,
55
68
* ,
69
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
70
+ timeout : Union [float , object ] = gapic_v1 .method .DEFAULT ,
56
71
metadata : Sequence [Tuple [str , str ]] = ()
57
72
):
58
73
"""Instantiate the pager.
@@ -64,12 +79,17 @@ def __init__(
64
79
The initial request object.
65
80
response (google.cloud.firestore_admin_v1.types.ListIndexesResponse):
66
81
The initial response object.
82
+ retry (google.api_core.retry.Retry): Designation of what errors,
83
+ if any, should be retried.
84
+ timeout (float): The timeout for this request.
67
85
metadata (Sequence[Tuple[str, str]]): Strings which should be
68
86
sent along with the request as metadata.
69
87
"""
70
88
self ._method = method
71
89
self ._request = firestore_admin .ListIndexesRequest (request )
72
90
self ._response = response
91
+ self ._retry = retry
92
+ self ._timeout = timeout
73
93
self ._metadata = metadata
74
94
75
95
def __getattr__ (self , name : str ) -> Any :
@@ -80,7 +100,12 @@ def pages(self) -> Iterator[firestore_admin.ListIndexesResponse]:
80
100
yield self ._response
81
101
while self ._response .next_page_token :
82
102
self ._request .page_token = self ._response .next_page_token
83
- self ._response = self ._method (self ._request , metadata = self ._metadata )
103
+ self ._response = self ._method (
104
+ self ._request ,
105
+ retry = self ._retry ,
106
+ timeout = self ._timeout ,
107
+ metadata = self ._metadata ,
108
+ )
84
109
yield self ._response
85
110
86
111
def __iter__ (self ) -> Iterator [index .Index ]:
@@ -115,6 +140,8 @@ def __init__(
115
140
request : firestore_admin .ListIndexesRequest ,
116
141
response : firestore_admin .ListIndexesResponse ,
117
142
* ,
143
+ retry : OptionalAsyncRetry = gapic_v1 .method .DEFAULT ,
144
+ timeout : Union [float , object ] = gapic_v1 .method .DEFAULT ,
118
145
metadata : Sequence [Tuple [str , str ]] = ()
119
146
):
120
147
"""Instantiates the pager.
@@ -126,12 +153,17 @@ def __init__(
126
153
The initial request object.
127
154
response (google.cloud.firestore_admin_v1.types.ListIndexesResponse):
128
155
The initial response object.
156
+ retry (google.api_core.retry.AsyncRetry): Designation of what errors,
157
+ if any, should be retried.
158
+ timeout (float): The timeout for this request.
129
159
metadata (Sequence[Tuple[str, str]]): Strings which should be
130
160
sent along with the request as metadata.
131
161
"""
132
162
self ._method = method
133
163
self ._request = firestore_admin .ListIndexesRequest (request )
134
164
self ._response = response
165
+ self ._retry = retry
166
+ self ._timeout = timeout
135
167
self ._metadata = metadata
136
168
137
169
def __getattr__ (self , name : str ) -> Any :
@@ -142,7 +174,12 @@ async def pages(self) -> AsyncIterator[firestore_admin.ListIndexesResponse]:
142
174
yield self ._response
143
175
while self ._response .next_page_token :
144
176
self ._request .page_token = self ._response .next_page_token
145
- self ._response = await self ._method (self ._request , metadata = self ._metadata )
177
+ self ._response = await self ._method (
178
+ self ._request ,
179
+ retry = self ._retry ,
180
+ timeout = self ._timeout ,
181
+ metadata = self ._metadata ,
182
+ )
146
183
yield self ._response
147
184
148
185
def __aiter__ (self ) -> AsyncIterator [index .Index ]:
@@ -181,6 +218,8 @@ def __init__(
181
218
request : firestore_admin .ListFieldsRequest ,
182
219
response : firestore_admin .ListFieldsResponse ,
183
220
* ,
221
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
222
+ timeout : Union [float , object ] = gapic_v1 .method .DEFAULT ,
184
223
metadata : Sequence [Tuple [str , str ]] = ()
185
224
):
186
225
"""Instantiate the pager.
@@ -192,12 +231,17 @@ def __init__(
192
231
The initial request object.
193
232
response (google.cloud.firestore_admin_v1.types.ListFieldsResponse):
194
233
The initial response object.
234
+ retry (google.api_core.retry.Retry): Designation of what errors,
235
+ if any, should be retried.
236
+ timeout (float): The timeout for this request.
195
237
metadata (Sequence[Tuple[str, str]]): Strings which should be
196
238
sent along with the request as metadata.
197
239
"""
198
240
self ._method = method
199
241
self ._request = firestore_admin .ListFieldsRequest (request )
200
242
self ._response = response
243
+ self ._retry = retry
244
+ self ._timeout = timeout
201
245
self ._metadata = metadata
202
246
203
247
def __getattr__ (self , name : str ) -> Any :
@@ -208,7 +252,12 @@ def pages(self) -> Iterator[firestore_admin.ListFieldsResponse]:
208
252
yield self ._response
209
253
while self ._response .next_page_token :
210
254
self ._request .page_token = self ._response .next_page_token
211
- self ._response = self ._method (self ._request , metadata = self ._metadata )
255
+ self ._response = self ._method (
256
+ self ._request ,
257
+ retry = self ._retry ,
258
+ timeout = self ._timeout ,
259
+ metadata = self ._metadata ,
260
+ )
212
261
yield self ._response
213
262
214
263
def __iter__ (self ) -> Iterator [field .Field ]:
@@ -243,6 +292,8 @@ def __init__(
243
292
request : firestore_admin .ListFieldsRequest ,
244
293
response : firestore_admin .ListFieldsResponse ,
245
294
* ,
295
+ retry : OptionalAsyncRetry = gapic_v1 .method .DEFAULT ,
296
+ timeout : Union [float , object ] = gapic_v1 .method .DEFAULT ,
246
297
metadata : Sequence [Tuple [str , str ]] = ()
247
298
):
248
299
"""Instantiates the pager.
@@ -254,12 +305,17 @@ def __init__(
254
305
The initial request object.
255
306
response (google.cloud.firestore_admin_v1.types.ListFieldsResponse):
256
307
The initial response object.
308
+ retry (google.api_core.retry.AsyncRetry): Designation of what errors,
309
+ if any, should be retried.
310
+ timeout (float): The timeout for this request.
257
311
metadata (Sequence[Tuple[str, str]]): Strings which should be
258
312
sent along with the request as metadata.
259
313
"""
260
314
self ._method = method
261
315
self ._request = firestore_admin .ListFieldsRequest (request )
262
316
self ._response = response
317
+ self ._retry = retry
318
+ self ._timeout = timeout
263
319
self ._metadata = metadata
264
320
265
321
def __getattr__ (self , name : str ) -> Any :
@@ -270,7 +326,12 @@ async def pages(self) -> AsyncIterator[firestore_admin.ListFieldsResponse]:
270
326
yield self ._response
271
327
while self ._response .next_page_token :
272
328
self ._request .page_token = self ._response .next_page_token
273
- self ._response = await self ._method (self ._request , metadata = self ._metadata )
329
+ self ._response = await self ._method (
330
+ self ._request ,
331
+ retry = self ._retry ,
332
+ timeout = self ._timeout ,
333
+ metadata = self ._metadata ,
334
+ )
274
335
yield self ._response
275
336
276
337
def __aiter__ (self ) -> AsyncIterator [field .Field ]:
0 commit comments