2020import pkg_resources
2121import warnings
2222
23- import google .api_core .client_options as ClientOptions # type: ignore
23+ from google .api_core .client_options import ClientOptions # type: ignore
2424from google .api_core import exceptions as core_exceptions # type: ignore
2525from google .api_core import gapic_v1 # type: ignore
2626from google .api_core import retry as retries # type: ignore
2727from google .auth import credentials as ga_credentials # type: ignore
2828from google .oauth2 import service_account # type: ignore
2929
30+ OptionalRetry = Union [retries .Retry , object ]
31+
3032from google .cloud .bigquery_datatransfer_v1 .services .data_transfer_service import pagers
3133from google .cloud .bigquery_datatransfer_v1 .types import datatransfer
3234from google .cloud .bigquery_datatransfer_v1 .types import transfer
@@ -183,18 +185,18 @@ def __init__(
183185
184186 async def get_data_source (
185187 self ,
186- request : datatransfer .GetDataSourceRequest = None ,
188+ request : Union [ datatransfer .GetDataSourceRequest , dict ] = None ,
187189 * ,
188190 name : str = None ,
189- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
191+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
190192 timeout : float = None ,
191193 metadata : Sequence [Tuple [str , str ]] = (),
192194 ) -> datatransfer .DataSource :
193195 r"""Retrieves a supported data source and returns its
194196 settings, which can be used for UI rendering.
195197
196198 Args:
197- request (:class:` google.cloud.bigquery_datatransfer_v1.types.GetDataSourceRequest` ):
199+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.GetDataSourceRequest, dict] ):
198200 The request object. A request to get data source info.
199201 name (:class:`str`):
200202 Required. The field will contain name of the resource
@@ -268,18 +270,18 @@ async def get_data_source(
268270
269271 async def list_data_sources (
270272 self ,
271- request : datatransfer .ListDataSourcesRequest = None ,
273+ request : Union [ datatransfer .ListDataSourcesRequest , dict ] = None ,
272274 * ,
273275 parent : str = None ,
274- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
276+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
275277 timeout : float = None ,
276278 metadata : Sequence [Tuple [str , str ]] = (),
277279 ) -> pagers .ListDataSourcesAsyncPager :
278280 r"""Lists supported data sources and returns their
279281 settings, which can be used for UI rendering.
280282
281283 Args:
282- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListDataSourcesRequest` ):
284+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListDataSourcesRequest, dict] ):
283285 The request object. Request to list supported data
284286 sources and their data transfer settings.
285287 parent (:class:`str`):
@@ -361,18 +363,18 @@ async def list_data_sources(
361363
362364 async def create_transfer_config (
363365 self ,
364- request : datatransfer .CreateTransferConfigRequest = None ,
366+ request : Union [ datatransfer .CreateTransferConfigRequest , dict ] = None ,
365367 * ,
366368 parent : str = None ,
367369 transfer_config : transfer .TransferConfig = None ,
368- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
370+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
369371 timeout : float = None ,
370372 metadata : Sequence [Tuple [str , str ]] = (),
371373 ) -> transfer .TransferConfig :
372374 r"""Creates a new data transfer configuration.
373375
374376 Args:
375- request (:class:` google.cloud.bigquery_datatransfer_v1.types.CreateTransferConfigRequest` ):
377+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.CreateTransferConfigRequest, dict] ):
376378 The request object. A request to create a data transfer
377379 configuration. If new credentials are needed for this
378380 transfer configuration, an authorization code must be
@@ -458,19 +460,19 @@ async def create_transfer_config(
458460
459461 async def update_transfer_config (
460462 self ,
461- request : datatransfer .UpdateTransferConfigRequest = None ,
463+ request : Union [ datatransfer .UpdateTransferConfigRequest , dict ] = None ,
462464 * ,
463465 transfer_config : transfer .TransferConfig = None ,
464466 update_mask : field_mask_pb2 .FieldMask = None ,
465- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
467+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
466468 timeout : float = None ,
467469 metadata : Sequence [Tuple [str , str ]] = (),
468470 ) -> transfer .TransferConfig :
469471 r"""Updates a data transfer configuration.
470472 All fields must be set, even if they are not updated.
471473
472474 Args:
473- request (:class:` google.cloud.bigquery_datatransfer_v1.types.UpdateTransferConfigRequest` ):
475+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.UpdateTransferConfigRequest, dict] ):
474476 The request object. A request to update a transfer
475477 configuration. To update the user id of the transfer
476478 configuration, an authorization code needs to be
@@ -550,18 +552,18 @@ async def update_transfer_config(
550552
551553 async def delete_transfer_config (
552554 self ,
553- request : datatransfer .DeleteTransferConfigRequest = None ,
555+ request : Union [ datatransfer .DeleteTransferConfigRequest , dict ] = None ,
554556 * ,
555557 name : str = None ,
556- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
558+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
557559 timeout : float = None ,
558560 metadata : Sequence [Tuple [str , str ]] = (),
559561 ) -> None :
560562 r"""Deletes a data transfer configuration,
561563 including any associated transfer runs and logs.
562564
563565 Args:
564- request (:class:` google.cloud.bigquery_datatransfer_v1.types.DeleteTransferConfigRequest` ):
566+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.DeleteTransferConfigRequest, dict] ):
565567 The request object. A request to delete data transfer
566568 information. All associated transfer runs and log
567569 messages will be deleted as well.
@@ -628,17 +630,17 @@ async def delete_transfer_config(
628630
629631 async def get_transfer_config (
630632 self ,
631- request : datatransfer .GetTransferConfigRequest = None ,
633+ request : Union [ datatransfer .GetTransferConfigRequest , dict ] = None ,
632634 * ,
633635 name : str = None ,
634- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
636+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
635637 timeout : float = None ,
636638 metadata : Sequence [Tuple [str , str ]] = (),
637639 ) -> transfer .TransferConfig :
638640 r"""Returns information about a data transfer config.
639641
640642 Args:
641- request (:class:` google.cloud.bigquery_datatransfer_v1.types.GetTransferConfigRequest` ):
643+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.GetTransferConfigRequest, dict] ):
642644 The request object. A request to get data transfer
643645 information.
644646 name (:class:`str`):
@@ -717,18 +719,18 @@ async def get_transfer_config(
717719
718720 async def list_transfer_configs (
719721 self ,
720- request : datatransfer .ListTransferConfigsRequest = None ,
722+ request : Union [ datatransfer .ListTransferConfigsRequest , dict ] = None ,
721723 * ,
722724 parent : str = None ,
723- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
725+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
724726 timeout : float = None ,
725727 metadata : Sequence [Tuple [str , str ]] = (),
726728 ) -> pagers .ListTransferConfigsAsyncPager :
727729 r"""Returns information about all transfer configs owned
728730 by a project in the specified location.
729731
730732 Args:
731- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListTransferConfigsRequest` ):
733+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListTransferConfigsRequest, dict] ):
732734 The request object. A request to list data transfers
733735 configured for a BigQuery project.
734736 parent (:class:`str`):
@@ -809,12 +811,12 @@ async def list_transfer_configs(
809811
810812 async def schedule_transfer_runs (
811813 self ,
812- request : datatransfer .ScheduleTransferRunsRequest = None ,
814+ request : Union [ datatransfer .ScheduleTransferRunsRequest , dict ] = None ,
813815 * ,
814816 parent : str = None ,
815817 start_time : timestamp_pb2 .Timestamp = None ,
816818 end_time : timestamp_pb2 .Timestamp = None ,
817- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
819+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
818820 timeout : float = None ,
819821 metadata : Sequence [Tuple [str , str ]] = (),
820822 ) -> datatransfer .ScheduleTransferRunsResponse :
@@ -825,7 +827,7 @@ async def schedule_transfer_runs(
825827 StartManualTransferRuns instead.
826828
827829 Args:
828- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ScheduleTransferRunsRequest` ):
830+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ScheduleTransferRunsRequest, dict] ):
829831 The request object. A request to schedule transfer runs
830832 for a time range.
831833 parent (:class:`str`):
@@ -910,9 +912,9 @@ async def schedule_transfer_runs(
910912
911913 async def start_manual_transfer_runs (
912914 self ,
913- request : datatransfer .StartManualTransferRunsRequest = None ,
915+ request : Union [ datatransfer .StartManualTransferRunsRequest , dict ] = None ,
914916 * ,
915- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
917+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
916918 timeout : float = None ,
917919 metadata : Sequence [Tuple [str , str ]] = (),
918920 ) -> datatransfer .StartManualTransferRunsResponse :
@@ -922,7 +924,7 @@ async def start_manual_transfer_runs(
922924 and end_time (exclusive), or for a specific run_time.
923925
924926 Args:
925- request (:class:` google.cloud.bigquery_datatransfer_v1.types.StartManualTransferRunsRequest` ):
927+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.StartManualTransferRunsRequest, dict] ):
926928 The request object. A request to start manual transfer
927929 runs.
928930 retry (google.api_core.retry.Retry): Designation of what errors, if any,
@@ -962,18 +964,18 @@ async def start_manual_transfer_runs(
962964
963965 async def get_transfer_run (
964966 self ,
965- request : datatransfer .GetTransferRunRequest = None ,
967+ request : Union [ datatransfer .GetTransferRunRequest , dict ] = None ,
966968 * ,
967969 name : str = None ,
968- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
970+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
969971 timeout : float = None ,
970972 metadata : Sequence [Tuple [str , str ]] = (),
971973 ) -> transfer .TransferRun :
972974 r"""Returns information about the particular transfer
973975 run.
974976
975977 Args:
976- request (:class:` google.cloud.bigquery_datatransfer_v1.types.GetTransferRunRequest` ):
978+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.GetTransferRunRequest, dict] ):
977979 The request object. A request to get data transfer run
978980 information.
979981 name (:class:`str`):
@@ -1045,17 +1047,17 @@ async def get_transfer_run(
10451047
10461048 async def delete_transfer_run (
10471049 self ,
1048- request : datatransfer .DeleteTransferRunRequest = None ,
1050+ request : Union [ datatransfer .DeleteTransferRunRequest , dict ] = None ,
10491051 * ,
10501052 name : str = None ,
1051- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1053+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
10521054 timeout : float = None ,
10531055 metadata : Sequence [Tuple [str , str ]] = (),
10541056 ) -> None :
10551057 r"""Deletes the specified transfer run.
10561058
10571059 Args:
1058- request (:class:` google.cloud.bigquery_datatransfer_v1.types.DeleteTransferRunRequest` ):
1060+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.DeleteTransferRunRequest, dict] ):
10591061 The request object. A request to delete data transfer
10601062 run information.
10611063 name (:class:`str`):
@@ -1122,17 +1124,17 @@ async def delete_transfer_run(
11221124
11231125 async def list_transfer_runs (
11241126 self ,
1125- request : datatransfer .ListTransferRunsRequest = None ,
1127+ request : Union [ datatransfer .ListTransferRunsRequest , dict ] = None ,
11261128 * ,
11271129 parent : str = None ,
1128- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1130+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
11291131 timeout : float = None ,
11301132 metadata : Sequence [Tuple [str , str ]] = (),
11311133 ) -> pagers .ListTransferRunsAsyncPager :
11321134 r"""Returns information about running and completed jobs.
11331135
11341136 Args:
1135- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListTransferRunsRequest` ):
1137+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListTransferRunsRequest, dict] ):
11361138 The request object. A request to list data transfer
11371139 runs. UI can use this method to show/filter specific
11381140 data transfer runs. The data source can use this method
@@ -1217,18 +1219,18 @@ async def list_transfer_runs(
12171219
12181220 async def list_transfer_logs (
12191221 self ,
1220- request : datatransfer .ListTransferLogsRequest = None ,
1222+ request : Union [ datatransfer .ListTransferLogsRequest , dict ] = None ,
12211223 * ,
12221224 parent : str = None ,
1223- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1225+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
12241226 timeout : float = None ,
12251227 metadata : Sequence [Tuple [str , str ]] = (),
12261228 ) -> pagers .ListTransferLogsAsyncPager :
12271229 r"""Returns user facing log messages for the data
12281230 transfer run.
12291231
12301232 Args:
1231- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListTransferLogsRequest` ):
1233+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListTransferLogsRequest, dict] ):
12321234 The request object. A request to get user facing log
12331235 messages associated with data transfer run.
12341236 parent (:class:`str`):
@@ -1310,10 +1312,10 @@ async def list_transfer_logs(
13101312
13111313 async def check_valid_creds (
13121314 self ,
1313- request : datatransfer .CheckValidCredsRequest = None ,
1315+ request : Union [ datatransfer .CheckValidCredsRequest , dict ] = None ,
13141316 * ,
13151317 name : str = None ,
1316- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1318+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
13171319 timeout : float = None ,
13181320 metadata : Sequence [Tuple [str , str ]] = (),
13191321 ) -> datatransfer .CheckValidCredsResponse :
@@ -1326,7 +1328,7 @@ async def check_valid_creds(
13261328 can create a transfer config.
13271329
13281330 Args:
1329- request (:class:` google.cloud.bigquery_datatransfer_v1.types.CheckValidCredsRequest` ):
1331+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.CheckValidCredsRequest, dict] ):
13301332 The request object. A request to determine whether the
13311333 user has valid credentials. This method is used to limit
13321334 the number of OAuth popups in the user interface. The
0 commit comments