diff --git a/google/cloud/bigquery_reservation_v1/services/reservation_service/client.py b/google/cloud/bigquery_reservation_v1/services/reservation_service/client.py index 0a659a66..b8079a65 100644 --- a/google/cloud/bigquery_reservation_v1/services/reservation_service/client.py +++ b/google/cloud/bigquery_reservation_v1/services/reservation_service/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources import warnings @@ -431,7 +431,7 @@ def __init__( def create_reservation( self, - request: gcbr_reservation.CreateReservationRequest = None, + request: Union[gcbr_reservation.CreateReservationRequest, dict] = None, *, parent: str = None, reservation: gcbr_reservation.Reservation = None, @@ -443,7 +443,7 @@ def create_reservation( r"""Creates a new reservation resource. Args: - request (google.cloud.bigquery_reservation_v1.types.CreateReservationRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.CreateReservationRequest, dict]): The request object. The request for [ReservationService.CreateReservation][google.cloud.bigquery.reservation.v1.ReservationService.CreateReservation]. parent (str): @@ -524,7 +524,7 @@ def create_reservation( def list_reservations( self, - request: reservation.ListReservationsRequest = None, + request: Union[reservation.ListReservationsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -535,7 +535,7 @@ def list_reservations( specified location. Args: - request (google.cloud.bigquery_reservation_v1.types.ListReservationsRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.ListReservationsRequest, dict]): The request object. The request for [ReservationService.ListReservations][google.cloud.bigquery.reservation.v1.ReservationService.ListReservations]. parent (str): @@ -605,7 +605,7 @@ def list_reservations( def get_reservation( self, - request: reservation.GetReservationRequest = None, + request: Union[reservation.GetReservationRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -615,7 +615,7 @@ def get_reservation( r"""Returns information about the reservation. Args: - request (google.cloud.bigquery_reservation_v1.types.GetReservationRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.GetReservationRequest, dict]): The request object. The request for [ReservationService.GetReservation][google.cloud.bigquery.reservation.v1.ReservationService.GetReservation]. name (str): @@ -677,7 +677,7 @@ def get_reservation( def delete_reservation( self, - request: reservation.DeleteReservationRequest = None, + request: Union[reservation.DeleteReservationRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -689,7 +689,7 @@ def delete_reservation( assignments. Args: - request (google.cloud.bigquery_reservation_v1.types.DeleteReservationRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.DeleteReservationRequest, dict]): The request object. The request for [ReservationService.DeleteReservation][google.cloud.bigquery.reservation.v1.ReservationService.DeleteReservation]. name (str): @@ -744,7 +744,7 @@ def delete_reservation( def update_reservation( self, - request: gcbr_reservation.UpdateReservationRequest = None, + request: Union[gcbr_reservation.UpdateReservationRequest, dict] = None, *, reservation: gcbr_reservation.Reservation = None, update_mask: field_mask_pb2.FieldMask = None, @@ -755,7 +755,7 @@ def update_reservation( r"""Updates an existing reservation resource. Args: - request (google.cloud.bigquery_reservation_v1.types.UpdateReservationRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.UpdateReservationRequest, dict]): The request object. The request for [ReservationService.UpdateReservation][google.cloud.bigquery.reservation.v1.ReservationService.UpdateReservation]. reservation (google.cloud.bigquery_reservation_v1.types.Reservation): @@ -825,7 +825,7 @@ def update_reservation( def create_capacity_commitment( self, - request: reservation.CreateCapacityCommitmentRequest = None, + request: Union[reservation.CreateCapacityCommitmentRequest, dict] = None, *, parent: str = None, capacity_commitment: reservation.CapacityCommitment = None, @@ -836,7 +836,7 @@ def create_capacity_commitment( r"""Creates a new capacity commitment resource. Args: - request (google.cloud.bigquery_reservation_v1.types.CreateCapacityCommitmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.CreateCapacityCommitmentRequest, dict]): The request object. The request for [ReservationService.CreateCapacityCommitment][google.cloud.bigquery.reservation.v1.ReservationService.CreateCapacityCommitment]. parent (str): @@ -920,7 +920,7 @@ def create_capacity_commitment( def list_capacity_commitments( self, - request: reservation.ListCapacityCommitmentsRequest = None, + request: Union[reservation.ListCapacityCommitmentsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -931,7 +931,7 @@ def list_capacity_commitments( project. Args: - request (google.cloud.bigquery_reservation_v1.types.ListCapacityCommitmentsRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.ListCapacityCommitmentsRequest, dict]): The request object. The request for [ReservationService.ListCapacityCommitments][google.cloud.bigquery.reservation.v1.ReservationService.ListCapacityCommitments]. parent (str): @@ -1003,7 +1003,7 @@ def list_capacity_commitments( def get_capacity_commitment( self, - request: reservation.GetCapacityCommitmentRequest = None, + request: Union[reservation.GetCapacityCommitmentRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1013,7 +1013,7 @@ def get_capacity_commitment( r"""Returns information about the capacity commitment. Args: - request (google.cloud.bigquery_reservation_v1.types.GetCapacityCommitmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.GetCapacityCommitmentRequest, dict]): The request object. The request for [ReservationService.GetCapacityCommitment][google.cloud.bigquery.reservation.v1.ReservationService.GetCapacityCommitment]. name (str): @@ -1087,7 +1087,7 @@ def get_capacity_commitment( def delete_capacity_commitment( self, - request: reservation.DeleteCapacityCommitmentRequest = None, + request: Union[reservation.DeleteCapacityCommitmentRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1099,7 +1099,7 @@ def delete_capacity_commitment( error code ``google.rpc.Code.FAILED_PRECONDITION``. Args: - request (google.cloud.bigquery_reservation_v1.types.DeleteCapacityCommitmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.DeleteCapacityCommitmentRequest, dict]): The request object. The request for [ReservationService.DeleteCapacityCommitment][google.cloud.bigquery.reservation.v1.ReservationService.DeleteCapacityCommitment]. name (str): @@ -1156,7 +1156,7 @@ def delete_capacity_commitment( def update_capacity_commitment( self, - request: reservation.UpdateCapacityCommitmentRequest = None, + request: Union[reservation.UpdateCapacityCommitmentRequest, dict] = None, *, capacity_commitment: reservation.CapacityCommitment = None, update_mask: field_mask_pb2.FieldMask = None, @@ -1174,7 +1174,7 @@ def update_capacity_commitment( ``google.rpc.Code.FAILED_PRECONDITION``. Args: - request (google.cloud.bigquery_reservation_v1.types.UpdateCapacityCommitmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.UpdateCapacityCommitmentRequest, dict]): The request object. The request for [ReservationService.UpdateCapacityCommitment][google.cloud.bigquery.reservation.v1.ReservationService.UpdateCapacityCommitment]. capacity_commitment (google.cloud.bigquery_reservation_v1.types.CapacityCommitment): @@ -1260,7 +1260,7 @@ def update_capacity_commitment( def split_capacity_commitment( self, - request: reservation.SplitCapacityCommitmentRequest = None, + request: Union[reservation.SplitCapacityCommitmentRequest, dict] = None, *, name: str = None, slot_count: int = None, @@ -1279,7 +1279,7 @@ def split_capacity_commitment( ``FLEX`` and then delete it. Args: - request (google.cloud.bigquery_reservation_v1.types.SplitCapacityCommitmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.SplitCapacityCommitmentRequest, dict]): The request object. The request for [ReservationService.SplitCapacityCommitment][google.cloud.bigquery.reservation.v1.ReservationService.SplitCapacityCommitment]. name (str): @@ -1351,7 +1351,7 @@ def split_capacity_commitment( def merge_capacity_commitments( self, - request: reservation.MergeCapacityCommitmentsRequest = None, + request: Union[reservation.MergeCapacityCommitmentsRequest, dict] = None, *, parent: str = None, capacity_commitment_ids: Sequence[str] = None, @@ -1371,7 +1371,7 @@ def merge_capacity_commitments( ``google.rpc.Code.FAILED_PRECONDITION``. Args: - request (google.cloud.bigquery_reservation_v1.types.MergeCapacityCommitmentsRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.MergeCapacityCommitmentsRequest, dict]): The request object. The request for [ReservationService.MergeCapacityCommitments][google.cloud.bigquery.reservation.v1.ReservationService.MergeCapacityCommitments]. parent (str): @@ -1460,7 +1460,7 @@ def merge_capacity_commitments( def create_assignment( self, - request: reservation.CreateAssignmentRequest = None, + request: Union[reservation.CreateAssignmentRequest, dict] = None, *, parent: str = None, assignment: reservation.Assignment = None, @@ -1508,7 +1508,7 @@ def create_assignment( the assignment does not match location of the reservation. Args: - request (google.cloud.bigquery_reservation_v1.types.CreateAssignmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.CreateAssignmentRequest, dict]): The request object. The request for [ReservationService.CreateAssignment][google.cloud.bigquery.reservation.v1.ReservationService.CreateAssignment]. Note: "bigquery.reservationAssignments.create" @@ -1580,7 +1580,7 @@ def create_assignment( def list_assignments( self, - request: reservation.ListAssignmentsRequest = None, + request: Union[reservation.ListAssignmentsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1611,7 +1611,7 @@ def list_assignments( **Note** "-" cannot be used for projects nor locations. Args: - request (google.cloud.bigquery_reservation_v1.types.ListAssignmentsRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.ListAssignmentsRequest, dict]): The request object. The request for [ReservationService.ListAssignments][google.cloud.bigquery.reservation.v1.ReservationService.ListAssignments]. parent (str): @@ -1686,7 +1686,7 @@ def list_assignments( def delete_assignment( self, - request: reservation.DeleteAssignmentRequest = None, + request: Union[reservation.DeleteAssignmentRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1711,7 +1711,7 @@ def delete_assignment( ``project2`` will switch to use on-demand mode. Args: - request (google.cloud.bigquery_reservation_v1.types.DeleteAssignmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.DeleteAssignmentRequest, dict]): The request object. The request for [ReservationService.DeleteAssignment][google.cloud.bigquery.reservation.v1.ReservationService.DeleteAssignment]. Note: "bigquery.reservationAssignments.delete" @@ -1767,7 +1767,7 @@ def delete_assignment( def search_assignments( self, - request: reservation.SearchAssignmentsRequest = None, + request: Union[reservation.SearchAssignmentsRequest, dict] = None, *, parent: str = None, query: str = None, @@ -1802,7 +1802,7 @@ def search_assignments( **Note** "-" cannot be used for projects nor locations. Args: - request (google.cloud.bigquery_reservation_v1.types.SearchAssignmentsRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.SearchAssignmentsRequest, dict]): The request object. The request for [ReservationService.SearchAssignments][google.cloud.bigquery.reservation.v1.ReservationService.SearchAssignments]. Note: "bigquery.reservationAssignments.search" @@ -1895,7 +1895,7 @@ def search_assignments( def search_all_assignments( self, - request: reservation.SearchAllAssignmentsRequest = None, + request: Union[reservation.SearchAllAssignmentsRequest, dict] = None, *, parent: str = None, query: str = None, @@ -1928,7 +1928,7 @@ def search_all_assignments( ``projects/*/locations/*reservations/*``. Args: - request (google.cloud.bigquery_reservation_v1.types.SearchAllAssignmentsRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.SearchAllAssignmentsRequest, dict]): The request object. The request for [ReservationService.SearchAllAssignments][google.cloud.bigquery.reservation.v1.ReservationService.SearchAllAssignments]. Note: "bigquery.reservationAssignments.search" @@ -2016,7 +2016,7 @@ def search_all_assignments( def move_assignment( self, - request: reservation.MoveAssignmentRequest = None, + request: Union[reservation.MoveAssignmentRequest, dict] = None, *, name: str = None, destination_id: str = None, @@ -2031,7 +2031,7 @@ def move_assignment( reservation. Args: - request (google.cloud.bigquery_reservation_v1.types.MoveAssignmentRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.MoveAssignmentRequest, dict]): The request object. The request for [ReservationService.MoveAssignment][google.cloud.bigquery.reservation.v1.ReservationService.MoveAssignment]. **Note**: "bigquery.reservationAssignments.create" @@ -2108,7 +2108,7 @@ def move_assignment( def get_bi_reservation( self, - request: reservation.GetBiReservationRequest = None, + request: Union[reservation.GetBiReservationRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -2118,7 +2118,7 @@ def get_bi_reservation( r"""Retrieves a BI reservation. Args: - request (google.cloud.bigquery_reservation_v1.types.GetBiReservationRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.GetBiReservationRequest, dict]): The request object. A request to get a singleton BI reservation. name (str): @@ -2178,7 +2178,7 @@ def get_bi_reservation( def update_bi_reservation( self, - request: reservation.UpdateBiReservationRequest = None, + request: Union[reservation.UpdateBiReservationRequest, dict] = None, *, bi_reservation: reservation.BiReservation = None, update_mask: field_mask_pb2.FieldMask = None, @@ -2196,7 +2196,7 @@ def update_bi_reservation( must be set to 0. Args: - request (google.cloud.bigquery_reservation_v1.types.UpdateBiReservationRequest): + request (Union[google.cloud.bigquery_reservation_v1.types.UpdateBiReservationRequest, dict]): The request object. A request to update a BI reservation. bi_reservation (google.cloud.bigquery_reservation_v1.types.BiReservation): diff --git a/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/base.py b/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/base.py index a15584b1..ff994fc1 100644 --- a/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/base.py +++ b/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/base.py @@ -122,7 +122,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc.py b/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc.py index 9e465a64..5164a2d9 100644 --- a/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc.py +++ b/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc.py @@ -98,16 +98,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc_asyncio.py b/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc_asyncio.py index 573bbf71..56fbb16f 100644 --- a/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc_asyncio.py +++ b/google/cloud/bigquery_reservation_v1/services/reservation_service/transports/grpc_asyncio.py @@ -145,16 +145,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/scripts/fixup_bigquery_reservation_v1_keywords.py b/scripts/fixup_bigquery_reservation_v1_keywords.py index 569bb151..c7cb6030 100644 --- a/scripts/fixup_bigquery_reservation_v1_keywords.py +++ b/scripts/fixup_bigquery_reservation_v1_keywords.py @@ -39,26 +39,26 @@ def partition( class bigquery_reservationCallTransformer(cst.CSTTransformer): CTRL_PARAMS: Tuple[str] = ('retry', 'timeout', 'metadata') METHOD_TO_PARAMS: Dict[str, Tuple[str]] = { - 'create_assignment': ('parent', 'assignment', 'assignment_id', ), - 'create_capacity_commitment': ('parent', 'capacity_commitment', 'enforce_single_admin_project_per_org', 'capacity_commitment_id', ), - 'create_reservation': ('parent', 'reservation_id', 'reservation', ), - 'delete_assignment': ('name', ), - 'delete_capacity_commitment': ('name', 'force', ), - 'delete_reservation': ('name', ), - 'get_bi_reservation': ('name', ), - 'get_capacity_commitment': ('name', ), - 'get_reservation': ('name', ), - 'list_assignments': ('parent', 'page_size', 'page_token', ), - 'list_capacity_commitments': ('parent', 'page_size', 'page_token', ), - 'list_reservations': ('parent', 'page_size', 'page_token', ), - 'merge_capacity_commitments': ('parent', 'capacity_commitment_ids', ), - 'move_assignment': ('name', 'destination_id', ), - 'search_all_assignments': ('parent', 'query', 'page_size', 'page_token', ), - 'search_assignments': ('parent', 'query', 'page_size', 'page_token', ), - 'split_capacity_commitment': ('name', 'slot_count', ), - 'update_bi_reservation': ('bi_reservation', 'update_mask', ), - 'update_capacity_commitment': ('capacity_commitment', 'update_mask', ), - 'update_reservation': ('reservation', 'update_mask', ), + 'create_assignment': ('parent', 'assignment', 'assignment_id', ), + 'create_capacity_commitment': ('parent', 'capacity_commitment', 'enforce_single_admin_project_per_org', 'capacity_commitment_id', ), + 'create_reservation': ('parent', 'reservation_id', 'reservation', ), + 'delete_assignment': ('name', ), + 'delete_capacity_commitment': ('name', 'force', ), + 'delete_reservation': ('name', ), + 'get_bi_reservation': ('name', ), + 'get_capacity_commitment': ('name', ), + 'get_reservation': ('name', ), + 'list_assignments': ('parent', 'page_size', 'page_token', ), + 'list_capacity_commitments': ('parent', 'page_size', 'page_token', ), + 'list_reservations': ('parent', 'page_size', 'page_token', ), + 'merge_capacity_commitments': ('parent', 'capacity_commitment_ids', ), + 'move_assignment': ('name', 'destination_id', ), + 'search_all_assignments': ('parent', 'query', 'page_size', 'page_token', ), + 'search_assignments': ('parent', 'query', 'page_size', 'page_token', ), + 'split_capacity_commitment': ('name', 'slot_count', ), + 'update_bi_reservation': ('bi_reservation', 'update_mask', ), + 'update_capacity_commitment': ('capacity_commitment', 'update_mask', ), + 'update_reservation': ('reservation', 'update_mask', ), } def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: @@ -77,7 +77,7 @@ def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: return updated kwargs, ctrl_kwargs = partition( - lambda a: not a.keyword.value in self.CTRL_PARAMS, + lambda a: a.keyword.value not in self.CTRL_PARAMS, kwargs )