$dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Zdd lmZdd lmZdd lmZdd lmZGdde j.Zy)z2Command for setting whether to auto-delete a disk.)absolute_import)division)unicode_literals) base_classes)instance_utils)base) exceptions)scope)flags)log) resources) console_iocReZdZdZddiZedZdZdZdZ dZ d Z d Z d Z y ) SetDiskAutoDeletea Set auto-delete behavior for disks. *${command}* is used to configure the auto-delete behavior for disks attached to Compute Engine virtual machines. When auto-delete is on, the persistent disk is deleted when the instance it is attached to is deleted. EXAMPLESay To enable auto-delete for a disk named 'my-disk' on an instance named 'my-instance', run: $ {command} my-instance --auto-delete --disk=my-disk To enable auto-delete for a device named 'my-device' on an instance named 'my-instance', run: $ {command} my-instance --auto-delete --device-name=my-device ctjj||jdddd|j d}|jdd |jd d y) Nz --auto-delete store_trueTz'Enables auto-delete for the given disk.)actiondefaulthelp)requiredz--diskz Specifies a disk to set auto-delete for by its resource name. If you specify a disk to set auto-delete for by persistent disk name, then you must not specify its device name using the ``--device-name'' flag. )rz --device-namea Specifies a disk to set auto-delete for by its device name, which is the name that the guest operating system sees. The device name is set at the time that the disk is attached to the instance, and need not be the same as the persistent disk name. If the disk's device name is specified, then its persistent disk name must not be specified using the ``--disk'' flag. )r INSTANCE_ARG AddArgument add_argumentadd_mutually_exclusive_group)parser disk_groups 5lib/surface/compute/instances/set_disk_auto_delete.pyArgszSetDiskAutoDelete.Args9s ""6*  6 8 44d4CJ     cltjj||tj|S)N) scope_lister)r rResolveAsResourceGetInstanceZoneScopeLister)selfclientr argss rCreateReferencez!SetDiskAutoDelete.CreateReferenceYs4    / / ie&F&Fv&N 0 PPr c|jjd|jjdi|j fS)NGet)apitools_client instancesmessagesComputeInstancesGetRequestAsDict)r%r& instance_refs r GetGetRequestzSetDiskAutoDelete.GetGetRequest]s@  " " , ,  6FOO 6 6 O9L9L9N O QQr c |jjd|jj|j|j |j |j|jfS)Nr) deviceNameinstanceprojectzone autoDelete) r,r-r.(ComputeInstancesSetDiskAutoDeleteRequestr4r5r6r7r8)r%r&r1 attached_disks r GetSetRequestzSetDiskAutoDelete.GetSetRequestbsc  " " , ,  OO D D(33%..$,,!&&(33 E 5 66r cg} tj|||j|jtj j }|j| tj|||j|jtj j}|j||S#tj$rYnwxYw#tj$rY|SwxYw)azGets the possible disks that match the provided disk name. First, we attempt to parse the provided disk name as a regional and as a zonal disk. Next, we iterate over the attached disks to find the ones that match the parsed regional and zonal disks. If the disk can match either a zonal or regional disk, we prompt the user to choose one. Args: resources: resources.Registry, The resource registry name: str, name of the disk. instance_ref: Reference of the instance instance. Returns: List of possible disks references that possibly match the provided disk name. ) rParseDiskResourcer6r7compute_scopes ScopeEnumREGIONappendcloud_resources WrongResourceCollectionExceptionZONE)r%r namer1possible_disks regional_disk zonal_disks r_GetPossibleDisksz#SetDiskAutoDelete._GetPossibleDisksls&N $66 T<//1B1B  " " ) )+mM* !33 T<//1B1B  " " ' ')jJ'   ; ;     ; ;   s%AB3AC 3C C  C#"C#c|j|||}g}|jD]U}tj||}|D]8} | s| j |j k(s(|j |:W|s:t jdj||j|jt|dk(r|dSg} |D]7}tj||} | j | j 9tj| dj|} | :t jdj||j|j|| S)aGets an attached disk with the specified disk name. First, we attempt to parse the provided disk name to find the possible disks that it may describe. Next, we iterate over the attached disks to find the ones that match the possible disks. If the disk can match multiple disks, we prompt the user to choose one. Args: resources: resources.Registry, The resource registry name: str, name of the attached disk. instance_ref: Reference of the instance instance. instance: Instance object. Returns: An attached disk object. Raises: exceptions.ArgumentError: If a disk with name cannot be found attached to the instance or if the user does not choose a specific disk when prompted. z8Disk [{}] is not attached to instance [{}] in zone [{}].rz([{}] matched multiple disks. Choose one:)optionsmessagezJFound multiple disks matching [{}] attached to instance [{}] in zone [{}].)rIdisksr!ParseDiskResourceFromAttachedDisk RelativeNamerAcompute_exceptions ArgumentErrorformatr5r7lenr PromptChoice) r%r rEr1r5rFmatched_attached_disksr: parsed_diskd choice_namesdisk_refidxs rGetAttachedDiskByNamez'SetDiskAutoDelete.GetAttachedDiskByNamesl.++It\JN! "DD ]$k! ![%=%=%?? ' ' 6( "  , , D 6$ --|/@/@ A CC # $ ) #A &&L/ AA ]$h(//120  ! !:AA$G IC {  , ,  6$ --|/@/@ A CC "# &&r c|jD]}|j|k(s|cStjdj ||j |j )aGets an attached disk with the specified device name. Args: resources: resources.Registry, The resource registry device_name: str, device name of the attached disk. instance_ref: Reference of the instance instance. instance: Instance object. Returns: An attached disk object. Raises: compute_exceptions.ArgumentError: If a disk with device name cannot be found attached to the instance. zHNo disk with device name [{}] is attached to instance [{}] in zone [{}].)rNr4rQrRrSr5r7)r%r device_namer1r5disks rGetAttachedDiskByDeviceNamez-SetDiskAutoDelete.GetAttachedDiskByDeviceNamesY" K '   * *   \22L4E4E F HHr ctj|j}|j}|j ||j |}|j ||}|j|g}|jr,|j|j |j||d}n+|j|j |j||d}|j|jk(r=tjj!dj#|dj$|S|j|_ |j|j'|||gS)Nrz.No change requested; skipping update for [{}].)rComputeApiHolder ReleaseTrackr&r(r r2 MakeRequestsr_r\r`r^r8 auto_deleter statusPrintrSrEr;)r%r'holderr&r1 get_requestobjectsr_s rRunzSetDiskAutoDelete.Runs8  * *4+<+<+> ?F ]]F''0@0@$GL$$V\:K!!;-0G yy  ' '(8(8$))(4gajBd - -f.>.>.2.>.> .5aj:d $*** jj : A Aajoo  n&&DO     FL$ 78 ::r N)__name__ __module__ __qualname____doc__ detailed_help staticmethodrr(r2r;rIr\r`rkr+r rrr"sT  -  >PQ 6#J5'nH4:r rN)ro __future__rrrgooglecloudsdk.api_lib.computerrgooglecloudsdk.callioper"googlecloudsdk.command_lib.computer rQr r>,googlecloudsdk.command_lib.compute.instancesr googlecloudsdk.corer r rBgooglecloudsdk.core.consoler UpdateCommandrr+r rrzsD 9&'89(OF>#<2X:**X:r