20
20
import pkg_resources
21
21
import warnings
22
22
23
- import google .api_core .client_options as ClientOptions # type: ignore
23
+ from google .api_core .client_options import ClientOptions # type: ignore
24
24
from google .api_core import exceptions as core_exceptions # type: ignore
25
25
from google .api_core import gapic_v1 # type: ignore
26
26
from google .api_core import retry as retries # type: ignore
27
27
from google .auth import credentials as ga_credentials # type: ignore
28
28
from google .oauth2 import service_account # type: ignore
29
29
30
+ OptionalRetry = Union [retries .Retry , object ]
31
+
30
32
from google .cloud .bigquery_datatransfer_v1 .services .data_transfer_service import pagers
31
33
from google .cloud .bigquery_datatransfer_v1 .types import datatransfer
32
34
from google .cloud .bigquery_datatransfer_v1 .types import transfer
@@ -183,18 +185,18 @@ def __init__(
183
185
184
186
async def get_data_source (
185
187
self ,
186
- request : datatransfer .GetDataSourceRequest = None ,
188
+ request : Union [ datatransfer .GetDataSourceRequest , dict ] = None ,
187
189
* ,
188
190
name : str = None ,
189
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
191
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
190
192
timeout : float = None ,
191
193
metadata : Sequence [Tuple [str , str ]] = (),
192
194
) -> datatransfer .DataSource :
193
195
r"""Retrieves a supported data source and returns its
194
196
settings, which can be used for UI rendering.
195
197
196
198
Args:
197
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.GetDataSourceRequest` ):
199
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.GetDataSourceRequest, dict] ):
198
200
The request object. A request to get data source info.
199
201
name (:class:`str`):
200
202
Required. The field will contain name of the resource
@@ -268,18 +270,18 @@ async def get_data_source(
268
270
269
271
async def list_data_sources (
270
272
self ,
271
- request : datatransfer .ListDataSourcesRequest = None ,
273
+ request : Union [ datatransfer .ListDataSourcesRequest , dict ] = None ,
272
274
* ,
273
275
parent : str = None ,
274
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
276
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
275
277
timeout : float = None ,
276
278
metadata : Sequence [Tuple [str , str ]] = (),
277
279
) -> pagers .ListDataSourcesAsyncPager :
278
280
r"""Lists supported data sources and returns their
279
281
settings, which can be used for UI rendering.
280
282
281
283
Args:
282
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListDataSourcesRequest` ):
284
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListDataSourcesRequest, dict] ):
283
285
The request object. Request to list supported data
284
286
sources and their data transfer settings.
285
287
parent (:class:`str`):
@@ -361,18 +363,18 @@ async def list_data_sources(
361
363
362
364
async def create_transfer_config (
363
365
self ,
364
- request : datatransfer .CreateTransferConfigRequest = None ,
366
+ request : Union [ datatransfer .CreateTransferConfigRequest , dict ] = None ,
365
367
* ,
366
368
parent : str = None ,
367
369
transfer_config : transfer .TransferConfig = None ,
368
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
370
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
369
371
timeout : float = None ,
370
372
metadata : Sequence [Tuple [str , str ]] = (),
371
373
) -> transfer .TransferConfig :
372
374
r"""Creates a new data transfer configuration.
373
375
374
376
Args:
375
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.CreateTransferConfigRequest` ):
377
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.CreateTransferConfigRequest, dict] ):
376
378
The request object. A request to create a data transfer
377
379
configuration. If new credentials are needed for this
378
380
transfer configuration, an authorization code must be
@@ -458,19 +460,19 @@ async def create_transfer_config(
458
460
459
461
async def update_transfer_config (
460
462
self ,
461
- request : datatransfer .UpdateTransferConfigRequest = None ,
463
+ request : Union [ datatransfer .UpdateTransferConfigRequest , dict ] = None ,
462
464
* ,
463
465
transfer_config : transfer .TransferConfig = None ,
464
466
update_mask : field_mask_pb2 .FieldMask = None ,
465
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
467
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
466
468
timeout : float = None ,
467
469
metadata : Sequence [Tuple [str , str ]] = (),
468
470
) -> transfer .TransferConfig :
469
471
r"""Updates a data transfer configuration.
470
472
All fields must be set, even if they are not updated.
471
473
472
474
Args:
473
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.UpdateTransferConfigRequest` ):
475
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.UpdateTransferConfigRequest, dict] ):
474
476
The request object. A request to update a transfer
475
477
configuration. To update the user id of the transfer
476
478
configuration, an authorization code needs to be
@@ -550,18 +552,18 @@ async def update_transfer_config(
550
552
551
553
async def delete_transfer_config (
552
554
self ,
553
- request : datatransfer .DeleteTransferConfigRequest = None ,
555
+ request : Union [ datatransfer .DeleteTransferConfigRequest , dict ] = None ,
554
556
* ,
555
557
name : str = None ,
556
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
558
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
557
559
timeout : float = None ,
558
560
metadata : Sequence [Tuple [str , str ]] = (),
559
561
) -> None :
560
562
r"""Deletes a data transfer configuration,
561
563
including any associated transfer runs and logs.
562
564
563
565
Args:
564
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.DeleteTransferConfigRequest` ):
566
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.DeleteTransferConfigRequest, dict] ):
565
567
The request object. A request to delete data transfer
566
568
information. All associated transfer runs and log
567
569
messages will be deleted as well.
@@ -628,17 +630,17 @@ async def delete_transfer_config(
628
630
629
631
async def get_transfer_config (
630
632
self ,
631
- request : datatransfer .GetTransferConfigRequest = None ,
633
+ request : Union [ datatransfer .GetTransferConfigRequest , dict ] = None ,
632
634
* ,
633
635
name : str = None ,
634
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
636
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
635
637
timeout : float = None ,
636
638
metadata : Sequence [Tuple [str , str ]] = (),
637
639
) -> transfer .TransferConfig :
638
640
r"""Returns information about a data transfer config.
639
641
640
642
Args:
641
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.GetTransferConfigRequest` ):
643
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.GetTransferConfigRequest, dict] ):
642
644
The request object. A request to get data transfer
643
645
information.
644
646
name (:class:`str`):
@@ -717,18 +719,18 @@ async def get_transfer_config(
717
719
718
720
async def list_transfer_configs (
719
721
self ,
720
- request : datatransfer .ListTransferConfigsRequest = None ,
722
+ request : Union [ datatransfer .ListTransferConfigsRequest , dict ] = None ,
721
723
* ,
722
724
parent : str = None ,
723
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
725
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
724
726
timeout : float = None ,
725
727
metadata : Sequence [Tuple [str , str ]] = (),
726
728
) -> pagers .ListTransferConfigsAsyncPager :
727
729
r"""Returns information about all transfer configs owned
728
730
by a project in the specified location.
729
731
730
732
Args:
731
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListTransferConfigsRequest` ):
733
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListTransferConfigsRequest, dict] ):
732
734
The request object. A request to list data transfers
733
735
configured for a BigQuery project.
734
736
parent (:class:`str`):
@@ -809,12 +811,12 @@ async def list_transfer_configs(
809
811
810
812
async def schedule_transfer_runs (
811
813
self ,
812
- request : datatransfer .ScheduleTransferRunsRequest = None ,
814
+ request : Union [ datatransfer .ScheduleTransferRunsRequest , dict ] = None ,
813
815
* ,
814
816
parent : str = None ,
815
817
start_time : timestamp_pb2 .Timestamp = None ,
816
818
end_time : timestamp_pb2 .Timestamp = None ,
817
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
819
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
818
820
timeout : float = None ,
819
821
metadata : Sequence [Tuple [str , str ]] = (),
820
822
) -> datatransfer .ScheduleTransferRunsResponse :
@@ -825,7 +827,7 @@ async def schedule_transfer_runs(
825
827
StartManualTransferRuns instead.
826
828
827
829
Args:
828
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ScheduleTransferRunsRequest` ):
830
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ScheduleTransferRunsRequest, dict] ):
829
831
The request object. A request to schedule transfer runs
830
832
for a time range.
831
833
parent (:class:`str`):
@@ -910,9 +912,9 @@ async def schedule_transfer_runs(
910
912
911
913
async def start_manual_transfer_runs (
912
914
self ,
913
- request : datatransfer .StartManualTransferRunsRequest = None ,
915
+ request : Union [ datatransfer .StartManualTransferRunsRequest , dict ] = None ,
914
916
* ,
915
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
917
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
916
918
timeout : float = None ,
917
919
metadata : Sequence [Tuple [str , str ]] = (),
918
920
) -> datatransfer .StartManualTransferRunsResponse :
@@ -922,7 +924,7 @@ async def start_manual_transfer_runs(
922
924
and end_time (exclusive), or for a specific run_time.
923
925
924
926
Args:
925
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.StartManualTransferRunsRequest` ):
927
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.StartManualTransferRunsRequest, dict] ):
926
928
The request object. A request to start manual transfer
927
929
runs.
928
930
retry (google.api_core.retry.Retry): Designation of what errors, if any,
@@ -962,18 +964,18 @@ async def start_manual_transfer_runs(
962
964
963
965
async def get_transfer_run (
964
966
self ,
965
- request : datatransfer .GetTransferRunRequest = None ,
967
+ request : Union [ datatransfer .GetTransferRunRequest , dict ] = None ,
966
968
* ,
967
969
name : str = None ,
968
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
970
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
969
971
timeout : float = None ,
970
972
metadata : Sequence [Tuple [str , str ]] = (),
971
973
) -> transfer .TransferRun :
972
974
r"""Returns information about the particular transfer
973
975
run.
974
976
975
977
Args:
976
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.GetTransferRunRequest` ):
978
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.GetTransferRunRequest, dict] ):
977
979
The request object. A request to get data transfer run
978
980
information.
979
981
name (:class:`str`):
@@ -1045,17 +1047,17 @@ async def get_transfer_run(
1045
1047
1046
1048
async def delete_transfer_run (
1047
1049
self ,
1048
- request : datatransfer .DeleteTransferRunRequest = None ,
1050
+ request : Union [ datatransfer .DeleteTransferRunRequest , dict ] = None ,
1049
1051
* ,
1050
1052
name : str = None ,
1051
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1053
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
1052
1054
timeout : float = None ,
1053
1055
metadata : Sequence [Tuple [str , str ]] = (),
1054
1056
) -> None :
1055
1057
r"""Deletes the specified transfer run.
1056
1058
1057
1059
Args:
1058
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.DeleteTransferRunRequest` ):
1060
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.DeleteTransferRunRequest, dict] ):
1059
1061
The request object. A request to delete data transfer
1060
1062
run information.
1061
1063
name (:class:`str`):
@@ -1122,17 +1124,17 @@ async def delete_transfer_run(
1122
1124
1123
1125
async def list_transfer_runs (
1124
1126
self ,
1125
- request : datatransfer .ListTransferRunsRequest = None ,
1127
+ request : Union [ datatransfer .ListTransferRunsRequest , dict ] = None ,
1126
1128
* ,
1127
1129
parent : str = None ,
1128
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1130
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
1129
1131
timeout : float = None ,
1130
1132
metadata : Sequence [Tuple [str , str ]] = (),
1131
1133
) -> pagers .ListTransferRunsAsyncPager :
1132
1134
r"""Returns information about running and completed jobs.
1133
1135
1134
1136
Args:
1135
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListTransferRunsRequest` ):
1137
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListTransferRunsRequest, dict] ):
1136
1138
The request object. A request to list data transfer
1137
1139
runs. UI can use this method to show/filter specific
1138
1140
data transfer runs. The data source can use this method
@@ -1217,18 +1219,18 @@ async def list_transfer_runs(
1217
1219
1218
1220
async def list_transfer_logs (
1219
1221
self ,
1220
- request : datatransfer .ListTransferLogsRequest = None ,
1222
+ request : Union [ datatransfer .ListTransferLogsRequest , dict ] = None ,
1221
1223
* ,
1222
1224
parent : str = None ,
1223
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1225
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
1224
1226
timeout : float = None ,
1225
1227
metadata : Sequence [Tuple [str , str ]] = (),
1226
1228
) -> pagers .ListTransferLogsAsyncPager :
1227
1229
r"""Returns user facing log messages for the data
1228
1230
transfer run.
1229
1231
1230
1232
Args:
1231
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.ListTransferLogsRequest` ):
1233
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.ListTransferLogsRequest, dict] ):
1232
1234
The request object. A request to get user facing log
1233
1235
messages associated with data transfer run.
1234
1236
parent (:class:`str`):
@@ -1310,10 +1312,10 @@ async def list_transfer_logs(
1310
1312
1311
1313
async def check_valid_creds (
1312
1314
self ,
1313
- request : datatransfer .CheckValidCredsRequest = None ,
1315
+ request : Union [ datatransfer .CheckValidCredsRequest , dict ] = None ,
1314
1316
* ,
1315
1317
name : str = None ,
1316
- retry : retries . Retry = gapic_v1 .method .DEFAULT ,
1318
+ retry : OptionalRetry = gapic_v1 .method .DEFAULT ,
1317
1319
timeout : float = None ,
1318
1320
metadata : Sequence [Tuple [str , str ]] = (),
1319
1321
) -> datatransfer .CheckValidCredsResponse :
@@ -1326,7 +1328,7 @@ async def check_valid_creds(
1326
1328
can create a transfer config.
1327
1329
1328
1330
Args:
1329
- request (:class:` google.cloud.bigquery_datatransfer_v1.types.CheckValidCredsRequest` ):
1331
+ request (Union[ google.cloud.bigquery_datatransfer_v1.types.CheckValidCredsRequest, dict] ):
1330
1332
The request object. A request to determine whether the
1331
1333
user has valid credentials. This method is used to limit
1332
1334
the number of OAuth popups in the user interface. The
0 commit comments