dZddlmZddlmZddlmZddlZddlmZddlZGddeZ Gd d ejeje Z Gd d e ZGd de ZdZdZy)zYUtilities for handling Compute InstantSnapshotsService and RegionInstantSnapshotsService.)absolute_import)division)unicode_literalsN)ErrorceZdZdZy)UnknownResourceErrorzORaised when a instant snapshot resource argument is neither regional nor zonal.N)__name__ __module__ __qualname____doc__=lib/googlecloudsdk/api_lib/compute/instant_snapshots_utils.pyrrsWrrceZdZdZdZdZejdZejdZ ejdZ y)_CommonInstantSnapshotz4Common class for InstantSnapshot Service API client.c|jSN)_serviceselfs r GetServicez!_CommonInstantSnapshot.GetService!s ==rcX|j}|jj|Sr) GetInstantSnapshotRequestMessagerGet)r request_msgs rGetInstantSnapshotResourcez1_CommonInstantSnapshot.GetInstantSnapshotResource$s%779K ==  [ ))rctrNotImplementedErrorrs rrz7_CommonInstantSnapshot.GetInstantSnapshotRequestMessage( rctrrrs rGetSetLabelsRequestMessagez1_CommonInstantSnapshot.GetSetLabelsRequestMessage,r rctrrrs r)GetSetInstantSnapshotLabelsRequestMessagez@_CommonInstantSnapshot.GetSetInstantSnapshotLabelsRequestMessage0r rN) r r r r rrabcabstractmethodrr"r$r rrrrs]<*rrc8eZdZdZdZedZdZdZdZ y)_InstantSnapshotz@A wrapper for Compute Engine InstantSnapshotsService API client.cztj|||_||_|j|_||_yr)r__init___ips_ref_clientinstantSnapshotsr _messagesrclientips_refmessagess rr*z_InstantSnapshot.__init__8s2##D)DMDL++DMDNrcy)Nzcompute.zoneOperationsr clss rGetOperationCollectionz'_InstantSnapshot.GetOperationCollection?s #rcj|jjdi|jjSNr )r.!ComputeInstantSnapshotsGetRequestr+AsDictrs rrz1_InstantSnapshot.GetInstantSnapshotRequestMessageCs0 ;4>> ; ; " --    ""rc.|jjSr)r.ZoneSetLabelsRequestrs rr"z+_InstantSnapshot.GetSetLabelsRequestMessageGs >> . ..rc |jj}||jj|jj|jj |jj |j|S)NlabelFingerprintlabels)projectresourcezonezoneSetLabelsRequest)r.'ComputeInstantSnapshotsSetLabelsRequestr+rAinstantSnapshotrCr<r?ripsr@reqs rr$z:_InstantSnapshot.GetSetInstantSnapshotLabelsRequestMessageJsk .. @ @C  %%.. ]]  !^^@@ 11&AB  CCrN r r r r r* classmethodr6rr"r$r rrr(r(5s.H$$"/Crr(c8eZdZdZdZedZdZdZdZ y)_RegionInstantSnapshotzEA wrapper for Compute Engine RegionInstantSnapshotService API client.cztj|||_||_|j|_||_yr)rr*r+r,regionInstantSnapshotsrr.r/s rr*z_RegionInstantSnapshot.__init__Ws2##D)DMDL11DMDNrcy)Nzcompute.regionOperationsr r4s rr6z-_RegionInstantSnapshot.GetOperationCollection^s %rcj|jjdi|jjSr8)r.'ComputeRegionInstantSnapshotsGetRequestr+r:rs rrz7_RegionInstantSnapshot.GetInstantSnapshotRequestMessagebs0 A4>> A A " --    ""rc.|jjSr)r.RegionSetLabelsRequestrs rr"z1_RegionInstantSnapshot.GetSetLabelsRequestMessagefs >> 0 00rc |jj}||jj|jj|jj |jj |j|S)Nr>)rArBregionregionSetLabelsRequest)r.-ComputeRegionInstantSnapshotsSetLabelsRequestr+rArFrVrTr?rGs rr$z@_RegionInstantSnapshot.GetSetInstantSnapshotLabelsRequestMessageisk .. F FC  %%..}}###~~DD 11& E B  CCrNrJr rrrMrMTs.M&&"1CrrMc|jdk(ry|jdk(rytdj|j)aChecks if a compute instant snapshot is zonal or regional. Args: ips_ref: the instant snapshot resource reference that is parsed from resource arguments to modify. Returns: Boolean, true when the compute instant snapshot resource to modify is a zonal compute instant snapshot resource, false when a regional compute instant snapshot resource. Raises: UnknownResourceError: when the compute instant snapshot resource is not in the correct format. zcompute.instantSnapshotsTzcompute.regionInstantSnapshotsFz3Unexpected instant snapshot resource argument of {}) Collectionrformat)r1s rIsZonalr\ssU( 77 ??  =DD     " ##rcLt|r t|||St|||S)a8Gets the zonal or regional instant snapshot api info. Args: ips_ref: the instant snapshot resource reference that is parsed from resource arguments. client: the compute api_tools_client. messages: the compute message module. Returns: _ZoneInstantSnapshot or _RegionInstantSnapshot. )r\r(rM)r1r0r2s rGetInstantSnapshotInfor^s* W FGX 66 !&'8 <ress`&' 0 X5X/S// VD.C-C>C3C>#<=r