A2dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddlm Zdd lmZdd lmZd Zd Zd ZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(d Z)d!ee*d"ee jVfd#Z,d$Z-y%)&z0Hooks for Backup for GKE command line arguments.)absolute_import)division)unicode_literals)Optional)messages)util) exceptions)log) console_io%cluster_resource_selected_group_kinds%cluster_resource_excluded_group_kinds cluster_resource_all_group_kindscluster_resource_no_group_kindsc~~d|_|S)NT)force)refargsrequests @lib/googlecloudsdk/command_lib/container/backup_restore/hooks.pyAddForceToDeleteRequestr$s  '- .c|sytj}g} |D]}|jd}t|dk(rd}|d}n*t|dk(r|\}}nt j |d|st j |d|j }||_||_|j||S#t$rt j |dwxYw) zProcess list of group kinds.N/rz7Cluster resource scope selected group kinds is invalid.z5Cluster resource scope selected group kinds is empty.) api_utilGetMessagesModulesplitlenr InvalidArgumentException GroupKind resourceGroup resourceKindappend ValueError) group_kindsflagmessagegksresource group_kindgroupkindgks rParseGroupKindsr0.s    & & (' #C>>#&j ZA !} z?a  t11  E  11  CE E    bbbo jjn' ( J C  - - A CCCs B B<< Cctj}|j}|jj t |d|S)Nz---cluster-resource-scope-selected-group-kinds)rrClusterResourceRestoreScopeselectedGroupKindsextendr0r'r)crrss rProcessSelectedGroupKindsr7PG  & & ('  , , .$   F +rctj}|j}|jj t |d|S)Nz---cluster-resource-scope-excluded-group-kinds)rrr2excludedGroupKindsr4r0r5s rProcessExcludedGroupKindsr;[r8rc\tj}|j}||_|SN)rrr2 allGroupKinds)all_group_kindsr)r6s rProcessAllGroupKindsr@fs+  & & ('  , , .$&$ +rc\tj}|j}||_|Sr=)rrr2 noGroupKinds)no_group_kindsr)r6s rProcessNoGroupKindsrDms+  & & ('  , , .$$$ +rc6|stjdd|S)Nz--all-namespacesz All namespaces can only be true.r r!)all_namespacess rProcessAllNamespacesrHts&   - -* ,, rc6|stjdd|S)Nz--no-namespaceszNo namespaces can only be true.rF) no_namespacess rProcessNoNamespacesrK|s&   - -) ++ rc6|stjdd|S)Nz--selected-namespacesz&Selected namespaces must not be empty.rF)selected_namespacess rProcessSelectedNamespacesrN&   - -0 22 rc6|stjdd|S)Nz--excluded-namespacesz&Excluded namespaces must not be empty.rF)excluded_namespacess rProcessExcludedNamespacesrRrOrcF|stjddtj}|j } |j dD]}|j d\}}|s%tjddj ||s%tjddj ||j}||_||_ |jj||S#t$r&tjddj |wxYw)z%Processes selected-applications flag.z--selected-applicationsz(Selected applications must not be empty.,rz/Namespace of selected application {0} is empty.z*Name of selected application {0} is empty.z%Selected applications {0} is invalid.) r r!rrNamespacedNamesrformatNamespacedNamename namespacenamespacedNamesr%r&)selected_applicationsr)sanamespaced_namerYrXnns rProcessSelectedApplicationsr_s-   - -!2 44  & & (' "066s;'--c2oi 11 % = D D !" "11 % 8 ? ? !" "  ! ! #bbgbl#< I   - -!/667LM s B2C11/D c|stjddtj}|j }|j dD]}}|j d}|dstjdd|j }|d|_t|dk(rdn|d|_ |jj||S) z)Processes selected-namespace-labels flag.z--selected-namespace-labelsz6Input for selected-namespace-labels must not be empty.rT=rz'Key of namespace label cannot be empty.rr) r r!rrResourceLabelsrLabelkeyr valueresourceLabelsr%)selected_namespace_labelsr)rlskey_value_pairpartsrls rProcessSelectedNamespaceLabelsrls "  - -%@   & & (' #177jAdD]v}|jCdrd}nO|jCdrd}n;|jCdrd}n'|jCdrd}n|jCdrd}|j;|xdjE||_|S)z-Preprocess request for updating restore plan.rGNrJrMrQr[substitution_rules_filez The given restore plan already has the transformation rules. Updating the restore plan with new substitution rules will delete the existing transformation rules. T) cancel_on_nornumberrepeatedz!restoreConfig.transformationRulestransformation_rules_filez The given restore plan already has the substitution rules. Updating the restore plan with new transformation rules will delete the existing substitution rules. zrestoreConfig.substitutionRulesrTz restoreConfig.selectedNamespacesz restoreConfig.excludedNamespacesz"restoreConfig.selectedApplicationszrestoreConfig.noNamespaceszrestoreConfig.allNamespaces)#hasattr%CLUSTER_RESOURCE_SELECTED_GROUP_KINDSrpr7r restorePlan restoreConfigclusterResourceRestoreScope%CLUSTER_RESOURCE_EXCLUDED_GROUP_KINDSr;r CLUSTER_RESOURCE_ALL_GROUP_KINDSr@rCLUSTER_RESOURCE_NO_GROUP_KINDSrDr noNamespacesruexcludedNamespacesrsrtbooltransformationRulesr PromptContinuer FieldList StringFieldr%substitutionRulesrzrr{r}rs rPreprocessUpdateRestorePlanrs   1 >?!$"L"LM %%A  1 >?!$"L"LM %%A T349I9I&: TBBC %%A T238H8H%9 D@@A %%A  &'59G%%2;?G%%8;?G%%8=AG%%: o&6:G%%3;?G%%8;?G%%8=AG%%: +,6:G%%359G%%2;?G%%8=AG%%: +,6:G%%359G%%2;?G%%8=AG%%: -.6:G%%359G%%2;?G%%8;?G%%8) 01 w""00DD E  =E 1d 5 6 )d 6 7 *d T,xx *' .rc b|s+tjtjddgStjdt j |d}tj}tj|j|tjddd  }|jS) zBReads content of the substitution rule file specified in file_arg.rTrzyThe substitutionRules field is deprecated and can only be managed via gcloud/API. Please migrate to transformation rules.Fbinary gkebackupv1SubstitutionRules message_typestream schema_path)rrrr warningr ReadFromFileOrStdinrr export_utilImport RestoreConfig GetSchemaPathrfile_argdatamstemp_restore_configs rReadSubstitutionRuleFilerms    h22!dKR PP++=  ' ' ?$!!#"#**## ++ t0  . ..rc |sytj|d}tj}t j |j |t jddd}|jS)zDReads content of the transformation rule file specified in file_arg.NFrrrTransformationRulesr) r rrrrrrrrrs rReadTransformationRuleFilersi    ' ' ?$!!#"#**## ++ t2  0 00rc |sytj|d}tj}t j |j |t jddd}|S)z>Reads content of the restore order file specified in file_arg.NFrrr RestoreOrderr)r rrrrrrr)rrrtemp_restore_orders rReadRestoreOrderFilers`    ' ' ?$!!#""))?? ++ t^ rc |sytj|d}tj}t j |j |t jddd}|jS)zAReads content of the exclusion window file specified in file_arg.NFrrrExclusionWindowsr) r rrrrr RpoConfigrexclusionWindows)rrrtemp_rpo_configs rReadExclusionWindowsFilersf    ' ' ?$!!#"&&<< ++ t//  ) ))rrreturnc |sytj|d}tj}t j |j |t jdddjS)z4Reads the volume data restore policy overrides file.NFrrr VolumeDataRestorePolicyOverridesr) r rrrrrRestorer volumeDataRestorePolicyOverrides)rrrs r(ReadVolumeDataRestorePolicyOverridesFilerse    ' ' ?$!!#"   :: ++ t?  %$ %rc :|sytj|d} tjt j j |tjddd}|S#t$r*}tjdd j|d}~wwxYw) z?Reads content of the restore filter file specified in file_arg.NFrrrFilterrz --filter-filez{0}) r rrrrrrr Exceptionr r!rV)rrrestore_filteres rReadRestoreFilterFilers    ' ' ?$  ''//188-- x N    - - Q sA A'' B0%BBN).__doc__ __future__rrrtypingrapitools.base.protorpcliter/googlecloudsdk.api_lib.container.backup_restorerrgooglecloudsdk.callioper !googlecloudsdk.command_lib.exportrgooglecloudsdk.corer googlecloudsdk.core.consoler rrrrrr0r7r;r@rDrHrKrNrRr_rlrrrrrrstrrrrrrrs7&'/L.A#2(O%(O%#E "CCDB 0-`m`/( 1  * %sm% h778%"r