58dZddlmZddlmZddlmZddlZddlmZmZm Z ddl m Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZGd deZGddeZy)z1Utilities for Backup and DR restore command apis.)absolute_import)division)unicode_literalsN)AnyDictList)encoding)alias_ip_range_utils) constants) exceptions)yaml)filesc deZdZdZedeeefdefdZ edeeefdefdZ ede jfdZ ede jded eefd Zede jd eeeeffd Zede jd eeeffdZede jdeeeffdZede jdededefdZede jdeeeffdZGddej.Zede jdefdZdZede jdedefdZy) ComputeUtilz1Util class for Restoring Compute Engine Instance.network_interfacereturnc d|vS)Nexternal-ipv6-addressrs 3lib/googlecloudsdk/api_lib/backupdr/restore_util.py_HasIpV6AccessConfigz ComputeUtil._HasIpV6AccessConfig$s "&7 77c d|vS)Naddressrrs r_HasIpV4AccessConfigz ComputeUtil._HasIpV4AccessConfig(s ) ))rclient_messagesc|yt}|D]f}|j}|j}|j}d|vr |d|_d|vr |d|_d|vr |d|_d|vr/|d|_d|vr |d|_ntjddd|vr |d|_ d |vr/|d |_ d |vr |d |_ ntjd dd |vrF|jj|d |_|jj|d |_d |vrtj |d |d |_d|vr#|jj%|d|_d|vr |d|_d|vr#|jj+|d|_d|vr |d|_t0j3|rB|jj4j6|_|j:j=|gnVt0j?|rA|jj4j@|_|jBj=|g|jE|i|S)a!Parses the network interface data into client messages. Args: client_messages: network_interfaces: A dictionary containing the network interface data Returns: List of parsed client messages for Network Interface Raises: InvalidArgumentException: Nnetworksubnetzprivate-network-ipzinternal-ipv6-addresszinternal-ipv6-prefix-lengthzDPrefix length of the provided IPv6 address is expected but not foundrrzexternal-ipv6-prefix-lengthz network-tieraliasesTz stack-typez queue-countznic-typeznetwork-attachment)#listNetworkInterface AccessConfigr subnetwork networkIP ipv6AddressinternalIpv6PrefixLengthr InvalidArgumentExceptionnatIP externalIpv6externalIpv6PrefixLengthNetworkTierValueValuesEnum networkTierr $CreateAliasIpRangeMessagesFromString aliasIpRangesStackTypeValueValuesEnum stackType queueCountNicTypeValueValuesEnumnicTypenetworkAttachmentrrTypeValueValuesEnum DIRECT_IPV6typeipv6AccessConfigsextendrONE_TO_ONE_NAT accessConfigsappend)rnetwork_interfacesmessagesrmessage access_configaccess_config_ipv6s rParserNetworkInterfacez"ComputeUtil.ParserNetworkInterface,s ! vH/002g%224m*779 ' '+I6 & &.x8 !2 2-.BC $5 5/0GH (,= =->+. ' *33+  ' '/ :  $5 5*; #+ ' ),= =8I+9  533+  , ,  ( ( C C!.1  !  ( ( C C!.1  & ' ' E E!),   * *  , , E E!,/   + +.}= ( (  , , C C!*-   !2 2$56J$K!  ) )*; <  ( ( < < H H  !!((*<)=>  + +,= >  ( ( < < K K  $$m_5oogi0j Orservice_accountscopeschd}|d}|tj}|j|||gS)aParses the service account data into client messages. Args: client_messages: service_account: An email id of the service account scopes: A list containing the scopes Returns: List of parsed client messages for Service Account ct}|D]4}tjj||g}|j |6|S)N)r"compute_constantsSCOPESgetr;)rFconverted_scopesscopes r_ConvertAliasToScopesz?ComputeUtil.ParserServiceAccount.._ConvertAliasToScopessD%!((,,UUG<&rdefault)emailrF)rIDEFAULT_SCOPESServiceAccount)rrErFrNs rParserServiceAccountz ComputeUtil.ParserServiceAccountsL!o ~ //f&&!*?*G '  rdisksc|yt}|D]v}|j}|j|_d|vr |d|_d|vr|d|j_d|vr|d|j_|j|x|S)zParses the disk data into client messages. Args: client_messages: disks: A list of dictionaries containing the disk data Returns: List of parsed client messages for Disk Nz device-namenamez replica-zones)r" AttachedDiskInitializeParamsinitializeParams deviceNamediskName replicaZonesr>)rrTr@diskrAs r ParserDiskszComputeUtil.ParserDiskss } vH,,.g!0!A!A!Cg $ !-0 4,0L  ) D 04_0E  -oog Ormetadatac |j|jDcgc]\}}|j||c}}Scc}}w)zParses the metadata data into client messages. Args: client_messages: metadata: A dictionary containing the metadata Returns: List of parsed client messages for Metadata keyvalue)items)MetadatardEntry)rr_rbrcs r ParseMetadatazComputeUtil.ParseMetadatasY  # #'nn. . U  ! !c ! 7.  $  sA labelsc |jj|jDcgc],\}}|jjj||.c}}Scc}}wzParses the labels data into client messages. Args: client_messages: labels: A dictionary containing the labels Returns: List of parsed client messages for Labels ra)additionalProperties) ComputeInstanceRestoreProperties LabelsValuerdAdditionalPropertyrrhrbrcs r ParseLabelszComputeUtil.ParseLabelssu  ; ; G G %lln  - U  < < H H [ [u \ -   H  1A# enable_uefi_networkingthreads_per_corevisible_core_countcj|||y|j}|||_|||_|||_|S)aParses the advanced machine features data into client messages. Args: client_messages: enable_uefi_networking: threads_per_core: visible_core_count: Returns: List of parsed client messages for AdvancedMachineFeatures N)AdvancedMachineFeaturesenableUefiNetworkingthreadsPerCorevisibleCoreCount)rrrrsrtrAs rParseAdvancedMachineFeaturesz(ComputeUtil.ParseAdvancedMachineFeaturessV& &  $  & 557G)%;g"#/g%!3g Nr acceleratorc^|d|vry|j|d|jddgS)zParses the accelerator data into client messages. Args: client_messages: accelerator: A dictionaries containing the accelerator data Returns: List of parsed client messages for Accelerator Nr9count)acceleratorTypeacceleratorCount)AcceleratorConfigrK)rr{s rParseAcceleratorzComputeUtil.ParseAcceleratorsIfK7  ))'/(__Wa8 *  rceZdZdZy)&ComputeUtil.NodeAffinityFileParseErrorz6Error raised when node affinity file cannot be parsed.N)__name__ __module__ __qualname____doc__rrrNodeAffinityFileParseErrorr2s@rr file_pathcz|ytj|}tj|}|stj dg}|D]}|stj d t j|j|}|jstj d|jr=tj dj|j|j|j||S#t$r*}tj dj||d}~wwxYw)aParses the node affinity data from file into client messages. Args: client_messages: file_path: A path to the file containing the node affinity data. Returns: List of parsed client messages for NodeAffinity Raises: NodeAffinityFileParseError: NziNo node affinity labels specified. You must specify at least one label to create a sole tenancy instance.z"Empty list item in JSON/YAML file.z6Failed to parse node affinity values from the file {}.z6A key must be specified for every node affinity label.z,Key [{0}] has invalid field formats for: {1})rReadFileContentsr loadrrr PyValueToMessage NodeAffinity Exceptionformatrball_unrecognized_fieldsr>)rrnode_affinities_fileaffinities_yamlnode_affinitiesaffinity node_affinityes rGetNodeAffinitiesFromFilez%ComputeUtil.GetNodeAffinitiesFromFile5sI"  11)<ii 45O   2 2 5  O# 44 0    11  ( ((   44 D    . . 044 : A A!!=#H#H#J   ]+3$4 #44 D K K    s$ D D:%D55D:z'compute.googleapis.com/reservation-namereservation_affinity reservationc|y|dk(r0|j|jjjS|dk(r0|j|jjjS|dk(rY|t j dd|j|jjj tj|gSy) aFParses the reservation affinity data into client messages. Args: client_messages: reservation_affinity: type of reservation affinity reservation: name of the specific reservation Returns: List of parsed client messages for ReservationAffinity Raises: InvalidArgumentException: Nany)consumeReservationTypenonespecificrz9Reservation is required for specific reservation affinity)rrbvalues) AllocationAffinity%ConsumeReservationTypeValueValuesEnumANY_RESERVATIONNO_RESERVATIONr r)SPECIFIC_RESERVATIONrRESERVATION_AFFINITY_KEY)rrrs rParseReservationAffinityz$ComputeUtil.ParseReservationAffinityps&# u$  / /!0!C!C!i!i!y!y0v%  / /!0!C!C!i!i!x!x0z)  11  G   / /!0!C!C!i!i!~!~220 rN)rrrr staticmethodrstrrboolrrtypes ModuleTyperDrrSr^rgrpintrzrcore_exceptionsErrorrrrrrrrrr!s798d38n888*d38n***g''ggR'':=GKCy>''04T#s(^0D6''37S>&5#3#3T#s(^&''" @''6:38n,A?#8#8A6''64766pG*''****rrcHeZdZdZedej deee ffdZ y)DiskUtilzUtil class for Restoring Disk.rrhc |jj|jDcgc],\}}|jjj||.c}}Scc}}wrj)DiskRestorePropertiesrmrdrnros rrpzDiskUtil.ParseLabelsst  0 0 < < %lln  - U  1 1 = = P Pu Q -   =  rqN) rrrrrrrrrrrprrrrrs3&5#3#3T#s(^rr)r __future__rrrrtypingrrrapitools.base.pyr googlecloudsdk.api_lib.computer r rIgooglecloudsdk.callioper googlecloudsdk.corerr googlecloudsdk.core.utilrobjectrrrrrrsL8&' ""%?I.=$*z&zz vr