dZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZd d d Zej(ej*j,Gddej.Zee_y)z5Set-custom-constraint command for the Org Policy CLI.)absolute_import)division)unicode_literals) exceptions) completers)service)base)utils)logz Sets a Custom Constraint from a JSON or YAML file. The custom constraint will be created if it does not exist, or updated if it already exists. z| To set the custom constraint from the file on the path './sample_path', run: $ {command} ./sample_path ) DESCRIPTIONEXAMPLESc&eZdZdZedZdZy)SetCustomConstraintz1Set a custom constraint from a JSON or YAML file.cJ|jddtjdy)Ncustom_constraint_fileCUSTOM_CONSTRAINT_FILEz@Path to JSON or YAML file that contains the organization policy.)metavar completerhelp) add_argumentrFilesCompleter)parsers 1lib/surface/org_policies/set_custom_constraint.pyArgszSetCustomConstraint.Args1s(  (++ O Qctj|j}tj|j |j}|j stjd|j jds%tjd|j zdz |j|j }||k(r|S|j!|}tj"|j d|S#tj$r6|j|}tj|j d|cYSwxYw)aCreates or updates a custom constraint from a JSON or YAML file. This first converts the contents of the specified file into a custom constraint object. It then fetches the current custom constraint using GetCustomConstraint. If it does not exist, the custom constraint is created using CreateCustomConstraint. If it does, the retrieved custom constraint is checked to see if it needs to be updated. If so, the custom constraint is updated using UpdateCustomConstraint. Args: args: argparse.Namespace, An object that contains the values for the arguments specified in the Args method. Returns: The created or updated custom constraint. z0Name field not present in the custom constraint.zorganizations/z+Name field contains invalid resource type: zC. Custom constraints can be created only on organization resources.zcustom constraint)org_policy_service OrgPolicyApi ReleaseTrackr "GetCustomConstraintMessageFromFilernamerInvalidInputError startswithGetCustomConstraintapi_exceptionsHttpNotFoundErrorCreateCustomConstraintr CreatedResourceUpdateCustomConstraintUpdatedResource)selfargsorg_policy_apiinput_custom_constraintcustom_constraintcreate_responseupdate_responses rRunzSetCustomConstraint.Run9sN"(44T5F5F5HIN#FF ##T%6%6%8: " ' '  ( ( < >> " ' ' 2 23C D  ( ( 7 ! & & ' O P QQ(<< ! & &(33 $;;!O/446IJ   + +&== !#o 1668KL  s4D AEEN)__name__ __module__ __qualname____doc__ staticmethodrr2rrrr-s9QQ)rrN)r6 __future__rrrapitools.base.pyrr% argcompleter googlecloudsdk.api_lib.orgpolicyrrgooglecloudsdk.callioper 'googlecloudsdk.command_lib.org_policiesr googlecloudsdk.corer DETAILED_HELP ReleaseTracksrGACommandr detailed_helpr8rrrEs}<&'9"J(>9#    D%%(()4$,,4*4n%2!r