<dZddlmZddlmZddlmZddlmZddlmZ ddl m Z dZ dd Z dd Zd Zd ZdZdZdZdZdZdZdZdZ ddZy )zJCommon utility functions to consturct compute future reservations message.)absolute_import)division)unicode_literals) exceptions)util)timesc:tj|t|dd}tj|t|dd}t ||j |j ||t|ddt|ddt|dd}t|ddrtj||nd}t|||j|}t|t|ddt|d dt|d d} t||jt|d dt|d d} t||t|d d} t|t|dd} d} |jdr t|d} d}|jdr t|d}d}|jdr t|d}t|dd}t|dd}d}|j!drt#|t|dd}t%||}d}|j!drt'|t|dd}d}|j!drt)|t|dd}d}|j!dr t|dd}t+||j-| | | | || |||||||||S)z9Construct future reservation message from args passed in. local_ssdN accelerator location_hintmaintenance_freeze_durationmaintenance_intervalsource_instance_template tpu_version chip_count workload_typeend_timeduration share_settingplanning_status%auto_delete_auto_created_reservations%auto_created_reservations_delete_time"auto_created_reservations_durationrequire_specific_reservationreservation_namedeployment_typescheduling_typereservation_modeenable_emergent_maintenance)reservation_util MakeLocalSsdsgetattrMakeGuestAcceleratorsMakeAllocatedInstanceProperties machine_typemin_cpu_platformResolveSourceInstanceTemplate MakeSpecificSKUPropertiesMessage total_count)MakeAggregateReservationPropertiesMessageMakeTimeWindowMessage start_timeMakeShareSettingsMakePlanningStatus IsSpecifiedIsKnownAndSpecifiedMakeDeploymentTypeMakeCommitmentInfoMakeSchedulingTypeMakeReservationModeMakeFutureReservationMessageName)messages resourcesargsfuture_reservation_ref local_ssds acceleratorsallocated_instance_propertiessource_instance_template_refsku_properties aggregate_reservation_properties time_windowshare_settingsrenable_auto_delete_reservationsrrrrrcommitment_inforrrs Blib/googlecloudsdk/command_lib/compute/future_reservations/util.py$MakeFutureReservationMessageFromArgsrFs --h.5dK.NP*!77mT24,#B   dOT* d148 d*D1 # 14 844T9E  4# " .&O dM4( dL$' dOT* &"&h&-dJ&E&-dJ&EG+%Xt%,T?D%IK.&x'.t5F'MO/%)! =>&- 5'#+/' =>,3 5-)(,$ :;)0 2*&") *D"T#5t</ /0()07H$)OQO'x6// /0('$ 148O 01*'$ 2D9!% ;<") +T# &!!#&%+("!# Nc|y|j}|||||} |r|| _|r |dz| _|r |jj|| _| S)zBConstructs an instance propteries for reservations message object.N) machineTypeguestAcceleratorsminCpuPlatform localSsdsi)9AllocationSpecificSKUAllocationReservedInstanceProperties locationHintmaintenanceFreezeDurationHours"MaintenanceIntervalValueValuesEnummaintenanceInterval) r7r%r&r;r<r freeze_durationfreeze_interval prop_msgsinstance_propertiess rEr$r$sz HH !$%  '4$9HD9P6JJ + +O <+ rGcf||yd}d}|r|j}n|}|j|||S)zHConstructs a specific sku properties message object if any is specified.N) totalCountsourceInstanceTemplateinstanceProperties)SelfLink&FutureReservationSpecificSKUProperties)r7rUr)r> propertiessource_instance_template_urls rEr(r(sV %A%I *!%!#?#H#H#J $J  8 89# 9 %%rGc|sy|jjj|jjj|jjjd}g}|j |}|j |j||j|jj|||}|r |jj||_ |S)z>Constructs an aggregate reservation properties message object.N)V5EV5PV6E)acceleratorCount)r )vmFamilyreservedResources) AllocationAggregateReservationVmFamilyValueValuesEnum(VM_FAMILY_CLOUD_TPU_LITE_POD_SLICE_CT5LP"VM_FAMILY_CLOUD_TPU_POD_SLICE_CT5P'VM_FAMILY_CLOUD_TPU_LITE_POD_SLICE_CT6E=AllocationAggregateReservationReservedResourceInfoAcceleratorappend2AllocationAggregateReservationReservedResourceInfoWorkloadTypeValueValuesEnum workloadType)r7rrrtpu_version_to_vm_familyreserved_resourcesr r@s rEr*r*s    1 1 I I r r  1 1 I I l l  1 1 I I q q LL%M AA!B &.%L%L66NN "; /+ &M&"//KK  %1 *)rGcr|r|j||S|j||j|S)z*Constructs the time window message object.) startTimeendTimeseconds)rrr)FutureReservationTimeWindowDuration)r7r,rrs rEr+r+sQ  / /h 0 00  / /x'8'8'8'J 0 LLrGc|r|dk(rW|jdrtjdd|j|jjj S|dk(rc|jdstjdd|j|jjj t|dd Syy) z-Constructs the share settings message object.local share_withz --share_withz_The scope this reservation is to be shared with must not be specified with share setting local.) shareTypeprojectszEThe projects this reservation is to be shared with must be specified.N)r{r|)r/rInvalidArgumentException ShareSettingsShareTypeValueValuesEnumLOCALSPECIFIC_PROJECTSr")r7r9setting_configss rEr-r-s'!  , '11  23 3 # #h.D.D%=%=ee$EE*$   l +11    # #**CC  4t4$66 % rGc|rJ|dk(r |jjjS|dk(r |jjjSy)z*Constructs the planning status enum value. SUBMITTEDDRAFTN)FutureReservationPlanningStatusValueValuesEnumrr)r7rs rEr.r.sH+%  ' ' E E O OO'!  ' ' E E K KK rGc|rJ|dk(r |jjjS|dk(r |jjjSy)z*Constructs the deployment type enum value.DENSEFLEXIBLEN)rDeploymentTypeValueValuesEnumrr)r7rs rEr1r1sH'!  ' ' E E K KK*$  ' ' E E N NN rGc t|dd}d}|jdrt|t|dd}d}|jdrt|t|dd}t |||gr|j |||Sy)z.Constructs the commitment info message object.commitment_nameNcommitment_planprevious_commitment_terms)commitmentNamecommitmentPlanpreviousCommitmentTerms)r"r0MakeCommitmentPlanMakePreviousCommitmentTermsanyFutureReservationCommitmentInfo)r7r9rrrs rEr2r2'sD"3T:// /0()07H$)OQO" 9: ;'$ ;TB! /?,E FG  3 3&& 9 4  rGc|rJ|dk(r |jjjS|dk(r |jjjSy)z*Constructs the commitment plan enum value. TWELVE_MONTHTHIRTY_SIX_MONTHN)rCommitmentPlanValueValuesEnumrr)r7rs rErr=sL.(66TTl,,66TT! rGcR|r%|dk(r |jjjSy)z4Constructs the previous commitment terms enum value.EXTENDN)r&PreviousCommitmentTermsValueValuesEnumr)r7rs rErrIs) H,66]]v rGc|rJ|dk(r |jjjS|dk(r |jjjSy)z*Constructs the scheduling type enum value.GROUPED INDEPENDENTN)rSchedulingTypeValueValuesEnumrr)r7rs rEr3r3RsI)#  ' ' E E M MM-'((FF{ rGc|rJ|dk(r |jjjS|dk(r |jjjSy)z+Constructs the reservation mode enum value.CALENDARDEFAULTN)rReservationModeValueValuesEnumrr)r7rs rEr4r4]sH:%  ' ' F F O OO9$  ' ' F F N NN rGc^|j|||}|r||_|r||_|r||_|||_|t j ||_| |j| |_ | | |_ | | |_ | | |_ | | |_ |||_|||_|||_|S)z/Constructs a future reservation message object.)name timeWindowplanningStatusrt)raggregateReservationspecificSkuProperties shareSettings!autoDeleteAutoCreatedReservationsrFormatDateTime!autoCreatedReservationsDeleteTimerwautoCreatedReservationsDurationspecificReservationRequiredreservationNamedeploymentTypecommitmentInfoschedulingTypereservationModeenableEmergentMaintenance)r7future_reservation_namerArBrr@r?rCrrrrrrDrrrfuture_reservation_messages rEr5r5gs( (99 "$ :  &(37E4/=,$0'@+6 BC@(3"DE>"-$:!1A. 0?- 0?- 0?-!1A. ,#8 $#rG)NNN)N) NNNNNNNNNNNN)__doc__ __future__rrrgooglecloudsdk.callioper/googlecloudsdk.command_lib.compute.reservationsrr googlecloudsdk.core.utilrrFr$r(r*r+r-r.r1r2rrr3r4r5rGrErsQ&'.T*pp374848D"& %,,*^L2,  &*$(*.'+!% $#E$rG