dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd l m Z dd lmZdd lmZgd ZdgZddZddZddZdZdZddZdZddZ d ddZ d d!dZy)"z6Flags for Security Profile Threat Prevention commands.)absolute_import) annotations)division)unicode_literals)sp_api) arg_parsers)parser_arguments)concepts)deps)concept_parsers)presentation_specs)DEFAULT_ACTIONALLOWALERTDENYTHREAT_PREVENTIONc|jd|}|jdtjdd|jdtjdd |jd tjd d y )z5Adds --antivirus, --severities, or --threat-ids flag.T)mutexrequiredz --severitiesSEVERITY_LEVELzeList of comma-separated severities where each value in the list indicates the severity of the threat.)typemetavarhelpz --threat-idsz THREAT-IDzList of comma-separated threat identifiers where each identifier in the list is a vendor-specified Signature ID representing a threat type. z --antivirusPROTOCOLznList of comma-separated protocols where each value in the list indicates the protocol of the antivirus threat.N) add_group add_argumentrArgList)parserr severity_threatid_antivirus_argss ;lib/googlecloudsdk/command_lib/network_security/sp_flags.py#AddSeverityorThreatIDorAntivirusArgr!"s%+%5%5 8&6&"#//     3 0#//      0 #//     = 0NcB|xst}|jd||dy)Nz--actionz8Action associated with antivirus, severity, or threat-id)rchoicesr)DEFAULT_ACTIONSr)ractionsrr$s r AddActionArgr'Es+  &' E r"c,|jd|dy)Nz --descriptionz)Brief description of the security profile)rr)r)rrs r AddProfileDescriptionr)Os 6r"c bd}tjddtj|tjdddtjddd tjdd d  }t j ||d d}tj|gj|S)z-Adds Security Profile Threat Prevention type.security_profilez8networksecurity.organizations.locations.securityProfiles organizationz2Organization ID to which the changes should apply.organizationsId)parameter_namelocationz$location of the {resource} - Global. locationsIdzName of the {resource}.securityProfilesId)resource_collection resource_name api_versionr-r0r1TzSecurity Profile Name.)name concept_specr group_help) r ResourceSpecr GetApiVersion ResourceParameterAttributeConfigr ResourcePresentationSpecr ConceptParser AddToParser)r release_trackr5 resource_specpresentation_specs r AddSecurityProfileResourcerAWs $'' D&&&}5??  >* ;;  0& "BB  #-!-,)AA  )   & &(9': ; G G OOr"cfdS)NcHtj|jzSN)rGetEffectiveApiEndpointr5)xr>s r z MakeGetUriFunc..zs611-@166Ir")r>s`r MakeGetUriFuncrIys IIr"c}dk(rd}g}r7|jtjfddj|t j dd|S)z.Gets Google Cloud location resource attribute.-z a wildcardcSrDrHdefaultsr rGz)LocationAttributeConfig..sGr"z(Location of the resource. Defaults to {}r/zLocation of the {resource}.)r5 help_text fallthroughs)appendr Fallthroughformatr r:)rNdefault_keywordrPs` r LocationAttributeConfigrU}sg/ ^"O,    6 = =o N   2 2 - r"c0tjddS)z2Gets Google Cloud organization resource attribute.r,z"Organization ID of the {resource}.)r5rO)r r:rHr"r OrgAttributeConfigrWs  2 2 4 r"cXtjddt|tS)z?Constructs and returns the Resource specification for Location.z'networksecurity.organizations.locationsr/rM)r3r0r-)r r8rUrWrMs r GetLocationResourceSpecrYs)   /)':(*  r"c|tjjdt|||j |y)aAdds a resource argument for Google Cloud location. Args: parser: The argparse.parser to add the resource arg to. help_text: str, the text of the help message. required: bool, whether the argument is required. default: Optional default value for the arg. z --locationrM)r5r?r7rN)r r< ForResourcerYr=rrOrrNs r AddLocationResourceArgr]s:++ +G< , Kr"cV|jdtjd|||y)zJAdds the `mirroringDeploymentGroups` arg for CustomMirroring SPs (Broker).z--mirroring-deployment-groupsMIRRORING_DEPLOYMENT_GROUPS)rrrrrNN)rrrr\s r %AddCustomMirroringDeploymentGroupsArgr`s2 %    +  r")T)NT)FrD)FN)r$parser_arguments.ArgumentInterceptorrOstrrbool)zKList of comma-separated full names of mirroring-deployment-group resources.FN)rrarOrbrrcrNzlist[str] | None)__doc__ __future__rrrr9googlecloudsdk.api_lib.network_security.security_profilesrgooglecloudsdk.callioperr googlecloudsdk.calliope.conceptsr r (googlecloudsdk.command_lib.util.conceptsr r r%DEFAULT_PROFILE_TYPESr!r'r)rArIrUrWrYr]r`rHr"r rks=&"'L/451DG>,- FPDJ,   00c $  0 r"