
    ,                     ^   S r SSKJr  SSKJr  SSKJr  SSKrSSKJr  SSKJr  SSK	J
r
  SS	KJr  SS
KJr  SrSr\
R                   R"                  S\
R                   R$                  S\
R                   R&                  S0rS r " S S\5      r " S S\5      r " S S\5      r " S S\5      rg)z'Utilities for Org Policy Simulator API.    )absolute_import)division)unicode_literalsN)apis)waiter)base)
properties)	resourcespolicysimulatori6 v1alphav1betav1c                 ,    [         R                  U 5      $ )z;Return the api version of the Org Policy Simulator service.)VERSION_MAPget)release_tracks    Elib/googlecloudsdk/api_lib/policy_intelligence/orgpolicy_simulator.pyGetApiVersionr   (   s    		''    c                   D  ^  \ rS rSrSrU 4S jrS rS rS r\	R                    SS j5       r\	R                       SS j5       r\	R                    SS	 j5       r\	R                    SS
 j5       r\	R                    SS j5       r\	R                  S 5       rSrU =r$ )OrgPolicySimulatorApi-   z'Base Class for OrgPolicy Simulator API.c                 D  > U[         R                  R                  :X  a  [        [        U ]  [        5      $ U[         R                  R                  :X  a  [        [        U ]  [        5      $ U[         R                  R                  :X  a  [        [        U ]  [        5      $ g N)r   ReleaseTrackALPHAsuperr   __new__OrgPolicySimulatorApiAlphaBETAOrgPolicySimulatorApiBetaGAOrgPolicySimulatorApiGA)clsr   	__class__s     r   r   OrgPolicySimulatorApi.__new__0   s    ))///(  :;<))...(  9:;)),,,(  789 -r   c                     [        U5      U l        [        R                  " [        U R                  5      U l        [        R                  " [        U R                  5      U l        g r   )r   api_versionr   GetClientInstance	_API_NAMEclientGetMessagesModulemessages)selfr   s     r   __init__OrgPolicySimulatorApi.__init__;   sC    $]3D((D4D4DEDK**9d6F6FGDMr   c                 *    UR                  S5      S   $ )N/)split)r.   operation_names     r   GetViolationsPreviewId,OrgPolicySimulatorApi.GetViolationsPreviewIdB   s    $R((r   c                    [         R                  " [        U R                  5      n[        R
                  R                  5       nUR                  SU R                  5        UR                  UR                  [        R                  R                  R                  R                  SU R                  UR                  5      S.SS9n[         R"                  " UR$                  5      n[         R&                  " XeU[(        S9$ )z#Wait for the operation to complete.r   global)organizationsIdlocationsIdorgPolicyViolationsPreviewsIdzNpolicysimulator.organizations.locations.orgPolicyViolationsPreviews.operations)params
collection)wait_ceiling_ms)r   r)   r*   r(   r
   REGISTRYCloneRegisterApiByNameParsenamer	   VALUESaccess_context_managerorganization	GetOrFailr6   r   CloudOperationPollerNoResources
operationsWaitFor_MAX_WAIT_TIME_MS)r.   	operationmessage	v1_clientregistryoperation_refpollers          r   WaitForOperation&OrgPolicySimulatorApi.WaitForOperationE   s     &&y$2B2BCI!!'')H0$2B2BCNN !!88EEOO#++INN;
 d # 	eM 33I4H4HIF>>w8IK Kr   c                     g r    r.   violations_previewparents      r   'CreateOrgPolicyViolationsPreviewRequest=OrgPolicySimulatorApi.CreateOrgPolicyViolationsPreviewRequestZ        	r   c                     g r   rV   r.   rD   overlayresource_countsstateviolations_counts         r   ,GetPolicysimulatorOrgPolicyViolationsPreviewBOrgPolicySimulatorApi.GetPolicysimulatorOrgPolicyViolationsPreview`   s     	r   c                     g r   rV   r.   custom_constraintspoliciess      r   GetOrgPolicyOverlay)OrgPolicySimulatorApi.GetOrgPolicyOverlayi   r\   r   c                     g r   rV   r.   policypolicy_parents      r   GetOrgPolicyPolicyOverlay/OrgPolicySimulatorApi.GetOrgPolicyPolicyOverlayo   s    
 	r   c                     g r   rV   r.   custom_constraintcustom_constraint_parents      r   #GetOrgPolicyCustomConstraintOverlay9OrgPolicySimulatorApi.GetOrgPolicyCustomConstraintOverlayv   r\   r   c                     g r   rV   r.   s    r   $GetOrgPolicyViolationsPreviewMessage:OrgPolicySimulatorApi.GetOrgPolicyViolationsPreviewMessage|   s    r   )r(   r+   r-   NNNNNNN)__name__
