-zdZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z d Z d Zd Zd Zd Zy)zGValidation of the arguments for the persistent-resources command group.)absolute_import)division)unicode_literals)util) exceptions) constants) validationcNtj|tjy)zFValidate whether the given region is allowed for persistent resources.)available_regionsN)r ValidateRegionrSUPPORTED_TRAINING_REGIONS)regions Dlib/googlecloudsdk/command_lib/ai/persistent_resources/validation.pyr r s  D DFc"|jrt|j|t|jtrt |j||js1t|jtst jddyy)z?Validate the argument values specified in the `create` command.--resource-pool-speczNo resource pools specified. At least one resource pool must be provided via a YAML config file (--config) or via the --resource-pool-spec arg.N)resource_pool_spec_ValidateResourcePoolSpecArgs isinstance resourcePoolslist$_ValidateResourcePoolSpecsFromConfigrInvalidArgumentException)argspersistent_resource_configversions rValidateCreateArgsr!s !$"9"97C*88$?("00';  ! ! /==t D  - - $ %% E "rcR|dstjddt||y)a Validates the argument values specified via `--resource-pool-spec` flags. Args: resource_pool_specs: List[dict], a list of resource pool specs specified via command line arguments. version: str, the API version this command will interact with, either GA or BETA. rrzEEmpty value is not allowed for the first `--resource-pool-spec` flag.N)rr'_ValidateHardwareInResourcePoolSpecArgs)resource_pool_specsrs rrr2s4 Q   - -O QQ**=wGrc |D]F}|sd|vrtjddd|vrd|vrtjddd|vrd|vrtjddd|vrd|vrtjdd d |vd |vk7rtjdd |jd d }|stjd|j }|j Dcgc]}|jdr|}}||vstjddj|djdt|Dy cc}w)a'Validates the hardware related fields specified in `--resource-pool-spec` flags. Args: resource_pool_specs: List[dict], a list of resource pool specs specified via command line arguments. version: str, the API version this command will interact with, either GA or BETA. z machine-typerz9Key [machine-type] required in dict arg but not provided.zmin-replica-countzmax-replica-countzVKey [max-replica-count] required in dict arg when key [min-replica-count] is provided.zVKey [min-replica-count] required in dict arg when key [max-replica-count] is provided.z replica-countz:Key [replica-count] required in dict arg but not provided.zaccelerator-countzaccelerator-typezPKey [accelerator-type] and [accelerator-count] are required to use accelerators.N MachineSpecNVIDIAzWFound invalid value of [accelerator-type]: {actual}. Available values are [{expected}]., c3 K|]}|ywN.0vs r z:_ValidateHardwareInResourcePoolSpecArgs..ys%E1DAa1D actualexpected) rrgetapi_util GetMessageAcceleratorTypeValueValuesEnumnames startswithformatjoinsorted)r rspecaccelerator_type type_enumtype valid_typess rrrCs"d t #11 " GI I  %,?t,K11 " /0 0  %,?t,K11 " /0 0  %,?t,K11 " HJ J  %+=+E F11 " ! ! "4d; '' 7$$B$B '__. .Tx( .   ; .33$45;V* II%E 1D%EE6<6GH HQ"F s$Ec |stjdd|D]X}|jstjdd|jjstjdd|js"|j stjdd|j r,|j j stjdd|j r,|j jstjdd|jjr,|jjstjdd|jrB|jjr,|jjstjdd |jjst|jjj}tj d |j"}|j%Dcgc]}|j'd r|}}||vstjdd j)|d j+dt-|Dycc}w)zJValidate ResourcePoolSpec message instances imported from the config file.z--configzDAt least one [resourcePools] required in but not provided in config.z;Field [machineSpec] required in but not provided in config.z;Field [machineType] required in but not provided in config.z.s#C/B!A/Br,r-N)rr machineSpec machineType replicaCountautoscalingSpecminReplicaCountmaxReplicaCountacceleratorCountacceleratorTypediskSpecbootDiskSizeGb bootDiskTypestrnamer1r2r3r4r5r6r7r8)resource_poolsrr9r:r;r<r=s rrr|sc   - -N PPd     / /  G II    ' '  / /  G II   (<(<  / /  H JJ t';';'K'K  / /  C EE t';';'K'K  / /  C EE ))    , ,  / /     }}$--66"mm88  / /     ''T--==BBC%% ""@"@%??,,4 __X & ,  ,11 2396())#Cvk/B#CC4:4EF FcXsI8N)__doc__ __future__rrrgooglecloudsdk.api_lib.airr1googlecloudsdk.calliopergooglecloudsdk.command_lib.airr r rrrrr'rrrSs>N&'6.34F %"H"6Hr<Fr