
    
                     6    S r SSKJr  SSKJr   " S S\5      rg)z7Utilities for Audit Manager API, Enrollments Endpoints.    )	constants)utilc                   d    \ rS rSrSr SS\R                  SS4S jjrS\S\S	\	S
\	4S jr
S rSrg)EnrollmentsClient   z+Client for operations in Audit Manager API.Napi_versionreturnc                     U=(       d    [         R                  " US9U l        U=(       d    [         R                  " XS9U l        g )N)r   )r   client)r   GetClientInstancer   GetMessagesModulemessages)selfr   r   r   s       7lib/googlecloudsdk/api_lib/audit_manager/enrollments.py__init__EnrollmentsClient.__init__   s7     KD22{KDK  6 6!DM    scopeeligible_gcs_bucketsis_parent_folderis_parent_organizationc                    U(       a  U R                   R                  nO4U(       a  U R                   R                  nOU R                   R                  nU R                  R                  5       n[        [        U R                  U5      5      Ul	        U(       a  U R                  R                  5       nO<U(       a  U R                  R                  5       nOU R                  R                  5       nXl        Xgl        UR                  U5      $ )a  Enrolls a resource to Audit Manager.

Args:
  scope: The scope to be enrolled.
  eligible_gcs_buckets: List of destination among which customer can choose
    to upload their reports during the audit process.
  is_parent_folder: Whether the parent is folder and not project.
  is_parent_organization: Whether the parent is organization and not
    project.

Returns:
  Described audit operation resource.
)r   folders_locationsorganizations_locationsprojects_locationsr   EnrollResourceRequestlistmapGcs_uri_to_eligible_destinationdestinations1AuditmanagerFoldersLocationsEnrollResourceRequest7AuditmanagerOrganizationsLocationsEnrollResourceRequest2AuditmanagerProjectsLocationsEnrollResourceRequestr   enrollResourceRequestEnrollResource)r   r   r   r   r   service	inner_reqreqs           r   AddEnrollmentsClient.Add    s    ( --g	33g..g335I!D002FGI MMKKMc	
--
O
O
Q 
 MMLLNcI )!!#&&r   c                 F    U R                   R                  5       nXl        U$ )N)r   EligibleDestinationeligibleGcsBucket)r   gcs_uridests      r   r   1EnrollmentsClient.Gcs_uri_to_eligible_destinationM   s    ==,,.D$Kr   )r   r   )NN)__name__
__module____qualname____firstlineno____doc__r   
ApiVersionr   strboolr)   r   __static_attributes__ r   r   r   r      sX    3 FJ"--+'+'  +' 	+'
 #+'Zr   r   N)r5   $googlecloudsdk.api_lib.audit_managerr   r   objectr   r:   r   r   <module>r=      s    > : 5; ;r   