84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2021 Google LLC. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Command for perform maintenance on Google Compute Engine instance."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
from googlecloudsdk.api_lib.compute import base_classes
|
|
from googlecloudsdk.api_lib.compute import exceptions
|
|
from googlecloudsdk.api_lib.compute import utils
|
|
from googlecloudsdk.calliope import base
|
|
from googlecloudsdk.command_lib.compute import flags
|
|
from googlecloudsdk.command_lib.compute.instances import flags as instance_flags
|
|
|
|
|
|
class PerformMaintenanceError(exceptions.Error):
|
|
"""Exception thrown when there is a problem with performing maintenance on an instance."""
|
|
|
|
|
|
@base.DefaultUniverseOnly
|
|
@base.ReleaseTracks(
|
|
base.ReleaseTrack.ALPHA, base.ReleaseTrack.BETA, base.ReleaseTrack.GA
|
|
)
|
|
class PerformMaintenance(base.UpdateCommand):
|
|
"""Perform maintenance of Google Compute Engine instance."""
|
|
|
|
@staticmethod
|
|
def Args(parser):
|
|
instance_flags.INSTANCES_ARG.AddArgument(parser)
|
|
|
|
def Run(self, args):
|
|
holder = base_classes.ComputeApiHolder(self.ReleaseTrack())
|
|
client = holder.client
|
|
messages = holder.client.messages
|
|
|
|
instance_refs = instance_flags.INSTANCES_ARG.ResolveAsResource(
|
|
args, holder.resources, scope_lister=flags.GetDefaultScopeLister(client)
|
|
)
|
|
requests = []
|
|
for instance_ref in instance_refs:
|
|
request_protobuf = messages.ComputeInstancesPerformMaintenanceRequest(
|
|
**instance_ref.AsDict()
|
|
)
|
|
request = (
|
|
client.apitools_client.instances,
|
|
'PerformMaintenance',
|
|
request_protobuf,
|
|
)
|
|
requests.append(request)
|
|
errors = []
|
|
holder.client.MakeRequests(requests, errors_to_collect=errors)
|
|
if errors:
|
|
utils.RaiseException(
|
|
errors,
|
|
PerformMaintenanceError,
|
|
error_message='Could not perform maintenance for resource:',
|
|
)
|
|
|
|
|
|
PerformMaintenance.detailed_help = {
|
|
'brief':
|
|
'Perform maintenance of Google Compute Engine instance',
|
|
'EXAMPLES':
|
|
"""\
|
|
To perform customer-triggered maintenance on an instance named ``{0}''
|
|
located in zone ``{1}'', run:
|
|
|
|
$ {2} {0} --zone={1}
|
|
""".format('test-instance', 'us-east1-d', '{command}')
|
|
}
|