s_<dZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd l m Z dd lmZdd lmZdd lmZddlmZeZ ddZ ddZ ddZ ddZ ddZ ddZdZ y)z:Convenience functions for dealing with instance templates.)absolute_import)division)unicode_literals)alias_ip_range_utils) constants) image_utils)instance_utils) kms_utils)utils)scope)flags) labels_util) propertiesNc0|j}|Ytjj|gtj j |||d}|j|_|X|j|dtjjjjid}|j|_ni|g|e|jt j"dtjjjjid}|j|_|||_| |jj'| |_|r~| dk7ry|j+t j,|j*j.j0}|t2k7r||_| |j*j7| |_|g|_d}| | G|j+t j<|j*j.j>}|g|_ | |j*j7| |_| | |_!||}||}|P|G|j+t j<|j*j.j>}|g|_ ||_"|P|G|j+t j<|j*j.j>}|g|_ ||_#|||_$|||_%|rtMjN|d ||_(| |jjS| |_*|||_+|||_,|||_-| |jj]||_/|r||_0|S) a Creates and returns a new NetworkInterface message. Args: resources: generates resource references, scope_lister: function, provides scopes for prompting subnet region, messages: GCE API messages, network: network, private_ip: IPv4 internal IP address to assign to the instance. subnet_region: region for subnetwork, subnet: regional subnetwork, address: specify static address for instance template * None - no address, * EPHEMERAL_ADDRESS - ephemeral address, * string - address name to be fetched from GCE API. alias_ip_ranges_string: command line string specifying a list of alias IP ranges. network_tier: specify network tier for instance template * None - no network tier * PREMIUM - network tier being PREMIUM * SELECT - network tier being SELECT * STANDARD - network tier being STANDARD stack_type: identify whether IPv6 features are enabled * IPV4_ONLY - can only have IPv4 address * IPV4_IPV6 - can have both IPv4 and IPv6 address * IPV6_ONLY - can only have IPv6 address ipv6_network_tier: specify network tier for IPv6 access config * PREMIUM - network tier being PREMIUM * STANDARD - network tier being STANDARD nic_type: specify the type of NetworkInterface Controller * GVNIC * VIRTIO_NET ipv6_public_ptr_domain: a string represents the custom PTR domain assigned to the interface. ipv6_address: a string represents the external IPv6 address reserved to the interface. ipv6_prefix_length: a string represents the external IPv6 address prefix length reserved to the interface. external_ipv6_address: a string represents the external IPv6 address reserved to the interface. external_ipv6_prefix_length: a string represents the external IPv6 address prefix length reserved to the interface. internal_ipv6_address: a string represents the internal IPv6 address reserved to the interface. internal_ipv6_prefix_length: the internal IPv6 address prefix length of the internal IPv6 address reserved to the interface. network_attachment: URL of a network attachment to connect the interface to. network_queue_count: the number of queues assigned to the network interface. vlan: the VLAN tag of the network interface. igmp_query: the IGMP query mode of the network interface. enable_vpc_scoped_dns: If True, indicates that this network interface can be used to send DNS queries to the VPC network's scope. Returns: network_interface: a NetworkInterface message object N) scope_listerrprojectzcompute.networks)params collection IPV6_ONLY)nametypeF)1NetworkInterface subnet_flagsSubnetworkResolverResolveResources compute_scope ScopeEnumREGIONSelfLink subnetworkParserVALUEScorer GetOrFailnetworkrDEFAULT_NETWORK networkIPStackTypeValueValuesEnum stackType AccessConfigDEFAULT_ACCESS_CONFIG_NAMETypeValueValuesEnumONE_TO_ONE_NATEPHEMERAL_ADDRESSnatIPNetworkTierValueValuesEnum networkTier accessConfigsDEFAULT_IPV6_ACCESS_CONFIG_NAME DIRECT_IPV6ipv6AccessConfigspublicPtrDomainName externalIpv6externalIpv6PrefixLength ipv6AddressinternalIpv6PrefixLengthr$CreateAliasIpRangeMessagesFromString aliasIpRangesNicTypeValueValuesEnumnicTypenetworkAttachment queueCountvlanIgmpQueryValueValuesEnum igmpQueryenableVpcScopedDns) resourcesrmessagesr% private_ip subnet_regionsubnetaddressalias_ip_ranges_string network_tier stack_typeipv6_network_tiernic_typeipv6_public_ptr_domain ipv6_addressipv6_prefix_lengthexternal_ipv6_addressexternal_ipv6_prefix_lengthinternal_ipv6_addressinternal_ipv6_prefix_lengthnetwork_attachmentnetwork_queue_countrA igmp_queryenable_vpc_scoped_dnsnetwork_interface subnet_ref network_ref access_configipv6_access_configs =lib/googlecloudsdk/api_lib/compute/instance_template_utils.pyCreateNetworkInterfaceMessagera$sh//1 002CC &&! D# $% &J $.#6#6#8  //:,,1199CCD%"'K!, 4 4 6~,4//!!:,,1199CCD%"'K!, 4 4 6",!!:::F {*))  1 1  " " 6 6 E E*GM ###m    : :< H(5o#"&<&H!..  6 6  " " 6 6 B B/D,>*>'"889JK"'-C*"(!("4&!#0088$$88DD1F.@,@)&;# ,!#0088$$88DD1F.@,@)2M/&$9! ,1L.AA e3 5#!!88B#*<'$#6  !!!:::F+@( cng}|r|D]}|jdd}d|vxs|jdd} |dk(s|| st}|jdd} |jdd} d} |rd|v} |jt||||jd d|jd d||jd d||jd d| f | |jd d|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jd|jdd|jdd|jdd| d|S)aCreate network interface messages. Args: resources: generates resource references, scope_lister: function, provides scopes for prompting subnet region, messages: creates resources. network_interface_arg: CLI argument specifying network interfaces. subnet_region: region of the subnetwork. support_enable_vpc_scoped_dns: Indicates whether setting enable vpc scoped dns on network interfaces is supported. Returns: list, items are NetworkInterfaceMessages. rJNz no-addressznetwork-attachmentz network-tierznic-typezenable-vpc-scoped-dnsr%zprivate-network-iprIaliasesz stack-typezipv6-network-tierzipv6-public-ptr-domainz ipv6-addresszipv6-prefix-lengthzexternal-ipv6-addresszexternal-ipv6-prefix-lengthzinternal-ipv6-addresszinternal-ipv6-prefix-lengthz queue-countrAz igmp-query)rOrMrNrPrQrRrSrTrUrVrWrXrArYrZ)getr.appendra) rErrFnetwork_interface_argrHsupport_enable_vpc_scoped_dnsresult interfacerJhas_no_addressrLrOrZs r`CreateNetworkInterfaceMessagesrms, &*  i.g$y0&IMM 5&n B7?N#]]>48lz40h" & 79 D mm 'mmIt,mm0$7mmHd+mmIt,$  "|T: ) .A4 H%.]]*D&%==>!*/CT!J$-MM)4%+4--/+%.MM)4%+4--/+"+/C!D"+-- t"D==."|T:$9G$ &!+n -rbc | r |jng} t|j|jxsg| } t |||t |dg||| | }|r^t |j|j|j|j|||||j|j|j g}nJ|r"|djr|jdg}n&| r"| djr| jdg}ng}tj |||j}tj"|||j}|| z|z|z|zS)z4Create disk messages for a single instance template.)container_mount_disk create_disk) support_kmssupport_multi_writersupport_replica_zonessupport_disk_labels) rF disk_typedisk_device_namedisk_auto_delete disk_size_gb image_urikms_argsrqdisk_provisioned_iopsdisk_provisioned_throughputdisk_interfacer)ro$CreatePersistentAttachedDiskMessagesrFdisk"CreatePersistentCreateDiskMessagesgetattr$CreateDefaultBootAttachedDiskMessageboot_disk_typeboot_disk_device_nameboot_disk_auto_deleteboot_disk_provisioned_iops boot_disk_provisioned_throughputboot_disk_interfacebootpop create_utilsCreateLocalNvdimmMessagesCreateLocalSsdMessages)argsclientrErryboot_disk_size_gbcreate_boot_diskrqrrmatch_container_mount_disksrsrtropersistent_diskspersistent_create_disksboot_disk_list local_nvdimms local_ssdss r`CreateDiskMessagesrDs~ $?dB+ // ))/r35 ?   dM2&/1- ,__))!77!77*#"&"A"A(,(M(M33 N#:1#=#B#B-11!45N,Q/44&**1-.NN88  oo- 22  oo*    rbc g}|D]X}|d}|jdd}|dk(r!|jjj}n |jjj}|jdd}|jdd} t j |||} |j| || |||jjj} |jdr\|jdd k(r!|jjj} n |jjj} | | _ |r| g|z}H|j| [|S) a;Returns a list of AttachedDisk messages and the boot disk's reference. Args: messages: GCE API messages, disks: disk objects - contains following properties * name - the name of disk, * mode - 'rw' (R/W), 'ro' (R/O) access mode, * boot - whether it is a boot disk, * auto-delete - whether disks is deleted when VM is deleted, * device-name - device name on VM. container_mount_disk: list of disks to be mounted to container, if any. Returns: list of API messages for attached disks rmoderwrF auto-delete) autoDeleter deviceNamersourcerrkSCSI)rf AttachedDiskModeValueValuesEnum READ_WRITE READ_ONLYr GetDiskDeviceNamer, PERSISTENTInterfaceValueValuesEnumrNVMErkrg) rFdisksrodisks_messagesrr mode_valuerr auto_delete device_name attached_diskrks r`r~r~sT&.d >HH]+&!!34 ?6KHH_-M$?? _i5N#P 1 1 34 "2kxx./H%-k"88N+L $k0  / / K K  66txx7IJk <<HHJ *//1- 1 U  + +5 + A1- ) * (,1I(JK%"hh~6K//.. $  __ ) ) = = H H"/K xx  + & ())BBGG ))BBGG 'k #}~5nK(M!P C- s(Pc d} |rtj||| d} |j|||} ||| _| | | _|j |d|| |j j j|j jj| } | rM| dk(r!|j jj}n |j jj}|| _ | S)z=Returns an AttachedDisk message for creating a new boot disk.NT)boot_disk_prefix)rrrrr)r MaybeGetKmsKeyrrrrrrr,rrrrrk)rFrurvrwrxryrzrqr{r|r}rinitialize_paramsboot_attached_diskrks r`rras(''(Ht=H;; y<J&(=% ,.I+,,! !(  4 4 ? ?  4 4 ? ? -''@@EEi''@@EEi#, rbcv|gS|d}t|jdd}|j||}|gS)aReturns a list of accelerator config messages for Instance Templates. Args: messages: tracked GCE API messages. accelerator: accelerator object with the following properties: * type: the accelerator's type. * count: the number of accelerators to attach. Optional, defaults to 1. Returns: a list of accelerator config messages that specify the type and number of accelerators to attach to an instance. rcount)acceleratorTypeacceleratorCount)intrfAcceleratorConfig)rF acceleratoraccelerator_typeaccelerator_countaccelerator_configs r`CreateAcceleratorConfigMessagesrsV I (+//'15611&9J2L  rb)NNNNNNNNNNNNNNNNN)F)NFFFFTF)N)FNFTF)NFNNN)!__doc__ __future__rrrgooglecloudsdk.api_lib.computerrrr r r /googlecloudsdk.api_lib.compute.instances.creater"googlecloudsdk.command_lib.computer r3googlecloudsdk.command_lib.compute.networks.subnetsr r$googlecloudsdk.command_lib.util.argsrgooglecloudsdk.corerobjectr.rarmrr~rrrrbr`rsA&'?46940QEU<*H   $ $ 3Kh#( Op %L`+/7~Qv $+\rb