: dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l mZdd l mZdd l mZdd l mZddlmZddlmZddlZdZdZdZdddddddZGddej4ZGddej4ZGddej4ZdZd Zd!Z d"Z!d#Z"d$Z#d%Z$d&Z%d'Z&d(Z'd)Z(d*Z)d+Z*d,Z+d-Z,d.Z-d/Z.d0Z/d1Z0d2Z1d3Z2d4Z3d5Z4d6Z5d7Z6d8Z7d9Z8d:Z9d;Z:d<Z;d=Zz.Argument processors for DLP surface arguments.)absolute_import)division)unicode_literalsN)apis) arg_parsers)base) arg_utils) exceptions)log) properties) resources)files)timesdlpv2zColors should be specified as a string of `r,g,b` float values in the interval [0,1] representing the amount of red, green, and blue in the color, respectively. For example, `black = 0,0,0`, `red = 1.0,0,0`, `white = 1.0,1.0,1.0`, and so on. IMAGE IMAGE_PNG IMAGE_JPEG IMAGE_SVG IMAGE_BMP)n_az.pngz.jpegz.jpgz.svgz.bmpceZdZdZy)ImageFileErrorz:Error if an image file is improperly formatted or missing.N__name__ __module__ __qualname____doc__+lib/googlecloudsdk/command_lib/dlp/hooks.pyrr7sBr rceZdZdZy)RedactColorErrorz;Error if a redact color is improperly formatted or missing.Nrrr r!r#r#;sCr r#ceZdZdZy)BigQueryTableNameErrorz7Error if a BigQuery table name is improperly formatted.Nrrr r!r%r%?s?r r%cVtjtt}t ||S)z3Get API message object for given message type name.)rGetMessagesModule_DLP_API_DLP_API_VERSIONgetattr) msg_type_namemsgs r!_GetMessageClassr-Ds" x)9:# m $$r c |sy|xr|tvS)z#Validate image file name extension.T)VALID_IMAGE_EXTENSIONS) extensions r!_ValidateExtensionr1Js    < %;;..`.r;r<r%r= table_name name_partss r!_ValidateAndParseInputTableNamerJjsD$*_  99? 9K MM r c|jd}t|dk7r(t|dk7rtdj||S)rDrEr:zInvalid BigQuery output table name [{}]. BigQuery tables are uniquely identified by their project_id, dataset_id, and or table_id in the format `..` or `..rFrGs r! _ValidateAndParseOutputTableNamerMvsQ$*_c*o2  &'-fZ&8  :: r c*td}||S)z=Return GooglePrivacyDlpV2InfoType message for a parsed value.GooglePrivacyDlpV2InfoTypenamer-)r?infotypes r!InfoTyperTs : ;( u r c*td}||S)z-Create a GooglePrivacyDlpV2FieldId for value.GooglePrivacyDlpV2FieldIdrPrR)r?field_ids r! PrivacyFieldrXs 9 :( u r cxt|\}}}td}td}||||}||}|S)aConvert BigQuery table name into GooglePrivacyDlpV2BigQueryOptions. Creates BigQuery input options for a job trigger. Args: table_name: str, BigQuery table name to create input options from in the form `..`. Returns: GooglePrivacyDlpV2BigQueryOptions, input options for job trigger. Raises: BigQueryTableNameError if table_name is improperly formatted. !GooglePrivacyDlpV2BigQueryOptionsGooglePrivacyDlpV2BigQueryTable datasetId projectIdtableIdtableReference)rJr-)rH project_id data_set_idtable_idbig_query_optionsbig_query_tabletableoptionss r!BigQueryInputOptionsrisQ'F'#*k8&'JK$%FG/ z8 E% U 3' .r cNtd}td}|||S)z-Return CloudStorageOptions for given GCS url.%GooglePrivacyDlpV2CloudStorageOptionsGooglePrivacyDlpV2FileSet)url)fileSetrR)rmcloud_storage_optionsfile_sets r!GcsInputOptionsrqs+*-/ 9 :( xC'8 99r cTtd}td}td}tjjjj d}|j d}t|dk(r|\}}||}||| } n||}|| } ||| S) amConvert Datastore arg value into GooglePrivacyDlpV2DatastoreOptions. Creates Datastore input options for a job trigger from datastore table name. Args: table_name: str, Datastore table name to create options from in the form `namespace:example-kind` or simply `example-kind`. Returns: GooglePrivacyDlpV2Action, output action for job trigger. "GooglePrivacyDlpV2DatastoreOptions GooglePrivacyDlpV2KindExpressionGooglePrivacyDlpV2PartitionIdTrequired:rLrP) namespaceIdr^)r^)kind partitionId)r-r VALUEScoreprojectGetr;r<) rHdata_store_optionsrz partition_idr~ split_name namespacergkind_exp partitions r!DatastoreInputOptionsrs((LM < =$!"AB,    " " * * . . . ='$*_!IuHgFI$Hw/I y AAr cNtd}td}|||S)z1Return PubSub DlpV2Action for given PubSub topic.GooglePrivacyDlpV2Action!GooglePrivacyDlpV2PublishToPubSub)topic)pubSubrR)r action_msg pubsub_actions r!PubSubTopicActionrs( :;*"#FG- =u5 66r ct|}|d}|d}d}t|dk(r|d}td}td}td}td }|||| } ||||   S)aConvert BigQuery formatted table name into GooglePrivacyDlpV2Action. Creates a BigQuery output action for a job trigger. Args: table_name: str, BigQuery table name to create action from in the form `..` or `.`. Returns: GooglePrivacyDlpV2Action, output action for job trigger. Raises: BigQueryTableNameError if table_name is improperly formatted. rr:rLrGooglePrivacyDlpV2SaveFindings%GooglePrivacyDlpV2OutputStorageConfigr[r\)rg) outputConfig) saveFindings)rMr<r-) rHrIrbrcrdrsave_findings_config output_configrfrgs r!BigQueryTableActionrs0 ;*!}*1 + (_!}H :;*)*JK"#JK-$%FG/ z8 E% '$513 44r cLtj|tjS)N)tzinfo)rFormatDateTimeUTCr?s r! DlpTimeStamprs   eEII 66r c~|jxs4tjjjj d}t j j|d}|j|_ |S)z%Set parent value for a DlpXXXRequest.Trvz dlp.projects) collection) r~r r|r}rr REGISTRYParse RelativeNameparent)refargsrequestr project_refs r!SetRequestParentrsc << L:,,1199==t=L&""((N(K+++-'. .r c8~~td}||_|S)z6Set cancel message on DlpProjectsDlpJobsCancelRequest.%GooglePrivacyDlpV2CancelDlpJobRequest)r-%googlePrivacyDlpV2CancelDlpJobRequest)rrrcancel_requests r!SetCancelRequestHookrs#  #$KL.2@2B'/ .r c~|jjjjj}|j r|r|j |j _|S)z1Update partitionId.projectId on DatastoreOptions.))googlePrivacyDlpV2CreateJobTriggerRequest jobTrigger inspectJob storageConfigdatastoreOptionsr~r{r^)rrrrs r!UpdateDataStoreOptionsr sN  77BBMM}%% \\(/3||"", .r c~|jjjj}|r=|jr1t d}|jDcgc] }|| c}|_|Scc}w)z>LL422 ;}|jdr|jddz}n|dz }|j|@dj ||_|S)zX NE ,  88>t  FFr cLtd}t|\}}}||||S)aConvert color_string into GooglePrivacyDlpV2Color. Creates a GooglePrivacyDlpV2Color message from input string to use for image redaction. Args: color_string: str, string representing red, green and blue color saturation percentages as float values between 0.0 and 1.0. For example, `black = 0,0,0`, `red = 1.0,0,0`, `white = 1.0,1.0,1.0` etc. Returns: GooglePrivacyDlpV2Color, color message. Raises: RedactColorError if color_string is improperly formatted. GooglePrivacyDlpV2Color)redbluegreen)r-rB) color_string color_msgrrrs r!GetRedactColorFromStringrqs/"89),\:#ud sU 33r cJdjtj|S)z;Return API required format for duration specified by value.z{}s)r=six text_typers r!GetJobScheduleDurationStringrs cmmE* ++r c^d}tjddtj|gS)z.Capture identifying fields for BigQuery table.zComma separated list of references to field names uniquely identifying rows within the BigQuery table. Nested fields in the format `person.birthdate.year` are allowed.z--identifying-fieldsIDENTIFYING_FIELDS)metavarrhelp)rArgumentrArgList) help_texts r!GetIdentifyingFieldsArgrs7A) mm &""$  r c|jsytj|j5}|j|dddtj j dj|jy#1swYBxYw)z@Helper function for writing redacted contents to an output file.Nz+The redacted contents can be viewed in [{}]) output_filerBinaryFileWriterwriter statusPrintr=)r? parsed_argsoutfiles r!*_PossiblyWriteRedactedResponseToOutputFilersg    k556' MM%7**@GG 76s BB cFt|jj||S)zHWrite the contents of the redacted text file to parsed_args.output_file.)ritemr?responsers r!-PossiblyWriteRedactedTextResponseToOutputFiler s,X]]-@-@+N /r c2t|j||S)z4Write the redacted image to parsed_args.output_file.)r redactedImager s r!.PossiblyWriteRedactedImageResponseToOutputFilers,X-C-C-8: /r c2tjddgS)z&Add --output-file to a redact command.z --output-filez/Path to the file to write redacted contents to.)r)rrrr r!AddOutputFileFlagrs! mm @B r )=r __future__rrrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioperr$googlecloudsdk.command_lib.util.apisr googlecloudsdk.corer r r r googlecloudsdk.core.utilrrrr(r)r>r/Errorrr#r%r-r1r7rBrJrMrTrXrirqrrrrrrrrrrrrrrrrrrr rrrr r!rsA5&' ,/(:*#*)**        CZ%%CDz''D@Z--@ % =  I    2:B674D7    " B  >4,,   r