
                             S 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                  4S	 jr " S
 S\5      rg)z5Utilities for calling the Composer ListWorkloads API.    )absolute_import)division)unicode_literals)util)base	workloadsi  c                 B    [         R                  " U 5      R                  $ N)api_utilGetClientInstanceprojects_locations_environmentsrelease_tracks    Blib/googlecloudsdk/api_lib/composer/environments_workloads_util.py
GetServicer      s    		#	#
445    c                   X    \ rS rSrSr\R                  R                  4S jrS r	S r
Srg)EnvironmentsWorkloadsService!   z3Provides workloads from Composer ListWorkloads API.c                 d    S U l         Xl        [        R                  " U R                  S9U l        g )Nr   )clientr   r   GetMessagesModulemessages)selfr   s     r   __init__%EnvironmentsWorkloadsService.__init__$   s'    DK&..T=O=OPDMr   c                     U R                   c/  [        R                  " U R                  5      R                  U l         U R                   $ r
   )r   r   r   r   )projects_locations_environments_workloads)r   s    r   	GetClient&EnvironmentsWorkloadsService.GetClient)   s<    {{..


GG k ;;r   c           	          U R                   R                  nU/n[        [        R                  " X R                  5       U[        [        5      5      $ )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   List!EnvironmentsWorkloadsService.List0   sK     	OO  &&I%%^^%y/9	
 r   )r   r   r   N)__name__
__module____qualname____firstlineno____doc__r   ReleaseTrackGAr   r   r*   __static_attributes__ r   r   r   r   !   s$    ;#'#4#4#7#7 Q
r   r   N)r0   
__future__r   r   r   googlecloudsdk.api_lib.composerr   r   googlecloudsdk.callioper   r%   r&   r1   r2   r   objectr   r4   r   r   <module>r9      sE    < &  ' < (	 "..11 5
6 r   