(ZdZddlmZddlmZddlmZddlmZdZdZdZ d d Z d Z y ) z7Utilities for AI Platform persistent resource commands.)absolute_import)division)unicode_literals) arg_utilsz1aiplatform.projects.locations.persistentResourcesc|jd}|jd}||jd}|jd}|rEtj||j|_t |jdd|_||_|jd}|rt ||_ |jd }|jd }|s|r=|jd } t || _ t || _ | |_ |jd } |jd } | r"|jd} | | | } | |_ |S)aConstructs a single resource pool spec. Args: aiplatform_client: The AI Platform API client used. spec: A dict whose fields represent a resource pool spec. Returns: A ResourcePoolSpec message instance for setting a resource pool in a Persistent Resource ResourcePool MachineSpecz machine-type) machineTypezaccelerator-typezaccelerator-countz replica-countzmin-replica-countzmax-replica-countResourcePoolAutoscalingSpecz disk-typez disk-sizeDiskSpec) bootDiskTypebootDiskSizeGb) GetMessagegetr ChoiceToEnumAcceleratorTypeValueValuesEnumacceleratorTypeintacceleratorCount machineSpec replicaCountminReplicaCountmaxReplicaCountautoscalingSpecdiskSpec)aiplatform_clientspec resource_poolmachine_spec_msg machine_specaccelerator_type replica_countmin_replica_countmax_replica_countautoscaling_spec disk_type disk_size disk_spec_msg disk_specs Rlib/googlecloudsdk/command_lib/ai/persistent_resources/persistent_resource_util.py _ConstructSingleResourcePoolSpecr,s_?#..~>@-&11-@!dhh~.FG,XX01#,#9#9*II$KL $'1Da(H$IL!*-((?+-!$]!3Mhh23hh23+C$$%BCE'*+<'=$'*+<'=$$4M!hh{#)hh{#)%00  r-Nc zg}t|jtr |j}|r|t||fi|z}|S)aYConstructs the resource pools to be used to create a Persistent Resource. Resource pools from the config file and arguments will be combined. Args: aiplatform_client: The AI Platform API client used. persistent_resource_config: A Persistent Resource configuration imported from a YAML config. resource_pool_specs: A dict of worker pool specification, usually derived from the gcloud command argument values. **kwargs: The keyword args to pass to construct the worker pool specs. Returns: An array of ResourcePool messages for creating a Persistent Resource. ) isinstance resourcePoolslistr4)rpersistent_resource_configr3r2resource_poolss r+ConstructResourcePoolsr;`sP,.*88$?/==N#&A.':28'::N r-c B||vxrt|j|S)N)boolr)keyr2s r+_IsKwargsDefinedr?s  04 300r-)NN) __doc__ __future__rrr$googlecloudsdk.command_lib.util.apisrPERSISTENT_RESOURCE_COLLECTIONr,r4r;r?r-r+rEs:>&':!T*Z6 $@1r-