<,dZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z dd lm Z dd l m Z dd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZdd lmZddlmZddlmZddlm Z dZ!GddejDZ#ejHejJejLjNejLjPejLjRGddejTZ+ddde+_,y)z2Command for converting a disk to a different type.)absolute_import)division)unicode_literalsN) base_classes) disks_util) kms_utils)name_generator)poller)waiter)base) exceptions) completers)flags) resource_args)log) console_io)progress_trackera>This command will permanently convert disk {0} to disk type: {1}. Please detach the disk from all instances before continuing. Data written to the original disk during conversion will not appear on the converted disk. Please see https://cloud.google.com/compute/docs/disks/automatically-convert-disks for more details.c$eZdZdZdZdZdZdZdZy) _ConvertStateN) __name__ __module__ __qualname__SNAPSHOT_CREATED DISK_RESTOREDORIGINAL_DISK_DELETEDORIGINAL_DISK_RECREATEDRESTORED_DISK_DELETEDSNAPSHOT_DELETED$lib/surface/compute/disks/convert.pyrr4s%-r&rceZdZdZej dZedZdZ ddZ dZ d Z dd Z d Zd Z dd ZdZddZdZdZdZdZdZy)ConvertzFConvert a Compute Engine Persistent Disk volume to a Hyperdisk volume.F)pluralctjj||jdtj ddt j|ddy)N--target-disk-typeTzSpecifies the type of Hyperdisk to convert to, for example, to convert a Hyperdisk Balanced volume, specify `hyperdisk-balanced`. To get a list of available disk types, run `gcloud compute disk-types list`. ) completerrequiredhelpdisk)region_fallthrough)r) _DISK_ARG AddArgument add_argumentrDiskTypesCompleterkms_resource_argsAddKmsKeyResourceArg)parsers r'Argsz Convert.ArgsFsT !!&) // **4r&c tj|j|_|jjj |_|jjj |_d|_i|_d|_ |jj||jjtj|jj|_|jj!dk(rt#j$dd|j&dk(rt#j$dd|j&|_t)j*|j|j|j }|j-}|j.rt#j0d t3j4t7j8t:j=|jj?|j&d  |jA|jj?5|jC|j|j&|jDtGjH||j d }ddd|jM|jr{|jtNjPtNjRtNjTfvr!tWjX|jStWjZ|jS#1swYxYw#tJ$r}|d}~wwxYw#|jM|jr{|jtNjPtNjRtNjTfvr tWjX|jwtWjZ|jwwxYw) N) scope_listerzcompute.regionDisksz--regionz2Regional disks are not supported for this command.z hyperdisk-mlr,z/Hyperdisk ML is not supported for this command.zHDisk is attached to instances. Please detach the disk before converting.T)message cancel_on_nodisk_encryption_key).rComputeApiHolder ReleaseTrackholderclientapitools_clientmessagesstatecreated_resources user_messagesr2ResolveAsResource resourcesrGetDefaultScopeListerdisk_ref Collectionr InvalidArgumentExceptiontarget_disk_typer GetDiskInfoGetDiskResourceusers ToolExceptionrPromptContinuetextwrapdedentCONTINUE_WITH_CONVERT_PROMPTformatName_CreateProgressTracker _ConvertDisksizeGbrMaybeGetKmsKey Exception_CleanUprr"r#r$rwarningerror)selfargs disk_info original_diskresultes r'Runz Convert.RunVs//0A0A0CDDK++$$44DKKK&&//DMDJDDNN44  001C1CD5DM  }}!%::  / /  >   .  / /  ; !11D&& t{{DMM3I--/M  $ $   ( / / ""$d&;&;   (  & &t}}'9'9'; <"" MM400-2F2F ) 8 8dmmT!#  = mmo   ::-??'=='88: : ++d(( ) M ))D&& ' M' = <  g mmo   ::-??'=='88: : ++d(( ) ))D&& ' s>)L??AL3L?3L<8L?? MM  MMBO+Nc|j||_|j||j}|jjj |jd|j id}|jtj||j||_ |jjj |j|j |jdd}|j|||||xs|}|jtj||j|xs|}|jtj|j!|j#||xs|}|jtj$|j|xs|}|jtj&|j)|xs|}|jtj*|S)Nprojectzcompute.snapshots)params collection)rkzonez compute.disksr?) _GenerateName snapshot_name_InsertSnapshotrCrKParserk _UpdateStaterrrestored_disk_namern_RestoreDiskFromSnapshotr _DeleteDiskr! _CloneDiskrZr"r#_DeleteSnapshotr$)rcrMrPsize_gbr@rg snapshot_refrestored_disk_refs r'r\zConvert._ConvertDisks++H5D  ! !(D,>,> ?F;;((.. 8++,&/L  m44lC"00:D --33 #++X]]C"4 %%     3 &     m113DE   h ' 16Fm99:  MMO  3       m;;<  / 0 :FFm99: ! !, / 96Fm445 Mr&c@|jj|j|jj||j |jjj j }|j|jjd|}|jjj|jd}tjt!j"|jj||j%dS)N)name sourceDisk snapshotType)rksnapshotInsertcompute.globalOperationsrmcustom_tracker max_wait_ms)rFComputeSnapshotsInsertRequestrkSnapshotSelfLinkSnapshotTypeValueValuesEnumSTANDARD _MakeRequestrD snapshotsrCrKrrselfLinkr WaitForr Poller_CreateNoOpProgressTracker)rcrMrprequest operation operation_refs r'rqzConvert._InsertSnapshotsmm99  ''((*//KKTT( :G!!$++"7"77KIKK))//-0M >> dkk++,668  r&c|jj|j|j}|j |j j d|}|jjj|jd}tjtj|j j ||jdS)N)rrkDeleterrr)rFComputeSnapshotsDeleteRequestrZrkrrDrrCrKrrrr rr DeletePollerr)rcrzrrrs r'rxzConvert._DeleteSnapshotsmm99""$$$:G!!$++"7"77KIKK))//-0M >>DKK112668  r&c |i}|r||d<|jjd|jtj|||j ||j d|}|jj||j|j}|j|jjd|} |j jj| jd} t!j"t%j&|jj| |j)dS) NdiskEncryptionKey)r}typer]sourceSnapshotr0rkrnrcompute.zoneOperationsrrr%)rFDiskrZrGetDiskTypeUrirCrComputeDisksInsertRequestrkrnrrDdisksrKrrrr rr rr) rcr{rz disk_typeryr@kwargsr0rrrs r'ruz Convert._RestoreDiskFromSnapshots)F$7f ! 4==     # # %  & & ($++ #,,.    Dmm55 !))  # #6G !!$++"3"3XwGIKK))//+0M >> dkk''(668  r&cVtjd|jddS)N-@)r GenerateRandomNamerZ)rc resource_refs r'rozConvert._GenerateNames.//1 2!L4E4E4G3H I#2 NNr&c|jj|j|j|j}|j |j jd|}|jjj|jd}tjtj|j j||j!dS)Nrrrrr)rFComputeDisksDeleteRequestrZrkrnrrDrrCrKrrrr rr rr)rcrMrrrs r'rvzConvert._DeleteDisksmm55 ]]_  ]]6G !!$++"3"3XwGIKK))//+0M >>DKK--.668  r&c"i}|r||d<|jjd||jd|}|jj||j|j }|j |jjd|}|jjj|jd}tj|jj} tj | ||j#dS) Nr)r}r~rrrrrr%)rFrrrrkrnrrDrrCrKrrrr rr rr) rcoriginal_disk_namer{r@rr0rrroperation_pollers r'rwzConvert._CloneDisk1sF$7f ! 4==    $--/   D mm55 !))  # #6G !!$++"3"3XwGIKK))//+0M}}T[[%6%67 >>668  r&cg}|jjj|||fg|}|rtj||stj d|dS)NzNo response receivedr)rCrD AsyncRequestscore_exceptions MultiError InternalError)rcresource_clientmethodrerrors_to_collect responsess r'rzConvert._MakeRequestNsh ""00 67 +,.?I  & &'8 99   ) )*@ AA Q<r&c6||_|r||j|<yyN)rGrH)rcrGcreated_resources r'rszConvert._UpdateStateYs!DJ&6dU#r&cR|jsd|jz|_y|jtjk(rSd|j d|jz|_|j |jtjy|jtjk(rd|jjd|jz|jz|_|j|jtj|j |jtjy|jtjk(rd|jjd|jjd|jjd|jd |j d |j d |j d|jjd |_y|jtj"k(r8d |j d|jz|jz|_y|jtj$k(r'd|j d|jz|_yy)NzCreating snapshot failed.zCreating disk from snapshot z failed. zDeleting original disk zRecreating original disk z1 failed. Please run `gcloud compute disks create  --zone=z --type=z --source-disk=zM` to recreate the original disk. Please run `gcloud compute snapshots delete zL` to delete the temporary snapshot. Please run `gcloud compute disks delete z` to delete the temporary disk.z;Conversion completed successfully, Deleting temporary disk z failed.z?Conversion completed successfully. Deleting temporary snapshot )rG_BuildCleanupSnapshotMessagerIrrrprxrHr rMrZ_BuildCleanupDiskMessagervr!rnrPrtr"r#rcs r'r`zConvert._CleanUp^s :: %(I(I(K K   zz]333 )););(,>+?xH]H]G^^mnroFoFnGG#1123%% &ht}}/A/A.BC!  !  }<< < %% &h 0  ) ) + ,  - - / 0  }:: :   ! +  - - / 0  ;r&c8tjd|ddS)NzConverting disk z...zConversion aborted.)r=aborted_message)rProgressTracker)rc disk_names r'r[zConvert._CreateProgressTrackers$  + +"9+S1- r&c0tjddS)NTr;) interruptabler)rNoOpProgressTrackerrs r'rz"Convert._CreateNoOpProgressTrackers  / /B r&c"d|jdS)Nz- Please run `gcloud compute snapshots delete z6` to delete the temporary snapshot if it still exists.)rprs r'rz$Convert._BuildCleanupSnapshotMessages 78J8J7KL= =r&cPd|jd|jjdS)Nz) Please run `gcloud compute disks delete rz2` to delete the temporary disk if it still exists.)rtrMrnrs r'rz Convert._BuildCleanupDiskMessages4  # # $HT]]-?-?,@A. .r&r)rrr__doc__ disks_flags MakeDiskArgr2 staticmethodr9rir\rqrxrurorvrwrrsr`r[rrrr%r&r'r)r)=s O%k%%U3)  AHFJ5n*. $LO&HL: 7 3j   r&r)a  Convert Compute Engine Persistent Disk volumes to Hyperdisk volumes. *{command}* converts a Compute Engine Persistent Disk volume to a Hyperdisk volume. For a comprehensive guide, refer to: https://cloud.google.com/sdk/gcloud/reference/compute/disks/convert. zThe following command converts a Persistent Disk volume to a Hyperdisk Balanced volume: $ {command} my-disk-1 --zone=ZONE --target-disk-type=hyperdisk-balanced ) DESCRIPTIONEXAMPLES)-r __future__rrrenumrVgooglecloudsdk.api_lib.computerrrr )googlecloudsdk.api_lib.compute.operationsr googlecloudsdk.api_lib.utilr googlecloudsdk.callioper r "googlecloudsdk.command_lib.computerr(googlecloudsdk.command_lib.compute.disksrgooglecloudsdk.command_lib.kmsrr6googlecloudsdk.corerrgooglecloudsdk.core.consolerrrXEnumrDefaultUniverseOnly ReleaseTracksrBALPHABETAGARestoreCommandr) detailed_helpr%r&r'rs9&' 7549<.(.94IM=#28DIID%%++%%**%%((*hd!!h*hX   r&