
    
                         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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diZdZej*                   G d dej,                               Zy)z#Command to list Composer workloads.    )absolute_import)division)unicode_literals)environments_util)environments_workloads_util)base)flags)image_versions_util)resource_args)utilEXAMPLESzz      To display Composer workloads for the environment named ``environment-1'', run:
      $ {command} environment-1
    zrThe operation is not supported for given environment. Composer version {composer_version} or greater is required.
c                   *    e Zd ZdZeZed        Zd Zy)ListWorkloadszIList Composer workloads, supported in Composer 3 environments or greater.c                 0    t        j                  | d       y )Nzfor which to display workloads)r   AddEnvironmentResourceArg)parsers    3lib/surface/composer/environments/list_workloads.pyArgszListWorkloads.Args-   s    ++0    c                    |j                   j                  j                         }| j                         }t	        j
                  || j                               }t        j                  |j                  j                  j                        s7t        j                  t        j                  t        j                               t#        j$                  |      }|j'                  |      S )N)release_track)image_version)composer_version)CONCEPTSenvironmentParseReleaseTrackenvironments_api_utilGetimage_versions_command_utilIsVersionComposer3CompatibleconfigsoftwareConfigimageVersioncommand_utilInvalidUserInputErrorCOMPOSER3_IS_REQUIRED_MSGformatr	   MIN_COMPOSER3_VERSIONworkloads_utilEnvironmentsWorkloadsServiceList)selfargsenv_refr   env_objworkloads_services         r   RunzListWorkloads.Run3   s    mm''--/G%%'M#''t002G 'CCnn33@@ ..
#
*
*$:: +   'CC !! r   N)	__name__
__module____qualname____doc__DETAILED_HELPdetailed_helpstaticmethodr   r2    r   r   r   r   '   s!    Q- 
r   r   N)r6   
__future__r   r   r   googlecloudsdk.api_lib.composerr   r   r   r*   googlecloudsdk.callioper   #googlecloudsdk.command_lib.composerr	   r
   r    r   r   r%   r7   r'   DefaultUniverseOnlyCommandr   r:   r   r   <module>rA      sd    * &  ' V Y ( 5 b = D  	
 
  DLL    r   