HdZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd lm Zddlm Zddlm Zdd lmZdd lmZddlZd d gZgdZd gZd dgZdZdZeddgZGdde j>Z dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ) d3dZ*d4d Z+d!Z,d"Z-d#Z.d$Z/d%Z0d&Z1d'Z2d(Z3d)Z4d*Z5d+Z6d5d,Z7d-Z8d.Z9 d6d/Z:d0Z;d4d1Z|jdr |jSy)N auto_renewF IsSpecifiedr2argss rTranslateAutoRenewArgForCreater7Ds l# ?? rc>|jdr |jSy)Nr2r3r5s rTranslateAutoRenewArgForUpdater9Js l# ?? rc|jdrd}d} tjj|jt|j}|r t jd}|jtjj|jt}|jt jjt}|s t d|Sy#t $rd}YwxYw#t $rY,wxYw)z7Translates the custom end time arg to a RFC3339 format.custom_end_timeFNTzAmerica/Los_AngeleszLInvalid custom end time. Expected format: YYYY-MM-DD or YYYY-MM-DDTHH:MM:SSZ) r4datetimestrptimer;RFC3339_FORMAT ValueErrorpytztimezonelocalizeDATE_ONLY_FORMAT astimezoneutcstrftime)r6try_date_only_parsefinal_date_timetzmidnight_date_time_mtvs rTranslateCustomEndTimeArgrKPs '(O!  !5!5~F,,o   ]]0 1!#    & &t';';=M N" 1;;DHHENN     "   7 !!!   s$:C(BC9( C65C69 DDc|j|d|jjj|j|ddz|jjjgS)Nramounttyperi)ResourceCommitmentTypeValueValuesEnumVCPUMEMORY)r* resources_args rTranslateResourcesArgrUwss!!v&**>>CC" !!x([9**>>EE"  rc|j}t||}d|jvrC|j|j |d|jj j |jdrS|j}|j|j |d|d|jj j|S)z,Util functions to parse ResourceCommitments. local-ssdrMresources_acceleratorcountrO)rNacceleratorTyperO) resourcesrUkeysappendrPrQ LOCAL_SSDr4rX ACCELERATOR)r*r6rTr[accelerator_args rTranslateResourcesArgGroupras..-#Hm<)M&&(( ## -,,@@JJ $  -.00O ##"7++F3,,@@LL $  rc,|r|jdSgS)z)List arguments are delineated by a comma.,)split)args rTranslateMergeArgrfs3&B&rc\tjdt|ddtjS)N commitmentr) resource_name completerpluralnameregional_collectionregion_explanation) compute_flagsResourceArgumentrREGION_PROPERTY_EXPLANATION)rks rMakeCommitmentArgrrs-  ' ' *  5&BB  rct||t|||||t|t|t |t |y)z4Add general arguments for `commitments create` flag.N)AddPlanForCreateAddReservationArgGroupAddResourcesArgGroupAddSplitSourceCommitmentAddMergeSourceCommitmentsAddCustomEndTime)parsersupport_share_settingsupport_stable_fleetsupport_existing_reservation"support_reservation_sharing_policysupport_60_month_plans rAddCreateFlagsrsH601 "( v6"F#6rc2t|t||y)z4Add general arguments for `commitments update` flag.N) AddAutoRenewAddPlanForUpdaterzrs rAddUpdateFlagsrsv601rcP|jdd|r tdStdS)N--planTDuration of the commitment.requiredchoiceshelp) add_argumentEXTENDED_VALID_PLANS VALID_PLANSrs rrtrtsB   &;" (  BM (  rcP|jdd|r tdStdS)NrFrr)rEXTENDED_VALID_UPDATE_PLANSVALID_UPDATE_PLANSrs rrrsD    * )    (  rc,|jddddS)Nz --auto-renew store_trueFz'Enable auto renewal for the commitment.)actiondefaultrrrzs rrrs&     4  rc|jddd|jddtd|jd dtd t|d y ) z6Add license based flags for `commitments create` flag.z --licenseTzApplicable license URI. For example: `https://www.googleapis.com/compute/v1/projects/suse-sap-cloud/global/licenses/sles-sap-12`rrz--cores-per-licenseFzZCore range of the instance. Must be one of: `1-2`, `3-4`, `5+`. Required for SAP licenses.rrOrz--amountzNumber of licenses purchased.)rN)rstrintrtrs rAddLicenceBasedFlagsrso h    5  4c0O67rc*|jdddS)Nz--split-source-commitmentFzsCreates the new commitment by splitting the specified source commitment and redistributing the specified resources.rrrs rrwrws&   ! J  rc*|jdddS)Nz--merge-source-commitmentsFzeCreates the new commitment by merging the specified source commitments and combining their resources.rrrs rrxrx s%   " >  rc4|jddtdS)Nz--custom-end-timeFzMSpecifies a custom future end date and extends the commitment's ongoing term.r)rrrs rryrys(       rc $|jdd}d}|jd|tjtttj dd }|jd |tjtt d y ) zDAdd the argument group for ResourceCommitment support in commitment.z/Manage the commitment for particular resources.TraResources to be included in the commitment. For details and examples of valid specifications, refer to the [custom machine type guide](https://cloud.google.com/compute/docs/instances/creating-instance-with-custom-machine-type#specifications). *memory*::: The size of the memory, should include units (e.g. 3072MB or 9GB). If no units are specified, GB is assumed. *vcpu*::: The number of the vCPU cores. *local-ssd*::: The size of local SSD. z --resources)rrWrspec)rrOaNManage the configuration of the type and number of accelerator cards to include in the commitment. *count*::: The number of accelerators to include. *type*::: The specific type (e.g. nvidia-tesla-k80 for NVIDIA Tesla K80) of the accelerator. Use `gcloud compute accelerator-types list` to learn about all available accelerator types. z--resources-accelerator)rYrON) add_grouprrArgDictr BinarySizer)rzresources_groupresources_helpaccelerator_helps rrvrv$s$$7$%/.    #..0        c3$? @rc`tjd|jjdddS)z:Helper to get a choice flag from the commitment type enum.z--typezaType of commitment. `memory-optimized` indicates that the commitment is for memory-optimized VMs.zgeneral-purposec |dk7S)NTYPE_UNSPECIFIEDr)xs rz#GetTypeMapperFlag..Ts q$66r)help_strrinclude_filter)r ChoiceEnumMapperr%rQ)r*s rGetTypeMapperFlagrJs4  # #-- 4 6  rc|jdd}t|d|jd}t|tjdj |tjdj |tj d j |tj d j |tjd j |tjd j |tjd j |tjdj ||S)z=Add reservation arguments to the update-reservations command.z8Manage reservations that are attached to the commitment.Tmutexz7Path to a YAML file of two reservations' configuration.) custom_textz:Specify source and destination reservations configuration.z--source-acceleratorz--dest-acceleratorz--source-local-ssdz--dest-local-ssdz--source-share-settingz--source-share-withz--dest-share-settingz--dest-share-with) rAddReservationsFromFileFlagAddReservationArgumentsreservation_flagsGetAcceleratorFlag AddToParserGetLocalSsdFlagGetSharedSettingFlagGetShareWithFlag)rzparent_reservations_groupreservations_groups rAddUpdateReservationGrouprXsK$..@/K1::B,-&&'=>JJ&&';<HH##$89EE##$67CC (()ABNN$$%:;GG(()?@LL$$%89EE -rc,d}ttttttd}|jdt j ||j dd|jdt j ||j d d|S) zDAdd --source-reservation and --dest-reservation arguments to parser.a~ {0} reservation configuration. *reservation*::: Name of the {0} reservation to operate on. *reservation-zone*::: Zone of the {0} reservation to operate on. *vm-count*::: The number of VM instances that are allocated to this reservation. The value of this field must be an int in the range [1, 1000]. *machine-type*::: The type of machine (name only) which has a fixed number of vCPUs and a fixed amount of memory. This also includes specifying custom machine type following `custom-number_of_CPUs-amount_of_memory` pattern, e.g. `custom-32-29440`. *min-cpu-platform*::: Optional minimum CPU platform of the reservation to create. *require-specific-reservation*::: Indicates whether the reservation can be consumed by VMs with "any reservation" defined. If enabled, then only VMs that target this reservation by name using `--reservation-affinity=specific` can consume from this reservation. ) reservationzreservation-zonezvm-countz machine-typezmin-cpu-platformzrequire-specific-reservationz--source-reservationrsourceT)rOrrz--dest-reservation destination)rrboolrrrformat)rz help_textreservation_specs rrrs )&*     $4 5   H %      $4 5   M *  -rcZ|r|nd}|jdtj|S)NzE" I*k// V $$F    O O4467+ rc|jdd}t||r t||jd}t j dj ||jddd gd d |jd }t|||r|jd d}t||r|jdd} t| yy)z0Adds all flags needed for reservations creation.z:Manage the reservations to be created with the commitment.Trz9Manage the reservation to be created with the commitment.)rF) positionalz--reservation-typespecificz*The type of the reservation to be created.)hiddenrrrz9Manage the specific SKU reservation properties to create.z7Manage the properties of a shared reservation to create)rrz?Manage the properties of a reservation sharing policy to createN) rrradd_argument_groupr GetReservationResourceArg AddArgumentrAddFlagsToSpecificSkuGroupAddFlagsToShareSettingGroup'AddFlagsToReservationSharingPolicyGroup) rzr{r|support_existing_reservationsr~reservations_manage_groupsingle_reservation_groupspecific_sku_reservation_groupshare_setting_reservation_group reservation_sharing_policy_groups rrurus%..B$/78"896II FJ))U;GG'' l 7 ($<#N#N F$O$ $&: 33J 4 $  ?@''?'R'R N(S($,,LM (rc^tjg}|D]}|j|y)z3Adds flags needed for a reservation sharing policy.N)rGetReservationSharingPolicyFlagrgroupr6res rrrs/779 $cOOErctjtjdtjtjdtj tj tjg}|r#|jtj|D]}|j|y)z6Adds flags needed for a specific sku zonal allocation.FrN) rGetRequireSpecificAllocationGetVmCountFlagGetMinCpuPlatformGetMachineTyperrGetResourcePolicyFlagr]instance_flagsAddMaintenanceIntervalr)rr|r6res rrr!s446&&6))+&&6'')**,--/ $KK5578 cOOErctjtjg}|D]}|j|y)z7Adds flags needed for an allocation with share setting.N)rrrrrs rrr4s;,,.((* $ cOOEr)FFFFF)Fr.)FFFF)>__doc__ __future__rrrr<googlecloudsdk.callioper"googlecloudsdk.command_lib.computercompute_completersrror r,googlecloudsdk.command_lib.compute.instancesr/googlecloudsdk.command_lib.compute.reservationsrr $googlecloudsdk.command_lib.util.apisr r@rrrrr>rCsorted_REQUIRED_RESOURCESListCommandCompleterrr,r0r7r9rKrUrarfrrrrrtrrrrwrxryrvrrrrrrrurrrrrrr s: >&'/OEEPVI: :& ; \):6%fh/0!3!H!H/  $N 4'  !&', .2 82 #L 'T%P ., "'', 3Nl&r