dZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z ddl m Z dd lmZd Zd Z dd Zd ZdZdZy)zFCode that's shared between multiple org firewall policies subcommands.)absolute_import)division)unicode_literalsN)network_security) exceptions)reference_utils)logzorganizations/c|j|d}d}|djD]}|j|k(s|j}n|8t j dj |tj|S)aKReturns the firewall policy id that matches the short_name in the org. Args: firewall_policy: the organization firewall policy. short_name: the short name of the firewall policy to be resolved. organization_id: the organization ID which the firewall policy belongs to. Returns: Firewall policy resource ID. F) parent_idonly_generate_requestNrzHInvalid short name: {0}. No Security Policy with this short name exists.) Listitems displayNamenamer errorformatsysexit)firewall_policy short_nameorganization_idresponsefp_idfps Slib/googlecloudsdk/command_lib/compute/firewall_policies/firewall_policies_utils.pyResolveFirewallPolicyIdr!s ! !u">( % QK  b ~~#gge   ]IIR  HHJ ,ctjd|s@|*tjd|t j t ||t|zS|S)aaReturns the firewall policy id that matches the short_name in the org. Args: firewall_policy_client: the organization firewall policy client. firewall_policy: the short name or ID of the firewall policy to be resolved. organization: the organization ID which the firewall policy belongs to. Returns: Firewall policy resource ID. \d{9,15}zAMust set --organization=ORGANIZATION when short name [%s]is used.)rematchr rrrrORGANIZATION_PREFIX)firewall_policy_clientr organizations rGetFirewallPolicyIdr%<sY + / ii %' hhj "#9?#6#E GG rcFtjd|s'|%tjddj ||}|sG|j |}|s$t jdj ||dj}d|vr|jdd}|S) aQReturns ID of the organization the given firewall policy belongs to. Args: firewall_policy_client: the organization firewall policy client. firewall_policy_id: the short name or ID of the firewall policy to be resolved. optional_organization: organization if provided. Returns: Firewall policy resource ID. rz--organizationzBMust set --organization=ORGANIZATION when short name [{0}]is used.)rz%Firewall Policy [{0}] does not exist.r/) r r!rRequiredArgumentExceptionrDescribecompute_exceptionsInvalidResourceErrorparentsplit)r#firewall_policy_idoptional_organizationr$fetched_policiess rGetFirewallPolicyOrganizationr2Vs +$ &*?*G  . . 6,- //', -66=O6P   3 3 1 8 89K L NN#A&--LL%%c*1-L rcd|vr|St|||}tjtjtj j j|d|S)aReturns Full URL reference of Security Profile Group. Args: security_profile_group: reference string provided by the user. optional_organization: organization if provided. firewall_policy_client: the organization firewall policy client. firewall_policy_id: the short name or ID of the firewall policy to be resolved. Returns: URL to Security Profile Group. r'r#r/r0securityProfileGroupsbase_uriorg_idcollection_name resource_name)r2r'BuildFullResourceUrlForOrgBasedResourcer GetApiBaseUrlbase ReleaseTrackGA)security_profile_groupr0r#r/r$s rBuildSecurityProfileGroupUrlrAusi  "" !!.3+13,  @ @--    , , / /-*  rc\d|vr|St|||}tjd|d|S)awReturns partial URL reference of Address Group. Args: address_group: reference string provided by the user. optional_organization: organization if provided. firewall_policy_client: the organization firewall policy client. firewall_policy_id: the short name or ID of the firewall policy to be resolved. Returns: partial URL to Address Group. r'r4 addressGroupsr6)r2rr;) address_groupr0r#r/r$s rBuildAddressGroupUrlrFsF  M .3+13,  @ @ %!  ##r)N)__doc__ __future__rrrr rgooglecloudsdk.api_librgooglecloudsdk.callioper"googlecloudsdk.command_lib.computer+rgooglecloudsdk.corer r"rr%r2rArFrrrNsNM&' 3.O>#&:&*4>B#r