*dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m cm cm cmcmZddlm cm cm cmcmZdZd Zd ZGd d e j,ZGd deZdZdZdZdZdZdZdZdZ dZ!dZ"dedejFdefdZ$GddejJZ&dZ'd Z(d!Z)d"Z*y)#z,Utilities for `gcloud network-connectivity`.)absolute_import)division)unicode_literalsN)Any) exceptionsz projects/z#location:projects/{0}/locations/{1}z-type:DYNAMIC_ROUTEceZdZdZy)NetworkConnectivityErrorz8Top-level exception for all Network Connectivity errors.N__name__ __module__ __qualname____doc__;lib/googlecloudsdk/command_lib/network_connectivity/util.pyr r #s@rr ceZdZdZy)InvalidInputErrorzException for invalid input.Nr rrrrr's$rra table( name.basename(), name.segment(3):label=LOCATION, hub.basename(), group.basename(), format( "{0}{1}{2}{3}{4}", linkedVpnTunnels.yesno(yes="VPN tunnel", no=""), linkedInterconnectAttachments.yesno(yes="VLAN attachment", no=""), linkedRouterApplianceInstances.yesno(yes="Router appliance", no=""), linkedVpcNetwork.yesno(yes="VPC network", no=""), gateway.yesno(yes="Gateway", no="") ):label=TYPE, firstof(linkedVpnTunnels.uris, linkedInterconnectAttachments.uris, linkedRouterApplianceInstances.instances).len().yesno(no="1"):label="RESOURCE COUNT", format( "{0}{1}", linkedVpcNetwork.yesno(yes="N/A", no=""), firstof(linkedVpnTunnels.siteToSiteDataTransfer, linkedInterconnectAttachments.siteToSiteDataTransfer, linkedRouterApplianceInstances.siteToSiteDataTransfer, Gateway).yesno(yes="On", no="") ).yesno(no="Off"):label="DATA TRANSFER", description ) an table( name.basename(), group.basename(), name.segment(1):label=PROJECT, name.segment(3):label=LOCATION, spokeType:label=TYPE, state, reasons.code.list():label="STATE REASON", etag, format( "{0}{1}", linkedVpcNetwork.yesno(yes="N/A", no=""), firstof(linkedVpnTunnels.siteToSiteDataTransfer, linkedInterconnectAttachments.siteToSiteDataTransfer, linkedRouterApplianceInstances.siteToSiteDataTransfer, gateway).yesno(yes="On", no="") ).yesno(no="Off").if(view=detailed):label="DATA TRANSFER", description.if(view=detailed) ) c0|xjdz c_|S)z$Add locations/global to parent path.z/locations/global)parent unused_ref unused_argsrequests rAppendLocationsGlobalToParentrXs ..''. .rcz|jds)|js |jrd|j_|S)z3Set legacy export_psc field based on new PSC flags. export_pscT) IsSpecified6export_psc_published_services_and_regional_google_apisexport_psc_global_google_apis'googleCloudNetworkconnectivityV1betaHub exportPscrargsrs rSetExportPscBetar$_s7  , ' CC  - -BFg55? .rct|vrtdj|||jtt tzd}|j dd}|S)z+Returns the project from a resource string.z7Resource must contain a project path, but received: {0}N/r)PROJECTS_RESOURCE_PATHrformatindexlensplit)resourceprojects rDeriveProjectFromResourcer.ksi8+ AHH    nn+,s3I/JJL ' MM# q !' .rc|jdrR|j}t|j}tj ||}dj |t |_|S)zGAppend filter to limit listing dynamic routes at an effective location.effective_locationz {0} OR {1})rr0r.rLOCATION_FILTER_FMTr(ROUTE_TYPE_FILTERfilter)rr#rlocationr-location_filters rAppendEffectiveLocationFilterr6zsX *+&&H'7G)00(CO!((:KLGN .rcy)zSet default location to global.globalrrrrSetGlobalLocationr9s rc||jdr*|jr|xjdz c_|Sd|_|S)zHandles clear_overlaps flag.clear_overlapsz ,overlapsoverlapsr updateMaskrr# patch_requests r ClearOverlapsrAsE &'+- ",m rc||jdr*|jr|xjdz c_|Sd|_|S)zHandles clear_labels flag. clear_labelsz,labelslabelsr=r?s r ClearLabelsrEsD n%)+ "*m rc|jdr|jdk7r/|jds|jdr td|S|jdr|jds td|jdr|jdk7r td |js td |j s td |j|j k(r td |S) z.Validates internal range migration parameters.usagez for-migrationmigration_sourcemigration_targetz_migration_source and migration_target can only be specified when usage is set to for-migration.z=Both migration_source and migration_target must be specified.peeringzfor-selfzCPeering must be set to for-self when usage is set to for-migration.z!migration_source cannot be empty.z!migration_target cannot be empty.z9migration_source and migration_target cannot be the same.)rrGrrJrHrIr"s rValidateMigrationrKs  ' "djjO&C *+t/?/?0  ,  n  , -T5E5E6 G  i T\\Z%? M     ? @@   ? @@ d333 C  .rrefr#rc~|jdr<|jdk(r-|jdr|jdkr td|S|jdr td|S)z,Validates internal range allocation options.allocation_strategyRANDOM_FIRST_N_AVAILABLE"first_available_ranges_lookup_sizezwfirst_available_ranges_lookup_size must be set and greater than 0 when allocation_strategy is RANDOM_FIRST_N_AVAILABLE.zhfirst_available_ranges_lookup_size can only be set when allocation_strategy is RANDOM_FIRST_N_AVAILABLE.)rrNrPr)rLr#rs rValidateAllocationOptionsrRs  ,-  " "&@ @   A B  2 2Q 6  C  .  <=  <  .rc&eZdZdZ dfd ZxZS)StoreGlobalActionz1Return "global" if the --global argument is used.c8tt| ||d|||y)Nr8)option_stringsdestconstdefaultrequiredhelp)superrT__init__)selfrVrWrYrZr[ __class__s rr]zStoreGlobalAction.__init__s, T+%   ,r)FN)r r r rr] __classcell__)r_s@rrTrTs:DH  rrTc|jr3tjjj|j _|S)zSet the route's `recipient` field based on boolean flags. Args: args: The command arguments. request: The request to set the `recipient` field on. Returns: The request with the `recipient` field set. )advertise_to_hubv1beta:GoogleCloudNetworkconnectivityV1betaGatewayAdvertisedRouteRecipientValueValuesEnumADVERTISE_TO_HUB:googleCloudNetworkconnectivityV1betaGatewayAdvertisedRoute recipientr"s r"SetGatewayAdvertisedRouteRecipientrjs7 IIbbss FFP .rcvd|jv}|jjd}|r |s td|S)aIf a spoke name is specified, then its region must also be specified. This is because CCFE doesn't support a wildcard ("-") in this case but returns a confusing error message. So we give the user a friendlier error. Args: request: The request object. We will inspect the parent field. Returns: The unmodified request object. Raises: InvalidInputError: If the region is unspecified when a spoke is. z /locations/-/z /spokes/-z7A region must be specified if a spoke name is specifiedrendswithr)rrrregion_wildcardspoke_wildcards r$CheckRegionSpecifiedIfSpokeSpecifiedrpsB$w~~5/>>**;7.^ A  .rcd|jv}|jjd}|r |s td|r |s td|S)aCheck that hub and route table are both specified or both unspecified. This is because CCFE doesn't support wildcards ("-") in this case but returns a confusing error message. So we give he user a friendlier error. Args: request: The request object. Returns: The unmodified request object. Raises: InvalidInputError: If the user needs to specify a hub name or route table name. z/hubs/-/z/routeTables/-z5A hub must be specified if a route table is specifiedz5A route table must be specified if a hub is specifiedrl)rrr hub_wildcardroute_table_wildcards r(CheckForRouteTableAndHubWildcardMismatchrts\w~~-, 001AB. ? , ?  .rct|jds|S|jjtjj j k7r|Std)aReject requests with HYBRID_INSPECTION preset topology. Args: request: A CreateHubRequest object. Returns: The unmodified request object. Raises: InvalidInputError: If the CreateHubRequest has the HYBRID_INSPECTION preset topology. presetTopologyzHYBRID_INSPECTION unsupported in the GA component; use the beta component instead. See https://cloud.google.com/sdk/gcloud#release_levels)hasattrhubrvv1HubPresetTopologyValueValuesEnumHYBRID_INSPECTIONrrs rProhibitHybridInspectionr}0sT . / N kk  ff**<<= N? r)+r __future__rrrargparsetypingrgooglecloudsdk.corer\googlecloudsdk.generated_clients.apis.networkconnectivity.v1.networkconnectivity_v1_messagesgenerated_clientsapisnetworkconnectivityrynetworkconnectivity_v1_messagesdgooglecloudsdk.generated_clients.apis.networkconnectivity.v1beta.networkconnectivity_v1beta_messagesrd#networkconnectivity_v1beta_messagesr'r1r2Errorr r LIST_FORMATLIST_SPOKES_FORMATrr$r.r6r9rArErK NamespacerR_StoreConstActionrTrjrprtr}rrrrs3&'*iiiuuu%;)Az//A%0%  0(   D3h.@.@3022$".8r