AdZddlmZddlmZddlmZddlZddlmZGddejZ Gd d e Z d Z d Z d Z dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZy)z*Utilities for `gcloud network-management`.)absolute_import)division)unicode_literalsN) exceptionsceZdZdZy)NetworkManagementErrorz6Top-level exception for all Network Management errors.N__name__ __module__ __qualname____doc__9lib/googlecloudsdk/command_lib/network_management/util.pyrrs>rrceZdZdZy)InvalidInputErrorzException for invalid input.Nr rrrrrs$rrcdg}t|dkDr|jdt|D]\}}|jdj|||dk(r t|dk(r|jdM|t|dz k(r|jd p|t|dz ks|jd |jd d j |S) zOCreates a message to specify at least one endpoint, separated by commas and or.zInvalid Connectivity Test. zAt least one of z--{endpoint_type}-{endpoint}) endpoint_typeendpointrz or z, or z, z must be specified.)lenappend enumerateformatjoin) endpointsr error_msgindexrs r"GetClearSingleEndpointAttrErrorMsgr!!s,-)^a '("9-oeX &--'( .   zc)n)v #i.1$ $w Y!# #t. ()  rc0|xjdz c_|S)zXAdd locations/global to parent path, since it isn't automatically populated by apitools./locations/global)parent unused_ref unused_argsrequests rAppendLocationsGlobalToParentr)8s ..''. .rc0|xjdz c_|S)Nr#)namer%s r"UpdateOperationRequestNameVariabler,>s ,,%%, .rcV|j}|s ||_|S||vr |dz|z|_|S)z"Adds name of field to update mask.,) updateMask)field patch_request update_masks rAddFieldToUpdateMaskr3CsB((+ $M  K*S058M rcJhd}||vrt||dyt||dy)N> cloudFunctionappEngineVersioncloudRunRevisionr)setattr)r endpoint_nameproto_endpoint_fieldss rClearEndpointValuer;Ms/ ++ HmT* HmR(rc|j}t||}hd}d}|D]}t||ds|dz }|dkDs t||ds2t||t|||||_t |dz|z|Sgd}|dk(r|j gd|d k(r|j gd t t|| zNChecks if given endpoint can be removed from Connectivity Test and removes it.> instance ipAddress redisClusterr5 redisInstanceforwardingRuler6r7cloudSqlInstancegkeMasterClusterrNr.)r>z ip-addresszgke-master-clusterzcloud-sql-instancesource)zcloud-functionzapp-engine-versionzcloud-run-revision destination)zforwarding-rulezredis-instancez redis-clusterconnectivityTestgetattrr;r8r3extendrr! r1rr9testrendpoint_fieldsnon_empty_endpoint_fieldsr0rs rClearSingleEndpointAttrrP[  ' '$ T= )( / ex%1$"' t+x/ D-*%)M" m+] I    % *9mD rcbgd}|D]%\}}}|j|st|||}'|Sz5Handles clear_source_* and clear_destination_* flags.))clear_source_instancerFr>)clear_source_ip_addressrFr?)clear_source_gke_master_clusterrFrD)clear_source_cloud_sql_instancerFrC)clear_source_cloud_functionrFr5)clear_source_app_engine_versionrFr6)clear_source_cloud_run_revisionrFr7)clear_destination_instancerGr>)clear_destination_ip_addressrGr?)$clear_destination_gke_master_clusterrGrD)$clear_destination_cloud_sql_instancerGrC)!clear_destination_forwarding_rulerGrB) clear_destination_redis_instancerGrA)clear_destination_redis_clusterrGr@) IsSpecifiedrPr&argsr1flags_and_endpointsflagrr9s rClearEndpointAttrsrgsH2-@(dM= -   m-@ rc|j}t||}hd}d}|D]}t||ds|dz }|dkDs t||ds2t||t|||||_t |dz|z|Sgd}|dk(r|j gd|d k(r|j gd t t||r=rHrLs rClearSingleEndpointAttrBetarirQrcbgd}|D]%\}}}|j|st|||}'|SrS)rbrircs rClearEndpointAttrsBetarksH2-@(dM= 1   m-@ rcddg}tjd}|D]L}|j|st||}|j |r3t dj |||S)z5Checks if all provided instances are in valid format.source_instancedestination_instancezEprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/zones/[-\w]+/instances/[-\w]+zInvalid value for flag {}: {} Expected instance in the following format: projects/my-project/zones/zone/instances/my-instancerecompilerbrJmatchrrr&rdr(flagsinstance_patternrfr>s rValidateInstanceNamesrvs %ZZNd t$h  # #H - EEKVhF   .rcddg}tjd}|D]L}|j|st||}|j |r3t dj |||S)z4Checks if all provided networks are in valid format.source_networkdestination_networkz>projects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/global/networks/[-\w]+zxInvalid value for flag {}: {} Expected network in the following format: projects/my-project/global/networks/my-networkro)r&rdr(rtnetwork_patternrfnetworks rValidateNetworkURIsr|)s %JJG/d d#g  " "7 + ??Evg@   .rcddg}tjd}|D]L}|j|st||}|j |r3t dj |||S)zFChecks if all provided GKE Master Clusters URIs are in correct format.source_gke_master_clusterdestination_gke_master_clusterzPprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/(zones|locations)/[-\w]+/clusters/[-\w]+zInvalid value for flag {}: {} Expected Google Kubernetes Engine master cluster in the following format: projects/my-project/location/location/clusters/my-clusterro)r&rdr(rtrurfclusters rValidateGKEMasterClustersURIsrAs"& %ZZYd d#g  # #G , IIOgJ   .rcddg}tjd}|D]L}|j|st||}|j |r3t dj |||S)zFChecks if all provided Cloud SQL Instances URIs are in correct format.source_cloud_sql_instancedestination_cloud_sql_instancez8projects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/instances/[-\w]+z~Invalid value for flag {}: {} Expected Cloud SQL instance in the following format: projects/my-project/instances/my-instancerorss rValidateCloudSQLInstancesURIsrXs~"& %ZZAd t$h  # #H - ::@&x:P   .rcdg}tjd}|D]L}|j|st||}|j |r3t dj |||S)zBChecks if all provided Cloud Functions URIs are in correct format.source_cloud_functionzIprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/locations/[-\w]+/functions/[-\w]+zInvalid value for flag {}: {} Expected Cloud Function in the following format: projects/my-project/locations/location/functions/my-functionro)r&rdr(rtfunction_patternrffunctions rValidateCloudFunctionsURIsrms %ZZRd   D !tT"H  ! !( +  K 6$ !    .rcdg}tjd}|D]L}|j|st||}|j |r3t dj |||S)zEChecks if all provided App Engine version URIs are in correct format.source_app_engine_versionzCapps/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/services/[-\w]+/versions/[-\w]+zInvalid value for flag {}: {} Expected App Engine version in the following format: apps/my-project/services/my-service/versions/my-versionro)r&rdr(rtversion_patternrfversions rValidateAppEngineVersionsURIsrs" %JJL/d   D !dD!G   )  FFLfGG    .rcdg}tjd}|D]L}|j|st||}|j |r3t dj |||S)zFChecks if all provided Cloud Run revisions URIs are in correct format.source_cloud_run_revisionzIprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/locations/[-\w]+/revisions/[-\w]+zInvalid value for flag {}: {} Expected Cloud Run revision in the following format: projects/my-project/locations/location/revisions/my-revisionro)r&rdr(rtrevision_patternrfrevisions rValidateCloudRunRevisionsURIsrs" %ZZRd   D !tT"H  ! !( +  K 6$ !    .rcdg}tjd}|D]<}|j|st||}|j |r3t d|S)zCChecks if all provided forwarding rules URIs are in correct format.destination_forwarding_rulezVprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/(global|regions/[-\w]+)/forwardingRules/[-\w]+zInvalid value for flag {flag}: {forwarding_rule} Expected forwarding rule in one of the following formats: projects/my-project/global/forwardingRules/my-forwarding-rule projects/my-project/regions/us-central1/forwardingRules/my-forwarding-rulerprqrbrJrrr)r&rdr(rtforwarding_rule_patternrfforwarding_rules rValidateForwardingRulesURIsrss$ %JJ_d   D !dD)O " ( ( 9  [   .rcdg}tjd}|D]<}|j|st||}|j |r3t d|S)zAChecks if all provided Redis Instance URIs are in correct format.destination_redis_instancezIprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/locations/[-\w]+/instances/[-\w]+zInvalid value for flag {flag}: {redis_instance} Expected Redis Instance in the following format: projects/my-project/locations/location/instances/my-redis-instance r)r&rdr(rtredis_instance_patternrfredis_instances rValidateRedisInstancesURIsrsm ' (%::Rd   D !T4(N ! ' ' 7  U   .rcdg}tjd}|D]<}|j|st||}|j |r3t d|S)z@Checks if all provided Redis Cluster URIs are in correct format.destination_redis_clusterzHprojects/(?:[a-z][a-z0-9-\.:]*[a-z0-9])/locations/[-\w]+/clusters/[-\w]+zInvalid value for flag {flag}: {redis_cluster} Expected Redis Cluster in the following format: projects/my-project/locations/location/clusters/my-redis-cluster r)r&rdr(rtredis_cluster_patternrf redis_clusters rValidateRedisClustersURIsrsm & '%**Qd   D !D$'M & &} 5  S   .rc`|jdr|jds td|S)zZChecks if destination-gke-master-cluster was specified when destination-fqdn is specified.destination_fqdnrzXdestination-fqdn can only be specified when destination-gke-master-cluster is specified )rbr)r&rdr(s r ValidateFqdnrs< ()$2B2B&3  9  .r)r __future__rrrrpgooglecloudsdk.corerErrorrrr!r)r,r3r;rPrgrirkrvr|rrrrrrrrrrrrrs1&' *?Z--?%.%.   )2j#L2j#L00.*.0..(( r