
                             d Z ddlm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e	j                  j                  fd	Z G d
 de      Zy)z5Utilities for calling the Composer ListWorkloads API.    )absolute_import)division)unicode_literals)util)base	workloadsi  c                 @    t        j                  |       j                  S N)api_utilGetClientInstanceprojects_locations_environmentsrelease_tracks    Blib/googlecloudsdk/api_lib/composer/environments_workloads_util.py
GetServicer      s    		#	#
445    c                   N    e Zd ZdZej
                  j                  fdZd Zd Z	y)EnvironmentsWorkloadsServicez3Provides workloads from Composer ListWorkloads API.c                 j    d | _         || _        t        j                  | j                        | _        y )Nr   )clientr   r   GetMessagesModulemessages)selfr   s     r   __init__z%EnvironmentsWorkloadsService.__init__$   s)    DK&D..T=O=OPDMr   c                     | j                   .t        j                  | j                        j                  | _         | j                   S r
   )r   r   r   r   )projects_locations_environments_workloads)r   s    r   	GetClientz&EnvironmentsWorkloadsService.GetClient)   s<    {{..


GG k ;;r   c           	          | j                   j                  }|g}t        t        j                  || j                         |t        t                    S )zERetrieves list of Composer workloads from Composer ListWorkloads API.)r   9ComposerProjectsLocationsEnvironmentsWorkloadsListRequestlistr   AggregateListResultsr   LIST_FIELD_NAME	PAGE_SIZE)r   project_location_refrequest	locationss       r   Listz!EnvironmentsWorkloadsService.List0   sM     	OO  &&I%%T^^%y/9	
 r   N)
__name__
__module____qualname____doc__r   ReleaseTrackGAr   r   r'    r   r   r   r   !   s$    ;#'#4#4#7#7 Q
r   r   N)r+   
__future__r   r   r   googlecloudsdk.api_lib.composerr   r   googlecloudsdk.callioper   r"   r#   r,   r-   r   objectr   r.   r   r   <module>r3      sE    < &  ' < (	 "..11 5
6 r   