|
19 | 19 | from typing import Dict, Sequence, Tuple, Type, Union |
20 | 20 | import pkg_resources |
21 | 21 |
|
22 | | -import google.api_core.client_options as ClientOptions # type: ignore |
| 22 | +from google.api_core.client_options import ClientOptions # type: ignore |
23 | 23 | from google.api_core import exceptions as core_exceptions # type: ignore |
24 | 24 | from google.api_core import gapic_v1 # type: ignore |
25 | 25 | from google.api_core import retry as retries # type: ignore |
26 | 26 | from google.auth import credentials as ga_credentials # type: ignore |
27 | 27 | from google.oauth2 import service_account # type: ignore |
28 | 28 |
|
| 29 | +OptionalRetry = Union[retries.Retry, object] |
| 30 | + |
29 | 31 | from google.cloud.iam_credentials_v1.types import common |
30 | 32 | from google.protobuf import duration_pb2 # type: ignore |
31 | 33 | from google.protobuf import timestamp_pb2 # type: ignore |
@@ -176,21 +178,21 @@ def __init__( |
176 | 178 |
|
177 | 179 | async def generate_access_token( |
178 | 180 | self, |
179 | | - request: common.GenerateAccessTokenRequest = None, |
| 181 | + request: Union[common.GenerateAccessTokenRequest, dict] = None, |
180 | 182 | *, |
181 | 183 | name: str = None, |
182 | 184 | delegates: Sequence[str] = None, |
183 | 185 | scope: Sequence[str] = None, |
184 | 186 | lifetime: duration_pb2.Duration = None, |
185 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 187 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
186 | 188 | timeout: float = None, |
187 | 189 | metadata: Sequence[Tuple[str, str]] = (), |
188 | 190 | ) -> common.GenerateAccessTokenResponse: |
189 | 191 | r"""Generates an OAuth 2.0 access token for a service |
190 | 192 | account. |
191 | 193 |
|
192 | 194 | Args: |
193 | | - request (:class:`google.cloud.iam_credentials_v1.types.GenerateAccessTokenRequest`): |
| 195 | + request (Union[google.cloud.iam_credentials_v1.types.GenerateAccessTokenRequest, dict]): |
194 | 196 | The request object. |
195 | 197 | name (:class:`str`): |
196 | 198 | Required. The resource name of the service account for |
@@ -308,21 +310,21 @@ async def generate_access_token( |
308 | 310 |
|
309 | 311 | async def generate_id_token( |
310 | 312 | self, |
311 | | - request: common.GenerateIdTokenRequest = None, |
| 313 | + request: Union[common.GenerateIdTokenRequest, dict] = None, |
312 | 314 | *, |
313 | 315 | name: str = None, |
314 | 316 | delegates: Sequence[str] = None, |
315 | 317 | audience: str = None, |
316 | 318 | include_email: bool = None, |
317 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 319 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
318 | 320 | timeout: float = None, |
319 | 321 | metadata: Sequence[Tuple[str, str]] = (), |
320 | 322 | ) -> common.GenerateIdTokenResponse: |
321 | 323 | r"""Generates an OpenID Connect ID token for a service |
322 | 324 | account. |
323 | 325 |
|
324 | 326 | Args: |
325 | | - request (:class:`google.cloud.iam_credentials_v1.types.GenerateIdTokenRequest`): |
| 327 | + request (Union[google.cloud.iam_credentials_v1.types.GenerateIdTokenRequest, dict]): |
326 | 328 | The request object. |
327 | 329 | name (:class:`str`): |
328 | 330 | Required. The resource name of the service account for |
@@ -434,20 +436,20 @@ async def generate_id_token( |
434 | 436 |
|
435 | 437 | async def sign_blob( |
436 | 438 | self, |
437 | | - request: common.SignBlobRequest = None, |
| 439 | + request: Union[common.SignBlobRequest, dict] = None, |
438 | 440 | *, |
439 | 441 | name: str = None, |
440 | 442 | delegates: Sequence[str] = None, |
441 | 443 | payload: bytes = None, |
442 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 444 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
443 | 445 | timeout: float = None, |
444 | 446 | metadata: Sequence[Tuple[str, str]] = (), |
445 | 447 | ) -> common.SignBlobResponse: |
446 | 448 | r"""Signs a blob using a service account's system-managed |
447 | 449 | private key. |
448 | 450 |
|
449 | 451 | Args: |
450 | | - request (:class:`google.cloud.iam_credentials_v1.types.SignBlobRequest`): |
| 452 | + request (Union[google.cloud.iam_credentials_v1.types.SignBlobRequest, dict]): |
451 | 453 | The request object. |
452 | 454 | name (:class:`str`): |
453 | 455 | Required. The resource name of the service account for |
@@ -546,20 +548,20 @@ async def sign_blob( |
546 | 548 |
|
547 | 549 | async def sign_jwt( |
548 | 550 | self, |
549 | | - request: common.SignJwtRequest = None, |
| 551 | + request: Union[common.SignJwtRequest, dict] = None, |
550 | 552 | *, |
551 | 553 | name: str = None, |
552 | 554 | delegates: Sequence[str] = None, |
553 | 555 | payload: str = None, |
554 | | - retry: retries.Retry = gapic_v1.method.DEFAULT, |
| 556 | + retry: OptionalRetry = gapic_v1.method.DEFAULT, |
555 | 557 | timeout: float = None, |
556 | 558 | metadata: Sequence[Tuple[str, str]] = (), |
557 | 559 | ) -> common.SignJwtResponse: |
558 | 560 | r"""Signs a JWT using a service account's system-managed |
559 | 561 | private key. |
560 | 562 |
|
561 | 563 | Args: |
562 | | - request (:class:`google.cloud.iam_credentials_v1.types.SignJwtRequest`): |
| 564 | + request (Union[google.cloud.iam_credentials_v1.types.SignJwtRequest, dict]): |
563 | 565 | The request object. |
564 | 566 | name (:class:`str`): |
565 | 567 | Required. The resource name of the service account for |
|
0 commit comments