S dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l mZdd l mZd Zd Zej$ej&j(Gddej*Zej$ej&j.ej&j0Gddej*Zy)zSet the IAM policy for a model.)absolute_import)division)unicode_literals)models)base)iam_util) endpoint_util)flags) models_util) region_utilctjdddj|tjdj|t j |y)NTzto set IAM policy for) positionalrequiredverb)include_global)r GetModelResourceArg AddToParser GetRegionArgrAddArgForPolicyFileparsers 0lib/surface/ai_platform/models/set_iam_policy.py_AddSetIamPolicyArgsrsI "$$/K$7D)55f= v&ctj|}tj|5t j t j|j|jcdddS#1swYyxYw)N)region) r GetRegionr MlEndpointOverridesr SetIamPolicyr ModelsClientmodel policy_file)argsrs r_Runr$&sT   &&((7  # #F$7$7$94::$($4$4 6877s =A33A<c&eZdZdZedZdZy)SetIamPolicyGASet the IAM policy for a model. Sets the IAM policy for the given model as defined in a JSON or YAML file. See https://cloud.google.com/iam/docs/managing-policies for details of the policy file format and contents. ## EXAMPLES The following command will read am IAM policy defined in a JSON file 'policy.json' and set it for the model `my_model`: $ {command} my_model policy.json ct|yNrrs rArgszSetIamPolicyGA.Args=  rct|Sr)r$selfr#s rRunzSetIamPolicyGA.RunA :rN__name__ __module__ __qualname____doc__ staticmethodr+r1rrr&r&-  !!rr&c&eZdZdZedZdZy)SetIamPolicyBetar'ct|yr)r*rs rr+zSetIamPolicyBeta.ArgsUr,rct|Sr)r.r/s rr1zSetIamPolicyBeta.RunYr2rNr3r9rrr<r<Er:rr<N)r7 __future__rrr googlecloudsdk.api_lib.ml_enginergooglecloudsdk.calliopergooglecloudsdk.command_lib.iamr$googlecloudsdk.command_lib.ml_enginer r r r rr$ ReleaseTracks ReleaseTrackGACommandr&ALPHABETAr<r9rrrJs&&'3(3>6<<'6D%%(()T\\*.D%%++T->->-C-CDt||Er