o B@s\dZddlmZddlmZddlmZddlmZddlmZddlmZ ddl Z d d Z d d Z d dZ ddZdZdZdZdZeeeZee eZ  d;ddZdd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Z d?d/d0Z!Gd1d2d2e"Z#Gd3d4d4e"Z$d5d6Z%dTsz$IsValidLabelValue..)lenall)valuer r rIsValidLabelValueFs rcCs|rt|ds dSt|S)zImplements the PCRE r'[\p{Ll}\p{Lo}][\p{Ll}\p{Lo}\p{N}_-]{0,62}'. The key must start with a lowercase character and must be a valid label value. Args: key: The label key, a string. Returns: True if the key is valid; False if not. rF)rr)keyr r rIsValidLabelKeyWs rzOnly hyphens (-), underscores (_), lowercase characters, and numbers are allowed. Keys must start with a lowercase character. International characters are allowed. Key length must not exceed 63 characters.zKeys must start with a lowercase character and contain only hyphens (`-`), underscores (```_```), lowercase characters, and numbers.zwOnly hyphens (-), underscores (_), lowercase characters, and numbers are allowed. International characters are allowed.zaValues must contain only hyphens (`-`), underscores (```_```), lowercase characters, and numbers.labelsTcC|rtnd}|r tnd}g}|r|t|r|tdg}|r)|d||r0||tjd|dt j ||dt j d|dS) z*Makes the base.Argument for --labels flag.Nz%List of label KEY=VALUE pairs to add. z--{} KEY=VALUEkey_type value_type metavartypeactionhelp KEY_FORMAT_VALIDATORVALUE_FORMAT_VALIDATORappendKEY_FORMAT_HELPVALUE_FORMAT_HELPjoinrArgumentformatrArgDict UpdateAction extra_message labels_name validate_keysvalidate_valuesr$r% format_help help_partsr r rGetCreateLabelsFlag}s*     r>cCstjd|ddj|ddS)Nz --clear-{} store_truea Remove all labels. If `--update-{labels}` is also specified then `--clear-{labels}` is applied first. For example, to remove all labels: $ {{command}} --clear-{labels} To remove all existing labels and create two new labels, ``foo'' and ``baz'': $ {{command}} --clear-{labels} --update-{labels} foo=bar,baz=qux r)r*r+)rr3r4)r9r r rGetClearLabelsFlags rAcCr ) z5Makes a base.Argument for the `--update-labels` flag.NzuList of label KEY=VALUE pairs to update. If a label exists, its value is modified. Otherwise, a new label is created.r!z --update-{}r"r#r&r'r,r7r r rGetUpdateLabelsFlags,     rBcCs,tjd|dttjdj|d|dS)Nz --remove-{}KEYz List of label keys to remove. If a label does not exist it is silently ignored. If `--update-{labels}` is also specified then `--update-{labels}` is applied first.r@r')rr3r4rArgListr6)r8r9r r rGetRemoveLabelsFlagsrEcCst|dS)zxAdds create command labels flags to an argparse parser. Args: parser: The argparse parser to add the flags to. N)r> AddToParser)parserr r rAddCreateLabelsFlagssrHcCsJt|||r|}t|t||dSt||dS)aAdds update command labels flags to an argparse parser. Args: parser: The argparse parser to add the flags to. extra_update_message: str, extra message to append to help text for --update-labels flag. extra_remove_message: str, extra message to append to help text for --delete-labels flag. enable_clear: bool, whether to include the --clear-labels flag. N)rBrFadd_mutually_exclusive_grouprArE)rGextra_update_messageextra_remove_message enable_clear remove_groupr r rAddUpdateLabelsFlagss   rNcCst|dr|jS|jS)zReturns the update labels dict from the parsed args. Args: args: The parsed args. Returns: The update labels dict from the parsed args. r)hasattrr update_labelsargsr r rGetUpdateLabelsDictFromArgss rScCs|jS)zReturns the remove labels list from the parsed args. Args: args: The parsed args. Returns: The remove labels list from the parsed args. ) remove_labelsrQr r rGetRemoveLabelsListFromArgss rUcCs"t|}|stdd|S)a{Validates and returns labels specific args for update. At least one of --update-labels, --clear-labels or --remove-labels must be provided. The --clear-labels flag *must* be a declared argument, whether it was specified on the command line or not. Args: parsed_args: The parsed args. Returns: (update_labels, remove_labels) update_labels contains values from --labels and --update-labels flags respectively. remove_labels contains values from --remove-labels flag Raises: RequiredArgumentException: if all labels arguments are absent. AttributeError: if the --clear-labels flag is absent. LABELSzVAt least one of --update-labels, --remove-labels, or --clear-labels must be specified.)DiffFromUpdateArgsMayHaveUpdatescalliope_exceptionsRequiredArgumentException) parsed_argsdiffr r rGetAndValidateOpsFromArgss r^cs"fddtt|DdS)Ncsg|] \}}j||dqS)rr)AdditionalProperty)rrr labels_clsr r (s z"_PackageLabels..additionalProperties)sortedsix iteritems)rbrr rar_PackageLabels&s   ricCs|siSdd|jDS)NcSsi|]}|j|jqSr r_)rlr r r 0sz*_GetExistingLabelsDict..rdr@r r r_GetExistingLabelsDict-srl=c Cs|sdSi}|D]O}z ||\}}Wnty#tdd|w|r5t|s5tddj|td|rFt|sFtddj|td||vrStdd||||<q|S) aValidates and returns labels in dictionary format. Args: list_of_labels: List of labels in format ["K1=V1", "K2=V2", ...]. delimiter: delimiters which separates key and its corresponding values. validate_keys: if true, performs regex validation. validate_values: if true, performs regex validation. Returns: None: if list_of_labels is empty. Otheriwse: dictionary of labels {"K1": "V1", "K2": "V2", ...}. Raises: InvalidArgumentException: if invalid format. Nz--labelszInvalid label format: {}z-Invalid key format: {key} {_KEY_FORMAT_ERROR})r_KEY_FORMAT_ERRORz3Invalid value format: {value} {_VALUE_FORMAT_ERROR})r_VALUE_FORMAT_ERRORzDuplicate key: {}) split ValueErrorrZInvalidArgumentExceptionr4rrnrro)list_of_labels delimiterr:r;dict_of_labelslabelrrr r rValidateAndParseLabels3s>      rwc@s,eZdZdZddZeddZddZdS) UpdateResulta-Result type for Diff application. Attributes: needs_update: bool, whether the diff resulted in any changes to the existing labels proto. _labels: LabelsValue, the new populated LabelsValue object. If needs_update is False, this is identical to the original LabelValue object. cCs||_||_dSr) needs_update_labels)selfryrr r r__init__os zUpdateResult.__init__cCs|jstd|jS)zUReturns the new labels. Raises: ValueError: if needs_update is False. zTIf no update is needed (self.needs_update == False), checking labels is unnecessary.)ryrqrzr{r r rrss zUpdateResult.labelscCsz|jWStyYdSw)aReturns the new labels if an update is needed or None otherwise. NOTE: If this function returns None, make sure not to include the labels field in the field mask of the update command. Otherwise, it's possible to inadvertently clear the labels on the resource. N)rrqr}r r r GetOrNones  zUpdateResult.GetOrNoneN)__name__ __module__ __qualname____doc__r|propertyrr~r r r rrxes    rxc@sReZdZdZdddZddZdd Zd d Zdd d ZddZ e dddZ dS)rWz%A change to the labels on a resource.NFcCs6|pi|_|pg|_||_|jr|jrtddSdS)aeInitialize a Diff. Only one of [subtractions, clear] may be specified. Args: additions: {str: str}, any label values to be updated subtractions: List[str], any labels to be removed clear: bool, whether to clear the labels Returns: Diff. Raises: ValueError: if both subtractions and clear are specified. z3Only one of [subtractions, clear] may be specified.N) _additions _subtractions_clearrq)r{ additions subtractionsclearr r rr|s   z Diff.__init__cCs&~|}|jD]}||dq|SzRemove labels.N)copyrpopr{existing_labels new_labelsrr r r _RemoveLabelss  zDiff._RemoveLabelscCs~iSrr r{rr r r _ClearLabelsszDiff._ClearLabelscCs|}||j|Sr)rupdater)r{rr r r _AddLabelss zDiff._AddLabelscCsZt|}|}|jr||}|jr||}|jr!|||}||k}t|t ||S)aApply this Diff to the (possibly non-existing) labels. First, makes any additions. Then, removes any labels. Args: labels_cls: type, the LabelsValue class for the resource. labels: LabelsValue, the existing LabelsValue object for the original resource (or None, which is treated the same as empty labels) Returns: labels_cls, the instantiated LabelsValue message with the new set up labels, or None if there are no changes. ) rlrrrrrrrrxri)r{rbrrrryr r rApplys   z Diff.ApplycCst|j|j|jgS)z'Returns true if this Diff is non-empty.)anyrrrr}r r rrYszDiff.MayHaveUpdatesTcCs |r|j}nd}||j|j|S)zBInitializes a Diff based on the arguments in AddUpdateLabelsFlags.N) clear_labelsrPrT)clsrRrLrr r rrXszDiff.FromUpdateArgs)NNFr)T) rrrrr|rrrrrY classmethodrXr r r rrWs  rWcCs(t|}|r |nd}|||S)aReturns the result of applying the diff constructed from args. Lazily fetches the original labels value if needed. Args: args: argparse.Namespace, the parsed arguments with update_labels, remove_labels, and clear_labels labels_cls: type, the LabelsValue class for the new labels. orig_labels_thunk: callable, a thunk which will return the original labels object (of type LabelsValue) when evaluated. Returns: UpdateResult: the result of applying the diff. N)rWrXrYr)rRrborig_labels_thunkr] orig_labelsr r rProcessUpdateArgsLazys  rcCs t||}|dur dSt||S)z5Initializes labels based on args and the given class.N)getattrri)rRrb labels_destrr r rParseCreateArgss  rc@s eZdZdZddZddZdS)ExplicitNullificationDiffzA change to labels for resources where API requires explicit nullification. That is, to clear a label {'foo': 'bar'}, you must pass {'foo': None} to the API. cCs8|}|jD]}||vrd||<q||vr||=q|Sr)rrrr r rrs  z'ExplicitNullificationDiff._RemoveLabelscCsdd|DS)NcSsi|]}|dqSrr )rrr r rrksz:ExplicitNullificationDiff._ClearLabels..r rr r rrsz&ExplicitNullificationDiff._ClearLabelsN)rrrrrrr r r rrs r)rrTTr@)rTT)rrT)NrmTT)(r __future__rrrgooglecloudsdk.callioperrrrZrgrrrrrnr0ror1CustomFunctionValidatorr-r.r>rArBrErHrNrSrUr^rirlrwobjectrxrWrrrr r r rsd "             2(V