
                         P    d Z ddlmZ ddlmZ ddlmZ ddlmZ d Zd Zd Z	d	 Z
y
)z7Common constants and methods for Org Policies commands.    )absolute_import)division)unicode_literals)org_policiesc                     t        j                         }| j                  r|j                  S | j                  r|j
                  S | j                  r|j                  S y N)r   OrgPoliciesClientprojectprojectsorganizationorganizationsfolderfolders)argsclients     Dlib/googlecloudsdk/command_lib/resource_manager/org_policies_base.pyOrgPoliciesServicer      sJ    ))+&	\\??{{>>    c                 Z   t        j                         }|j                  t        j                  | j                              }t        |       }| j                  r|j                  ||      S | j                  r|j                  ||      S | j                  r|j                  ||      S y)zConstructs a resource-dependent GetOrgPolicyRequest.

  Args:
    args: Command line arguments.

  Returns:
    Resource-dependent GetOrgPolicyRequest.
  )
constraint)
projectsIdgetOrgPolicyRequest)organizationsIdr   )	foldersIdr   N)r   OrgPoliciesMessagesGetOrgPolicyRequestFormatConstraintidGetResourcer
   /CloudresourcemanagerProjectsGetOrgPolicyRequestr   4CloudresourcemanagerOrganizationsGetOrgPolicyRequestr   .CloudresourcemanagerFoldersGetOrgPolicyRequest)r   messagesrequestresource_ids       r   r   r   $   s     --/(((..tww7 ) 9'D!+	\\CCG D = =HH# I B B{{BB7 C < <	r   c                     t        j                         }t        |       }|j                  |      }| j                  r|j                  ||      S | j                  r|j                  ||      S | j                  r|j                  ||      S y)zConstructs a resource-dependent SetOrgPolicyRequest.

  Args:
    args: Command line arguments.
    policy: OrgPolicy for resource-dependent SetOrgPolicyRequest.

  Returns:
    Resource-dependent SetOrgPolicyRequest.
  )policy)r   setOrgPolicyRequest)r   r(   )r   r(   N)
r   r   r   SetOrgPolicyRequestr
   /CloudresourcemanagerProjectsSetOrgPolicyRequestr   4CloudresourcemanagerOrganizationsSetOrgPolicyRequestr   .CloudresourcemanagerFoldersSetOrgPolicyRequest)r   r'   r#   r%   r$   s        r   r)   r)   =   s     --/(D!+(((7'	\\CCG D = =HH# I B B{{BB7 C < <	r   c                     | j                   r| j                   S | j                  r| j                  S | j                  r| j                  S y r   )r
   r   r   )r   s    r   r   r   W   s<    	\\<<{{;;r   N)__doc__
__future__r   r   r   'googlecloudsdk.api_lib.resource_managerr   r   r   r)   r    r   r   <module>r2      s*    > &  ' @	24r   