(jdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZdd lmZGd d ej"ZGddej&ZdZdZdZdZdZdZdZdZdZdZdZd(dZ d)dZ!d)dZ"d)dZ#dZ$d Z% d(d!Z&d"Z'd#Z(d$Z)d%Z*d&Z+y')*zrrAzureNodePoolAttributeConfigrBHs  2 2 !? rc0tjddS)NclientzAzure client of the {resource}.r:r<r>rrAzureClientAttributeConfigrENs  2 2 @ rctjddtjtj j jgS)z.Gets Google Cloud location resource attribute.locationz)Google Cloud location for the {resource}.r5r; fallthroughs)rr=rPropertyFallthroughr rcontainer_azurerGr>rrLocationAttributeConfigrLTs@  2 2 ;  " ":#4#4#D#D#M#M N rcz|stjStjddtt gS)NrzGProject number of the Google Cloud Platform project for the {resource}.rH)r DEFAULT_PROJECT_ATTRIBUTE_CONFIGr=r*ruse_project_numbers rProjectAttributeConfigrQ_s?   4 44  2 2   ' ( + -  rc0tjddS)N operationzOperation for the {resource}.r:r<r>rrOperationAttributeConfigrTos  2 2 "A rc`tjddttjS)Nz gkemulticloud.projects.locationsrG) resource_name locationsId projectsId)r ResourceSpecrLrNr>rrGetLocationResourceSpecrZus*   ()+::  rcrtjddtttjS)Nr0rS)rV operationsIdrWrX)rrYrTrLrNr>rrGetOperationResourceSpecr]~s0   3+-)+::  rcrtjddtttjS)N.gkemulticloud.projects.locations.azureClustersr9)rVazureClustersIdrWrX)rrYr?rLrNr>rrGetAzureClusterResourceSpecras0   613)+::  rctjddttt tj S)N=gkemulticloud.projects.locations.azureClusters.azureNodePoolsrA)rVazureNodePoolsIdr`rWrX)rrYrBr?rLrNr>rrGetAzureNodePoolResourceSpecres6   E3513)+::  rc htjddttt |S)agGets the resource spec for an Azure client. Args: use_project_number: bool, whether to enforce using project number in the resource spec. Project number should always be used when a resource is referenced in a field e.g. the client field when creating/updating Azure clusters. Returns: The resource spec for an Azure client. -gkemulticloud.projects.locations.azureClientsrD)rVazureClientsIdrWrX)rrYrErLrQrOs rGetAzureClientResourceSpecris2   5/1)+'(:;  rc|rdnd}tjj|tdj |dj |y)zAdds a resource argument for an Azure cluster. Args: parser: The argparse.parser to add the resource arg to. verb: str, the verb to describe the resource, such as 'to update'. positional: bool, whether the argument is positional or not. r9z --clusterz cluster {}.TrequiredN)r ConceptParser ForResourceraformat AddToParserparserverb positionalr5s rAddAzureClusterResourceArgrusL!k$++ !#4  , Krc|rdnd}tjj|tdj |dj |y)aAdds a resource argument for an Azure node pool. Args: parser: The argparse.parser to add the resource arg to. verb: str, the verb to describe the resource, such as 'to update'. positional: bool, whether the argument is positional or not. node_poolz --node-poolz node pool {}.TrkN)r rmrnrerorprqs rAddAzureNodePoolResourceArgrxsL# $++ "$T" , Krc|rdnd}tjj|tdj |dj |y)zAdds a resource argument for an Azure client. Args: parser: The argparse.parser to add the resource arg to. verb: str, the verb to describe the resource, such as 'to update'. positional: bool, whether the argument is positional or not. rD--clientzAzure client {}.TrkN)r rmrnrirorprqs rAddAzureClientResourceArgr{sL Z$++  "% , Krctjjdtdj |dj |y)zAdds a resource argument for Google Cloud location. Args: parser: The argparse.parser to add the resource arg to. verb: str, the verb to describe the resource, such as 'to list'. z --locationzGoogle Cloud location {}.TrkN)r rmrnrZrorprrrss rAddLocationResourceArgr~sC++!((. , Krctjjdtdj |dj |y)zAdds a resource argument for operation in Azure. Args: parser: The argparse parser to add the resource arg to. verb: str, the verb to describe the resource, such as 'to update'. operation_idz operation {}.TrkN)r rmrnr]rorpr}s rAddOperationResourceArgrsC++ T" , Krc 2tjtjdt dj |rdnddtjdt dd j |rdnd d d d i|gddgi}|j|y)aAAdds resource arguments for both Azure cluster and client. This is used for the create and update command. Args: parser: The argparse.parser to add the resource arg to. auth_config_group: The Authentication Configuration argument group. update: bool, whether the resources are used in update command. r9zAzure cluster to {}.updatecreateTrkrzrOz#Azure client to use for cluster {}.creationFrG)rlflag_name_overridesgroupz--client.locationzcluster.location)command_level_fallthroughsN)r rmr ResourcePresentationSpecrarorirp)rrauth_config_groupr arg_parsers r$AddAzureClusterAndClientResourceArgsrs,,  5 5)+$++HHM   5 5(DA3::$(*#-r"2% $#68J7K!L'**  rctjj|jjj j dS)Nrgr1)r r3r4CONCEPTSrDParse RelativeNameargss rParseAzureClientResourceArgr'sB    - - mm  "//1@ . rctjj|jjj j dS)Nr_r1)r r3r4rr9rrrs rParseAzureClusterResourceArgr.sB    - - mm!!#002A . rctjj|jjj j dS)Nrcr1)r r3r4rrwrrrs rParseAzureNodePoolResourceArgr5sB    - - mm##%224 I . rctjj|jjj j dSr/)r r3r4rrrrrs rParseOperationResourceArgr>sB    - - mm  &&(557> . rcX||S|jr|Stj|S)N)isdigit project_utilGetProjectNumber)ps rrrEs+Y HYY[ H  & &q ))rN)F)T),r& __future__rrr googlecloudsdk.calliope.conceptsrr#googlecloudsdk.command_lib.projectsrr(googlecloudsdk.command_lib.util.conceptsr r googlecloudsdk.corer r rJrArgFallthroughr*r7r?rBrErLrQrTrZr]rarerirurxr{r~rrrrrrrr>rrrsC&'51DDG*) (@(@  4#6#6        *"""   ',!!H*r