__module____qualname____firstlineno____doc__r   r/   r6   rS   abcabstractmethodrZ   rc   ri   ro   ru   ry   __static_attributes____classcell__)r%   s   @r   r   r   -   s    /	9H)K* AE59	 	
 8<;?CG9=DH	 	 -1#'	 	
  	 	 <@CG	 	
 	 	r   r   c                   j    \ rS rSrSr  SS jr     SS jr  SS jr  SS jr  SS jr	S	 r
S
rg)r      z-Base Class for OrgPolicy Simulator API Alpha.Nc                 6    U R                   R                  UUS9$ )N);googleCloudPolicysimulatorV1alphaOrgPolicyViolationsPreviewrY   r-   MPolicysimulatorOrganizationsLocationsOrgPolicyViolationsPreviewsCreateRequestrW   s      r   rZ   BOrgPolicySimulatorApiAlpha.CreateOrgPolicyViolationsPreviewRequest   s'     ==ffDV g  r   c                 <    U R                   R                  UUUUUS9$ N)rD   r_   resourceCountsra   violationsCountr-   ;GoogleCloudPolicysimulatorV1alphaOrgPolicyViolationsPreviewr^   s         r   rc   GOrgPolicySimulatorApiAlpha.GetPolicysimulatorOrgPolicyViolationsPreview   s0     ==TT&( U * *r   c                 6    U R                   R                  UUS9$ N)customConstraintsrh   )r-   1GoogleCloudPolicysimulatorV1alphaOrgPolicyOverlayrf   s      r   ri   .OrgPolicySimulatorApiAlpha.GetOrgPolicyOverlay   s'     ==JJ, K  r   c                 6    U R                   R                  UUS9$ N)rm   policyParent)r-   >GoogleCloudPolicysimulatorV1alphaOrgPolicyOverlayPolicyOverlayrl   s      r   ro   4OrgPolicySimulatorApiAlpha.GetOrgPolicyPolicyOverlay   s'     ==WW" X $ $r   c                 6    U R                   R                  UUS9$ N)customConstraintcustomConstraintParent)r-   HGoogleCloudPolicysimulatorV1alphaOrgPolicyOverlayCustomConstraintOverlayrr   s      r   ru   >OrgPolicySimulatorApiAlpha.GetOrgPolicyCustomConstraintOverlay   s'     ==aa*7 b 9 9r   c                 2    U R                   R                  nU$ r   r   r.   ms     r   ry   ?OrgPolicySimulatorApiAlpha.GetOrgPolicyViolationsPreviewMessage   s    QQAHr   rV   r{   r|   r}   r~   r   r   r   rZ   rc   ri   ro   ru   ry   r   rV   r   r   r   r      sW    5 BF59 9=;?CG9=DH* .2#' (,.2$ =ACG9r   r   c                   j    \ rS rSrSr  SS jr     SS jr  SS jr  SS jr  SS jr	S	 r
S
rg)r!      z,Base Class for OrgPolicy Simulator API Beta.Nc                 6    U R                   R                  UUS9$ )N):googleCloudPolicysimulatorV1betaOrgPolicyViolationsPreviewrY   r   rW   s      r   rZ   AOrgPolicySimulatorApiBeta.CreateOrgPolicyViolationsPreviewRequest   s'     ==ffCU g  r   c                 <    U R                   R                  UUUUUS9$ r   r-   :GoogleCloudPolicysimulatorV1betaOrgPolicyViolationsPreviewr^   s         r   rc   FOrgPolicySimulatorApiBeta.GetPolicysimulatorOrgPolicyViolationsPreview   s0     ==SS&( T * *r   c                 6    U R                   R                  UUS9$ r   )r-   0GoogleCloudPolicysimulatorV1betaOrgPolicyOverlayrf   s      r   ri   -OrgPolicySimulatorApiBeta.GetOrgPolicyOverlay   s'     ==II, J  r   c                 6    U R                   R                  UUS9$ r   )r-   =GoogleCloudPolicysimulatorV1betaOrgPolicyOverlayPolicyOverlayrl   s      r   ro   3OrgPolicySimulatorApiBeta.GetOrgPolicyPolicyOverlay   s'     ==VV" W $ $r   c                 6    U R                   R                  UUS9$ r   )r-   GGoogleCloudPolicysimulatorV1betaOrgPolicyOverlayCustomConstraintOverlayrr   s      r   ru   =OrgPolicySimulatorApiBeta.GetOrgPolicyCustomConstraintOverlay   s'     ==``*7 a 9 9r   c                 2    U R                   R                  nU$ r   r   r   s     r   ry   >OrgPolicySimulatorApiBeta.GetOrgPolicyViolationsPreviewMessage   s    PPAHr   rV   r{   r|   r   rV   r   r   r!   r!      sW    4 BF59 9=;?CG9=DH* .2#' (,.2$ =ACG9r   r!   c                   j    \ rS rSrSr  SS jr     SS jr  SS jr  SS jr  SS jr	S	 r
S
rg)r#      z*Base Class for OrgPolicy Simulator API GA.Nc                 6    U R                   R                  UUS9$ )N)6googleCloudPolicysimulatorV1OrgPolicyViolationsPreviewrY   r   rW   s      r   rZ   ?OrgPolicySimulatorApiGA.CreateOrgPolicyViolationsPreviewRequest   s'     ==ff?Q g  r   c                 <    U R                   R                  UUUUUS9$ r   r-   6GoogleCloudPolicysimulatorV1OrgPolicyViolationsPreviewr^   s         r   rc   DOrgPolicySimulatorApiGA.GetPolicysimulatorOrgPolicyViolationsPreview   s0     ==OO&( P * *r   c                 6    U R                   R                  UUS9$ r   )r-   ,GoogleCloudPolicysimulatorV1OrgPolicyOverlayrf   s      r   ri   +OrgPolicySimulatorApiGA.GetOrgPolicyOverlay   s'     ==EE, F  r   c                 6    U R                   R                  UUS9$ r   )r-   9GoogleCloudPolicysimulatorV1OrgPolicyOverlayPolicyOverlayrl   s      r   ro   1OrgPolicySimulatorApiGA.GetOrgPolicyPolicyOverlay   s'     ==RR" S $ $r   c                 6    U R                   R                  UUS9$ r   )r-   CGoogleCloudPolicysimulatorV1OrgPolicyOverlayCustomConstraintOverlayrr   s      r   ru   ;OrgPolicySimulatorApiGA.GetOrgPolicyCustomConstraintOverlay  s'     ==\\*7 ] 9 9r   c                 .    U R                   R                  $ r   r   rx   s    r   ry   <OrgPolicySimulatorApiGA.GetOrgPolicyViolationsPreviewMessage  s    ==OOOr   rV   r{   r|   r   rV   r   r   r#   r#      sX    2 BF59 9=;?CG9=DH* .2#' (,.2$ =ACG9Pr   r#   )r   
__future__r   r   r   r   googlecloudsdk.api_lib.utilr   r   googlecloudsdk.callioper   googlecloudsdk.corer	   r
   r*   rL   r   r   r    r"   r   r   objectr   r   r!   r#   rV   r   r   <module>r      s    . &  ' 
 , . ( * ) 	"  	YH$(
Q	F Q	h.!6 .b. 5 .`-P3 -Pr   