dZddlmZddlmZddlmZddlZddlZddlmZ ddl m Z ddl m Zdd l mZdd l mZdd lmZddlZd d iZdZdZdZdZdZdZdZe j:e j<j>e j@Gdde jBZ"e j:e j<jFe j<jHe j@Gdde"Z%y)z@Command to list all PyPI modules installed in an Airflow worker.)absolute_import)division)unicode_literalsN)environments_util)base)image_versions_util) resource_args)util)logEXAMPLESaG The following command: $ {command} myenv runs the "python -m pip list" command on a worker and returns the output. The following command: $ {command} myenv --tree runs the "python -m pipdeptree --warn" command on a worker and returns the output. workerzairflow-worker g?g?c<eZdZdZeZedZdZdZ dZ dZ y)Runz5List all PyPI modules installed in an Airflow worker.cZtj|d|jddddy)Nzin which to list PyPI modulesz--tree store_truez List PyPI packages, their versions and a dependency tree, as displayed by the "python -m pipdeptree --warn" command. )defaultactionhelp)r AddEnvironmentResourceArg add_argument)clsparsers 2lib/surface/composer/environments/list_packages.pyArgszRun.ArgsDs9++F,KM   c~|SN)selferrorenv_objs rConvertKubectlErrorzRun.ConvertKubectlErrorQs  Lrc`|jj}tj|}d|v}tj||d5 |jj j }tj|}tjt|}tjjdj|d|dg} |r| jd| jdt dg|j"r| jgd n| jgd tj$tj&|| tj(j  dddy#tj*$r} |j-| |d} ~ wwxYw#1swYyxYw) Nzno-tty) pod_substrkubectl_namespacez?Executing within the following Kubernetes cluster namespace: {}execz--stdinz--ttyz --containerz--)python-m pipdeptree--warn)r*r+piplist)out_func)config gkeCluster command_utilExtractGkeClusterLocationIdTemporaryKubeconfigsoftwareConfig imageVersionFetchKubectlNamespace GetGkePodWORKER_POD_SUBSTRr statusPrintformatappendextendWORKER_CONTAINERtreeRunKubectlCommandAddKubectlNamespaceout KubectlErrorr%) r"argsr$ cluster_idcluster_location_idtty image_version kubectl_nspod kubectl_argses r _RunKubectlzRun._RunKubectlUsg**J&BB7K $ C  ) )Z 355BB !77 F $$(JH   # %Y/    g &],&&  , ,Z FWW]] $/  4 & &3&&q'2235  s+F$ D$E77F! FF!!F$$F-c ~g}|jrd}|jdnd}tjd|g||j }|j st jdd}d}t}d} d } |s4d} tj|tjt tztj|j |j|j |||j } d } | j"}| j$} | j'd | t(k(rt jd | st+|t,zt.}nLt}| D]/} t1j2| j4r | j4nd1| dj6dz}|s4| r|| j8ro| j8j:rXt1j<dj?| j8j<tA| j8j:yyyy#| dz } YxYw)Nr,r-zpip listz list-packages)command subcommand parametersenvironment_ref release_trackzBCannot execute subcommand for environment. Got empty execution Id.Fr) execution_idpod_name pod_namespacenext_line_numberrTrUc|jSr ) lineNumber)lines rzRun._RunApi..sDOOr)keyz*Cannot fetch list-packages command status.zCommand exit code: {})!rAr>environments_api_utilExecuteAirflowCommand ReleaseTrack executionIdr3ErrorDEFAULT_POLL_TIME_SECONDStimesleeprandomuniformPOLL_JITTER_SECONDSPollAirflowCommandrL podNamespace outputEndoutputsortMAX_CONSECUTIVE_POLL_ERRORSminEXP_BACKOFF_MULTIPLIERMAX_POLL_TIME_SECONDSr r<contentr\exitInfoexitCoder#r=exit) r"rFenv_ref cmd_paramsrRexecute_result output_end next_linewait_time_seconds poll_resultcur_consequetive_poll_errorslinesr]s r_RunApiz Run._RunApixsJ yyj!j*@@'') N  % %    N JI1K#$ e* nn113FG H ,>>'33#''(55&#++-  ()$ ** "" 3 4 &)D D  !MNN   6 68M 6D ))DLLDLLb 9"I((1, CF{++ 0D0D0M0M ii'..{/C/C/I/IJK ;   ( ()1N+{!*$)$s B)H22H<cH|jjj}tj||j }t j|jjjr|j||y|j||y)N)rU)rJ) CONCEPTS environmentParserbGetrdimage_versions_command_util%IsVersionAirflowCommandsApiCompatibler1r6r7rrO)r"rFrzr$s rrzRun.Runsymm''--/G#''t0024G"HHnn33@@ ll4! tW%rN) __name__ __module__ __qualname____doc__ DETAILED_HELP detailed_help classmethodrr%rOrrr!rrrr=s4>-    !3F?*B &rrceZdZdZdZy)RunBetaa[List all PyPI modules installed in an Airflow worker. ## EXAMPLES The following command: {command} myenv runs the "python -m pip list" command on a worker and returns the output. The following command: {command} myenv --tree runs the "python -m pipdeptree --warn" command on a worker and returns the output. c|jjxr |jjj}|r+tjt j |dzS|S)Nz Make sure you have followed https://cloud.google.com/composer/docs/how-to/accessing/airflow-cli#running_commands_on_a_private_ip_environment to enable access to your private Cloud Composer environment from your machine.)r1privateEnvironmentConfigenablePrivateEnvironmentr3rfsix text_type)r"r#r$ is_privates rr%zRunBeta.ConvertKubectlErrorsb// I//HH    --    LrN)rrrrr%r!rrrrs $ rr)&r __future__rrrrjrhgooglecloudsdk.api_lib.composerrrbgooglecloudsdk.callioper#googlecloudsdk.command_lib.composerrrr r r3googlecloudsdk.corer rrr:r@rgrrrurtrl ReleaseTracksrdGADefaultUniverseOnlyCommandrBETAALPHArr!rrrsG&' V(b=D#   $# D%%(()C&$,,C&*C&LD%%**D,=,=,C,CDcEr