274 lines
8.9 KiB
Python
274 lines
8.9 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2017 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 creating Compute Engine commitments."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import re
|
|
|
|
from googlecloudsdk.api_lib.compute import base_classes
|
|
from googlecloudsdk.api_lib.compute import request_helper
|
|
from googlecloudsdk.api_lib.compute import utils
|
|
from googlecloudsdk.api_lib.util import apis
|
|
from googlecloudsdk.calliope import base
|
|
from googlecloudsdk.calliope import exceptions
|
|
from googlecloudsdk.command_lib.compute import flags as compute_flags
|
|
from googlecloudsdk.command_lib.compute.commitments import flags
|
|
from googlecloudsdk.command_lib.compute.commitments import reservation_helper
|
|
from googlecloudsdk.core import properties
|
|
|
|
|
|
_MISSING_COMMITMENTS_QUOTA_REGEX = r'Quota .COMMITMENTS. exceeded.+'
|
|
|
|
|
|
def _CommonArgs(track, parser):
|
|
"""Add common flags."""
|
|
flags.MakeCommitmentArg(False).AddArgument(parser, operation_type='create')
|
|
flags.AddAutoRenew(parser)
|
|
messages = apis.GetMessagesModule('compute', track)
|
|
flags.GetTypeMapperFlag(messages).choice_arg.AddToParser(parser)
|
|
|
|
|
|
@base.UniverseCompatible
|
|
@base.ReleaseTracks(base.ReleaseTrack.GA)
|
|
class Create(base.Command):
|
|
"""Create Compute Engine commitments."""
|
|
|
|
_support_share_setting = True
|
|
_support_stable_fleet = False
|
|
_support_existing_reservation = True
|
|
_support_reservation_sharing_policy = False
|
|
_support_60_month_plan = False
|
|
_support_24_month_plan = False
|
|
|
|
detailed_help = {'EXAMPLES': """
|
|
To create a commitment called ``commitment-1'' in the ``us-central1''
|
|
region, with a ``12-month'' plan, ``9GB'' of memory and 4 vcpu cores,
|
|
run:
|
|
|
|
$ {command} commitment-1 --plan=12-month --resources=memory=9GB,vcpu=4 --region=us-central1
|
|
"""}
|
|
|
|
@classmethod
|
|
def Args(cls, parser):
|
|
_CommonArgs('v1', parser)
|
|
flags.AddCreateFlags(
|
|
parser,
|
|
support_share_setting=cls._support_share_setting,
|
|
support_stable_fleet=cls._support_stable_fleet,
|
|
support_existing_reservation=cls._support_existing_reservation,
|
|
support_reservation_sharing_policy=cls._support_reservation_sharing_policy,
|
|
support_60_month_plan=cls._support_60_month_plan,
|
|
support_24_month_plan=cls._support_24_month_plan,
|
|
)
|
|
|
|
def _MakeCreateRequest(
|
|
self,
|
|
args,
|
|
messages,
|
|
project,
|
|
region,
|
|
commitment_ref,
|
|
existing_reservations,
|
|
holder,
|
|
):
|
|
|
|
if (
|
|
args.split_source_commitment is not None
|
|
and args.merge_source_commitments is not None
|
|
):
|
|
raise exceptions.ConflictingArgumentsException(
|
|
"It's not possible to merge and split in one request"
|
|
)
|
|
|
|
commitment_type_flag = flags.GetTypeMapperFlag(messages)
|
|
commitment_type = commitment_type_flag.GetEnumForChoice(args.type)
|
|
commitment = messages.Commitment(
|
|
reservations=reservation_helper.MakeReservations(
|
|
args, messages, holder
|
|
),
|
|
name=commitment_ref.Name(),
|
|
plan=flags.TranslatePlanArg(messages, args.plan),
|
|
resources=flags.TranslateResourcesArgGroup(messages, args),
|
|
type=commitment_type,
|
|
autoRenew=flags.TranslateAutoRenewArgForCreate(args),
|
|
splitSourceCommitment=args.split_source_commitment,
|
|
mergeSourceCommitments=flags.TranslateMergeArg(
|
|
args.merge_source_commitments,
|
|
),
|
|
existingReservations=existing_reservations,
|
|
)
|
|
commitment.customEndTimestamp = flags.TranslateCustomEndTimeArg(args)
|
|
return messages.ComputeRegionCommitmentsInsertRequest(
|
|
commitment=commitment,
|
|
project=project,
|
|
region=commitment_ref.region,
|
|
)
|
|
|
|
def Run(self, args):
|
|
holder = base_classes.ComputeApiHolder(self.ReleaseTrack())
|
|
resources = holder.resources
|
|
commitment_ref = flags.MakeCommitmentArg(False).ResolveAsResource(
|
|
args,
|
|
resources,
|
|
scope_lister=compute_flags.GetDefaultScopeLister(holder.client),
|
|
)
|
|
existing_reservations = flags.ResolveExistingReservationArgs(
|
|
args, resources
|
|
)
|
|
messages = holder.client.messages
|
|
region = properties.VALUES.compute.region.Get()
|
|
project = properties.VALUES.core.project.Get()
|
|
create_request = self._MakeCreateRequest(
|
|
args,
|
|
messages,
|
|
project,
|
|
region,
|
|
commitment_ref,
|
|
existing_reservations,
|
|
holder,
|
|
)
|
|
|
|
service = holder.client.apitools_client.regionCommitments
|
|
batch_url = holder.client.batch_url
|
|
http = holder.client.apitools_client.http
|
|
errors = []
|
|
result = list(
|
|
request_helper.MakeRequests(
|
|
requests=[(service, 'Insert', create_request)],
|
|
http=http,
|
|
batch_url=batch_url,
|
|
errors=errors,
|
|
)
|
|
)
|
|
for i, error in enumerate(errors):
|
|
if hasattr(error[1], 'message') and isinstance(error[1].message, str):
|
|
err_msg = error[1].message
|
|
else:
|
|
err_msg = error[1]
|
|
|
|
if re.match(_MISSING_COMMITMENTS_QUOTA_REGEX, err_msg):
|
|
errors[i] = (
|
|
error[0],
|
|
err_msg
|
|
+ (
|
|
' You can request commitments quota on '
|
|
'https://cloud.google.com/compute/docs/instances/'
|
|
'signing-up-committed-use-discounts#quota'
|
|
),
|
|
)
|
|
|
|
if errors:
|
|
utils.RaiseToolException(errors)
|
|
return result
|
|
|
|
|
|
@base.UniverseCompatible
|
|
@base.ReleaseTracks(base.ReleaseTrack.BETA)
|
|
class CreateBeta(Create):
|
|
"""Create Compute Engine commitments."""
|
|
|
|
_support_share_setting = True
|
|
_support_stable_fleet = True
|
|
_support_existing_reservation = True
|
|
_support_reservation_sharing_policy = True
|
|
_support_60_month_plan = False
|
|
_support_24_month_plan = False
|
|
|
|
@classmethod
|
|
def Args(cls, parser):
|
|
_CommonArgs('beta', parser)
|
|
flags.AddCreateFlags(
|
|
parser,
|
|
support_share_setting=cls._support_share_setting,
|
|
support_stable_fleet=cls._support_stable_fleet,
|
|
support_existing_reservation=cls._support_existing_reservation,
|
|
support_reservation_sharing_policy=cls._support_reservation_sharing_policy,
|
|
support_60_month_plan=cls._support_60_month_plan,
|
|
support_24_month_plan=cls._support_24_month_plan,
|
|
)
|
|
|
|
|
|
@base.UniverseCompatible
|
|
@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
|
|
class CreateAlpha(CreateBeta):
|
|
"""Create Compute Engine commitments."""
|
|
|
|
_support_share_setting = True
|
|
_support_stable_fleet = True
|
|
_support_existing_reservation = True
|
|
_support_reservation_sharing_policy = True
|
|
_support_60_month_plan = True
|
|
_support_24_month_plan = True
|
|
|
|
@classmethod
|
|
def Args(cls, parser):
|
|
_CommonArgs('alpha', parser)
|
|
flags.AddCreateFlags(
|
|
parser,
|
|
support_share_setting=cls._support_share_setting,
|
|
support_stable_fleet=cls._support_stable_fleet,
|
|
support_existing_reservation=cls._support_existing_reservation,
|
|
support_reservation_sharing_policy=cls._support_reservation_sharing_policy,
|
|
support_60_month_plan=cls._support_60_month_plan,
|
|
support_24_month_plan=cls._support_24_month_plan,
|
|
)
|
|
|
|
def _MakeCreateRequest(
|
|
self,
|
|
args,
|
|
messages,
|
|
project,
|
|
region,
|
|
commitment_ref,
|
|
existing_reservations,
|
|
holder,
|
|
):
|
|
|
|
if (
|
|
args.split_source_commitment is not None
|
|
and args.merge_source_commitments is not None
|
|
):
|
|
raise exceptions.ConflictingArgumentsException(
|
|
"It's not possible to merge and split in one request"
|
|
)
|
|
|
|
commitment_type_flag = flags.GetTypeMapperFlag(messages)
|
|
commitment_type = commitment_type_flag.GetEnumForChoice(args.type)
|
|
commitment = messages.Commitment(
|
|
reservations=reservation_helper.MakeReservations(
|
|
args, messages, holder
|
|
),
|
|
name=commitment_ref.Name(),
|
|
plan=flags.TranslatePlanArg(messages, args.plan),
|
|
resources=flags.TranslateResourcesArgGroup(messages, args),
|
|
type=commitment_type,
|
|
autoRenew=flags.TranslateAutoRenewArgForCreate(args),
|
|
splitSourceCommitment=args.split_source_commitment,
|
|
mergeSourceCommitments=flags.TranslateMergeArg(
|
|
args.merge_source_commitments
|
|
),
|
|
existingReservations=existing_reservations,
|
|
)
|
|
|
|
commitment.customEndTimestamp = flags.TranslateCustomEndTimeArg(args)
|
|
return messages.ComputeRegionCommitmentsInsertRequest(
|
|
commitment=commitment,
|
|
project=project,
|
|
region=commitment_ref.region,
|
|
)
|