dZddlmZddlmZddlmZddlZddlmZddl m Z ddl m Z dd lmZddlmZdd lmZdd lmZd d dZe j(e j*j,Gdde j.Zee_y)z%Reset command for the Org Policy CLI.)absolute_import)division)unicode_literalsN) exceptions)service)base) arguments)utils)logzA Resets the policy to the default for the constraint. z To reset the policy associated with the constraint 'gcp.resourceLocations' and the Project 'foo-project', run: $ {command} gcp.resourceLocations --project=foo-project ) DESCRIPTIONEXAMPLEScJeZdZdZedZdZdZdZdZ dZ dZ d Z y ) Resetz3Reset the policy to the default for the constraint.ctj|tj|tj|y)N)r AddConstraintArgToParserAddResourceFlagsToParserAddUpdateMaskArgToParser)parsers !lib/surface/org_policies/reset.pyArgsz Reset.Args2s, &&v. &&v. &&v.c&|duxs |dk(xs|dk(S)N*z policy.specself update_masks rShouldUpdateLiveSpeczReset.ShouldUpdateLiveSpec8s)t ( #  ( - 'rc|dk(xs|dk(S)Nrzpolicy.dry_run_specrrs rShouldUpdateDryRunSpeczReset.ShouldUpdateDryRunSpec?s #  E0E!EErc2tj|}|!|jrtjd|j |r^|j s|jj|_d|j _ g|j _ d|j _ |j|r^|js|jj|_d|j_ g|j_ d|j_ |S)aSets the reset field on the policy to True. If reset is set to True, no rules can be set on the policy and inheritFromParent has to be False. As such, this also deletes all rules on the policy and sets inheritFromParent to False. Args: policy: messages.GoogleCloudOrgpolicyV2alpha1Policy, The policy to be updated. update_mask: Specifies whether live/dryrun spec needs to be reset. Returns: The updated policy. z@update_mask is required if there is dry_run_spec in the request.TF) copydeepcopy dryRunSpecrInvalidInputErrorrspecorg_policy_apiCreateEmptyPolicySpecresetrulesinheritFromParentr )rpolicyr new_policys r ResetPolicyzReset.ResetPolicyBsv&Jz44  ( ( L    - __--CCE "joo joo*/joo' "";/  " " $ 3 3 I I K $(j!$&j!05j- rcl |jj|S#tj$rYywxYw)zGet the policy from the service. Args: policy_name: Name of the policy to be retrieved. Returns: The retrieved policy, or None if not found. N)r' GetPolicyapi_exceptionsHttpNotFoundError)r policy_names rr0zReset.GetPolicyes6  * *; 77  + + s 33c|jj|}|j||}|jj|}t j |d|S)zCreate the policy on the service if needed. Args: policy_name: Name of the policy to be created update_mask: Specifies whether live/dryrun spec needs to be created. Returns: The created policy. r,)r'BuildEmptyPolicyr. CreatePolicyr CreatedResource)rr3r empty_policyr-create_responses rr6zReset.CreatePolicyssX&&77 DL!!, >rN) __name__ __module__ __qualname____doc__ staticmethodrrr r.r0r6r;rFrrrrr.s<;// F!F $,.?rr)rJ __future__rrrr"apitools.base.pyrr1 googlecloudsdk.api_lib.orgpolicyrr@googlecloudsdk.callioper'googlecloudsdk.command_lib.org_policiesr r googlecloudsdk.corer DETAILED_HELP ReleaseTracksrBGACommandr detailed_helprrrrWs,&' 9J(=>9#   D%%(()Z?DLLZ?*Z?z$r