|
34 | 34 | from google.api_core.client_options import ClientOptions
|
35 | 35 | from google.auth import credentials as ga_credentials # type: ignore
|
36 | 36 | from google.oauth2 import service_account # type: ignore
|
37 |
| -import pkg_resources |
| 37 | + |
| 38 | +from google.cloud.dataflow_v1beta3 import gapic_version as package_version |
38 | 39 |
|
39 | 40 | try:
|
40 | 41 | OptionalRetry = Union[retries.Retry, gapic_v1.method._MethodDefault]
|
@@ -218,7 +219,7 @@ async def create_job(
|
218 | 219 | request: Optional[Union[jobs.CreateJobRequest, dict]] = None,
|
219 | 220 | *,
|
220 | 221 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
221 |
| - timeout: Optional[float] = None, |
| 222 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
222 | 223 | metadata: Sequence[Tuple[str, str]] = (),
|
223 | 224 | ) -> jobs.Job:
|
224 | 225 | r"""Creates a Cloud Dataflow job.
|
@@ -308,7 +309,7 @@ async def get_job(
|
308 | 309 | request: Optional[Union[jobs.GetJobRequest, dict]] = None,
|
309 | 310 | *,
|
310 | 311 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
311 |
| - timeout: Optional[float] = None, |
| 312 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
312 | 313 | metadata: Sequence[Tuple[str, str]] = (),
|
313 | 314 | ) -> jobs.Job:
|
314 | 315 | r"""Gets the state of the specified Cloud Dataflow job.
|
@@ -399,7 +400,7 @@ async def update_job(
|
399 | 400 | request: Optional[Union[jobs.UpdateJobRequest, dict]] = None,
|
400 | 401 | *,
|
401 | 402 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
402 |
| - timeout: Optional[float] = None, |
| 403 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
403 | 404 | metadata: Sequence[Tuple[str, str]] = (),
|
404 | 405 | ) -> jobs.Job:
|
405 | 406 | r"""Updates the state of an existing Cloud Dataflow job.
|
@@ -491,7 +492,7 @@ async def list_jobs(
|
491 | 492 | request: Optional[Union[jobs.ListJobsRequest, dict]] = None,
|
492 | 493 | *,
|
493 | 494 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
494 |
| - timeout: Optional[float] = None, |
| 495 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
495 | 496 | metadata: Sequence[Tuple[str, str]] = (),
|
496 | 497 | ) -> pagers.ListJobsAsyncPager:
|
497 | 498 | r"""List the jobs of a project.
|
@@ -602,7 +603,7 @@ async def aggregated_list_jobs(
|
602 | 603 | request: Optional[Union[jobs.ListJobsRequest, dict]] = None,
|
603 | 604 | *,
|
604 | 605 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
605 |
| - timeout: Optional[float] = None, |
| 606 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
606 | 607 | metadata: Sequence[Tuple[str, str]] = (),
|
607 | 608 | ) -> pagers.AggregatedListJobsAsyncPager:
|
608 | 609 | r"""List the jobs of a project across all regions.
|
@@ -702,7 +703,7 @@ async def check_active_jobs(
|
702 | 703 | request: Optional[Union[jobs.CheckActiveJobsRequest, dict]] = None,
|
703 | 704 | *,
|
704 | 705 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
705 |
| - timeout: Optional[float] = None, |
| 706 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
706 | 707 | metadata: Sequence[Tuple[str, str]] = (),
|
707 | 708 | ) -> jobs.CheckActiveJobsResponse:
|
708 | 709 | r"""Check for existence of active jobs in the given
|
@@ -774,7 +775,7 @@ async def snapshot_job(
|
774 | 775 | request: Optional[Union[jobs.SnapshotJobRequest, dict]] = None,
|
775 | 776 | *,
|
776 | 777 | retry: OptionalRetry = gapic_v1.method.DEFAULT,
|
777 |
| - timeout: Optional[float] = None, |
| 778 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
778 | 779 | metadata: Sequence[Tuple[str, str]] = (),
|
779 | 780 | ) -> snapshots.Snapshot:
|
780 | 781 | r"""Snapshot the state of a streaming job.
|
@@ -859,14 +860,9 @@ async def __aexit__(self, exc_type, exc, tb):
|
859 | 860 | await self.transport.close()
|
860 | 861 |
|
861 | 862 |
|
862 |
| -try: |
863 |
| - DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( |
864 |
| - gapic_version=pkg_resources.get_distribution( |
865 |
| - "google-cloud-dataflow-client", |
866 |
| - ).version, |
867 |
| - ) |
868 |
| -except pkg_resources.DistributionNotFound: |
869 |
| - DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo() |
| 863 | +DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( |
| 864 | + gapic_version=package_version.__version__ |
| 865 | +) |
870 | 866 |
|
871 | 867 |
|
872 | 868 | __all__ = ("JobsV1Beta3AsyncClient",)
|
0 commit comments