Skip to content
This repository has been archived by the owner on Dec 31, 2023. It is now read-only.

Commit

Permalink
feat: Update Compute Engine API to revision 20230307 (#786) (#403)
Browse files Browse the repository at this point in the history 
* feat: Update Compute Engine API to revision 20230307 (#786)

Source-Link: googleapis/googleapis@f64a3c8

Source-Link: https://github.com/googleapis/googleapis-gen/commit/808541b7e7697e2984a0ea0eba1adb5560dade97
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiODA4NTQxYjdlNzY5N2UyOTg0YTBlYTBlYmExYWRiNTU2MGRhZGU5NyJ9

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
  • Loading branch information
@gcf-owl-bot
gcf-owl-bot[bot] and gcf-owl-bot[bot] authored Mar 21, 2023
1 parent 54636e1 commit acc70b8
Show file tree
Hide file tree
Showing 9 changed files with 1,232 additions and 4 deletions.
4 changes: 4 additions & 0 deletions google/cloud/compute/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1029,6 +1029,7 @@
NodeGroupsListNodes,
NodeGroupsScopedList,
NodeGroupsSetNodeTemplateRequest,
NodeGroupsSimulateMaintenanceEventRequest,
NodeTemplate,
NodeTemplateAggregatedList,
NodeTemplateList,
Expand Down Expand Up @@ -1365,6 +1366,7 @@
ShieldedInstanceIntegrityPolicy,
SignedUrlKey,
SimulateMaintenanceEventInstanceRequest,
SimulateMaintenanceEventNodeGroupRequest,
Snapshot,
SnapshotList,
SourceDiskEncryptionKey,
Expand Down Expand Up @@ -2449,6 +2451,7 @@
"NodeGroupsListNodes",
"NodeGroupsScopedList",
"NodeGroupsSetNodeTemplateRequest",
"NodeGroupsSimulateMaintenanceEventRequest",
"NodeTemplate",
"NodeTemplateAggregatedList",
"NodeTemplateList",
Expand Down Expand Up @@ -2785,6 +2788,7 @@
"ShieldedInstanceIntegrityPolicy",
"SignedUrlKey",
"SimulateMaintenanceEventInstanceRequest",
"SimulateMaintenanceEventNodeGroupRequest",
"Snapshot",
"SnapshotList",
"SourceDiskEncryptionKey",
Expand Down
4 changes: 4 additions & 0 deletions google/cloud/compute_v1/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -925,6 +925,7 @@
NodeGroupsListNodes,
NodeGroupsScopedList,
NodeGroupsSetNodeTemplateRequest,
NodeGroupsSimulateMaintenanceEventRequest,
NodeTemplate,
NodeTemplateAggregatedList,
NodeTemplateList,
Expand Down Expand Up @@ -1261,6 +1262,7 @@
ShieldedInstanceIntegrityPolicy,
SignedUrlKey,
SimulateMaintenanceEventInstanceRequest,
SimulateMaintenanceEventNodeGroupRequest,
Snapshot,
SnapshotList,
SourceDiskEncryptionKey,
Expand Down Expand Up @@ -2293,6 +2295,7 @@
"NodeGroupsListNodes",
"NodeGroupsScopedList",
"NodeGroupsSetNodeTemplateRequest",
"NodeGroupsSimulateMaintenanceEventRequest",
"NodeTemplate",
"NodeTemplateAggregatedList",
"NodeTemplateList",
Expand Down Expand Up @@ -2665,6 +2668,7 @@
"ShieldedInstanceIntegrityPolicy",
"SignedUrlKey",
"SimulateMaintenanceEventInstanceRequest",
"SimulateMaintenanceEventNodeGroupRequest",
"Snapshot",
"SnapshotList",
"SnapshotsClient",
Expand Down
5 changes: 5 additions & 0 deletions google/cloud/compute_v1/gapic_metadata.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1954,6 +1954,11 @@
"set_node_template"
]
},
"SimulateMaintenanceEvent": {
"methods": [
"simulate_maintenance_event"
]
},
"TestIamPermissions": {
"methods": [
"test_iam_permissions"
Expand Down
277 changes: 277 additions & 0 deletions google/cloud/compute_v1/services/node_groups/client.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2601,6 +2601,283 @@ def error_code(self):
# Done; return the response.
return response

def simulate_maintenance_event_unary(
self,
request: Optional[
Union[compute.SimulateMaintenanceEventNodeGroupRequest, dict]
] = None,
*,
project: Optional[str] = None,
zone: Optional[str] = None,
node_group: Optional[str] = None,
node_groups_simulate_maintenance_event_request_resource: Optional[
compute.NodeGroupsSimulateMaintenanceEventRequest
] = None,
retry: OptionalRetry = gapic_v1.method.DEFAULT,
timeout: Union[float, object] = gapic_v1.method.DEFAULT,
metadata: Sequence[Tuple[str, str]] = (),
) -> compute.Operation:
r"""Simulates maintenance event on specified nodes from
the node group.
Args:
request (Union[google.cloud.compute_v1.types.SimulateMaintenanceEventNodeGroupRequest, dict]):
The request object. A request message for
NodeGroups.SimulateMaintenanceEvent. See the method
description for details.
project (str):
Project ID for this request.
This corresponds to the ``project`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
zone (str):
The name of the zone for this
request.
This corresponds to the ``zone`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
node_group (str):
Name of the NodeGroup resource whose
nodes will go under maintenance
simulation.
This corresponds to the ``node_group`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
node_groups_simulate_maintenance_event_request_resource (google.cloud.compute_v1.types.NodeGroupsSimulateMaintenanceEventRequest):
The body resource for this request
This corresponds to the ``node_groups_simulate_maintenance_event_request_resource`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
retry (google.api_core.retry.Retry): Designation of what errors, if any,
should be retried.
timeout (float): The timeout for this request.
metadata (Sequence[Tuple[str, str]]): Strings which should be
sent along with the request as metadata.
Returns:
google.api_core.extended_operation.ExtendedOperation:
An object representing a extended
long-running operation.
"""
# Create or coerce a protobuf request object.
# Quick check: If we got a request object, we should *not* have
# gotten any keyword arguments that map to the request.
has_flattened_params = any(
[
project,
zone,
node_group,
node_groups_simulate_maintenance_event_request_resource,
]
)
if request is not None and has_flattened_params:
raise ValueError(
"If the `request` argument is set, then none of "
"the individual field arguments should be set."
)

# Minor optimization to avoid making a copy if the user passes
# in a compute.SimulateMaintenanceEventNodeGroupRequest.
# There's no risk of modifying the input as we've already verified
# there are no flattened fields.
if not isinstance(request, compute.SimulateMaintenanceEventNodeGroupRequest):
request = compute.SimulateMaintenanceEventNodeGroupRequest(request)
# If we have keyword arguments corresponding to fields on the
# request, apply these.
if project is not None:
request.project = project
if zone is not None:
request.zone = zone
if node_group is not None:
request.node_group = node_group
if node_groups_simulate_maintenance_event_request_resource is not None:
request.node_groups_simulate_maintenance_event_request_resource = (
node_groups_simulate_maintenance_event_request_resource
)

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = self._transport._wrapped_methods[
self._transport.simulate_maintenance_event
]

# Certain fields should be provided within the metadata header;
# add these here.
metadata = tuple(metadata) + (
gapic_v1.routing_header.to_grpc_metadata(
(
("project", request.project),
("zone", request.zone),
("node_group", request.node_group),
)
),
)

# Send the request.
response = rpc(
request,
retry=retry,
timeout=timeout,
metadata=metadata,
)

# Done; return the response.
return response

def simulate_maintenance_event(
self,
request: Optional[
Union[compute.SimulateMaintenanceEventNodeGroupRequest, dict]
] = None,
*,
project: Optional[str] = None,
zone: Optional[str] = None,
node_group: Optional[str] = None,
node_groups_simulate_maintenance_event_request_resource: Optional[
compute.NodeGroupsSimulateMaintenanceEventRequest
] = None,
retry: OptionalRetry = gapic_v1.method.DEFAULT,
timeout: Union[float, object] = gapic_v1.method.DEFAULT,
metadata: Sequence[Tuple[str, str]] = (),
) -> extended_operation.ExtendedOperation:
r"""Simulates maintenance event on specified nodes from
the node group.
Args:
request (Union[google.cloud.compute_v1.types.SimulateMaintenanceEventNodeGroupRequest, dict]):
The request object. A request message for
NodeGroups.SimulateMaintenanceEvent. See the method
description for details.
project (str):
Project ID for this request.
This corresponds to the ``project`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
zone (str):
The name of the zone for this
request.
This corresponds to the ``zone`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
node_group (str):
Name of the NodeGroup resource whose
nodes will go under maintenance
simulation.
This corresponds to the ``node_group`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
node_groups_simulate_maintenance_event_request_resource (google.cloud.compute_v1.types.NodeGroupsSimulateMaintenanceEventRequest):
The body resource for this request
This corresponds to the ``node_groups_simulate_maintenance_event_request_resource`` field
on the ``request`` instance; if ``request`` is provided, this
should not be set.
retry (google.api_core.retry.Retry): Designation of what errors, if any,
should be retried.
timeout (float): The timeout for this request.
metadata (Sequence[Tuple[str, str]]): Strings which should be
sent along with the request as metadata.
Returns:
google.api_core.extended_operation.ExtendedOperation:
An object representing a extended
long-running operation.
"""
# Create or coerce a protobuf request object.
# Quick check: If we got a request object, we should *not* have
# gotten any keyword arguments that map to the request.
has_flattened_params = any(
[
project,
zone,
node_group,
node_groups_simulate_maintenance_event_request_resource,
]
)
if request is not None and has_flattened_params:
raise ValueError(
"If the `request` argument is set, then none of "
"the individual field arguments should be set."
)

# Minor optimization to avoid making a copy if the user passes
# in a compute.SimulateMaintenanceEventNodeGroupRequest.
# There's no risk of modifying the input as we've already verified
# there are no flattened fields.
if not isinstance(request, compute.SimulateMaintenanceEventNodeGroupRequest):
request = compute.SimulateMaintenanceEventNodeGroupRequest(request)
# If we have keyword arguments corresponding to fields on the
# request, apply these.
if project is not None:
request.project = project
if zone is not None:
request.zone = zone
if node_group is not None:
request.node_group = node_group
if node_groups_simulate_maintenance_event_request_resource is not None:
request.node_groups_simulate_maintenance_event_request_resource = (
node_groups_simulate_maintenance_event_request_resource
)

# Wrap the RPC method; this adds retry and timeout information,
# and friendly error handling.
rpc = self._transport._wrapped_methods[
self._transport.simulate_maintenance_event
]

# Certain fields should be provided within the metadata header;
# add these here.
metadata = tuple(metadata) + (
gapic_v1.routing_header.to_grpc_metadata(
(
("project", request.project),
("zone", request.zone),
("node_group", request.node_group),
)
),
)

# Send the request.
response = rpc(
request,
retry=retry,
timeout=timeout,
metadata=metadata,
)

operation_service = self._transport._zone_operations_client
operation_request = compute.GetZoneOperationRequest()
operation_request.project = request.project
operation_request.zone = request.zone
operation_request.operation = response.name

get_operation = functools.partial(operation_service.get, operation_request)
# Cancel is not part of extended operations yet.
cancel_operation = lambda: None

# Note: this class is an implementation detail to provide a uniform
# set of names for certain fields in the extended operation proto message.
# See google.api_core.extended_operation.ExtendedOperation for details
# on these properties and the expected interface.
class _CustomOperation(extended_operation.ExtendedOperation):
@property
def error_message(self):
return self._extended_operation.http_error_message

@property
def error_code(self):
return self._extended_operation.http_error_status_code

response = _CustomOperation.make(get_operation, cancel_operation, response)

# Done; return the response.
return response

def test_iam_permissions(
self,
request: Optional[
Expand Down
14 changes: 14 additions & 0 deletions google/cloud/compute_v1/services/node_groups/transports/base.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -187,6 +187,11 @@ def _prep_wrapped_messages(self, client_info):
default_timeout=None,
client_info=client_info,
),
self.simulate_maintenance_event: gapic_v1.method.wrap_method(
self.simulate_maintenance_event,
default_timeout=None,
client_info=client_info,
),
self.test_iam_permissions: gapic_v1.method.wrap_method(
self.test_iam_permissions,
default_timeout=None,
Expand Down Expand Up @@ -313,6 +318,15 @@ def set_node_template(
]:
raise NotImplementedError()

@property
def simulate_maintenance_event(
self,
) -> Callable[
[compute.SimulateMaintenanceEventNodeGroupRequest],
Union[compute.Operation, Awaitable[compute.Operation]],
]:
raise NotImplementedError()

@property
def test_iam_permissions(
self,
Expand Down
Loading

0 comments on commit acc70b8

Please sign in to comment.