dZddlmZddlmZddlmZddlZddlmZddl mZddl m Z ddl m Z dd l mZddlmZdd lmZdd lmZd Zd ZGddej,ZddZdZdZdZdZdZy)z%CLI Utilities for cloud tpu commands.)absolute_import)division)unicode_literalsN) exceptions)peering)apis)util) properties) resourcesz;Error determining VPC peering status for network [{}]: [{}]z7Network [{}] is invalid for use with Service NetworkingceZdZdZy)ServiceNetworkingExceptionzEException for creation failures involving Service Networking/Peering.N)__name__ __module__ __qualname____doc__3lib/googlecloudsdk/command_lib/compute/tpus/util.pyr r &sMrr c.tjd|S)Ntpu)rGetMessagesModule)versions rrr*s  w //rcH~~tj}||_|S)z/Declarative request hook for TPU Start command.)rStartNodeRequeststartNodeRequest)refargsrequest start_requests rStartRequestHookr .s&  #%668-*' .rcH~~tj}||_|S)z.Declarative request hook for TPU Stop command.)rStopNodeRequeststopNodeRequest)rrr stop_requests rStopRequestHookr%7s&  "$446,(' .rc tjj}|j|d}|j}t j|S#tj $r|}Y,wxYw)z!@K$,,  / /0B CC &%&s=AA,+A,cP~|j}|r|jxs4tjjjj d} t |j|}tj|dtjj|j}|s(t#t(j'|j|S#tjtj f$r/}t#t$j'|j||d}~wwxYw)aValidates that supplied network has been peered to a GoogleOrganization. Uses the Service Networking API to check if the network specified via --network flag has been peered to Google Organization. If it has, proceeds with TPU create operation otherwise will raise ServiceNetworking exception. Check is only valid if --use-service-networking has been specified otherwise check will return immediately. Args: ref: Reference to the TPU Node resource to be created. args: Argument namespace. request: TPU Create requests message. Returns: request: Passes requests through if args pass validation Raises: ServiceNetworkingException: if network is not properly peered T)requiredz servicenetworking.googleapis.comN)use_service_networkingr+r VALUEScoreGetr4r/rListConnectionsospathbasenamer(ListConnectionsPermissionDeniedExceptionapitools_exceptions HttpErrorr _PROJECT_LOOKUP_ERRORformat_PEERING_VALIDATION_ERROR)rrrservice_networking_enabledr+network_project_number lookup_resultes rCreateValidateVPCHookrINs( #::llOj//44<<@@$@OG B=dll>E G-- "D ''  4<< (*m  & # * *4<< 8 :: .  ? ?  ) ) +B &  & &t||Wa @ BBBsAC#D%6*D  D%c~g}|D];}|jD]*}||j|j|jdgz },=|jd|S)z;Reformat to extract topologies and sort by acceleratorType.)topologytypeacceleratorTypecBt|djddS)NrM-)intsplit)xs rz,ListTopologiesResponseHook..sc!$5"6"<"resr,&' >63,M=*)27N!6!6N0 D(V r