U eb@sxddlZddlZddlZddlZddlmZddlmZGdddeej Z Gdddeej Z Gdd d eej Z dS) N)BaseCommandMixin)CONFIGcsZeZdZdZejjjejjjejjjdZ fddZ e ddZ ddZ d d ZZS) RootCommandzAThis class defers to either the PluginCommand or the builtin cmds)infotoolsdevc sddl}ddl}ddddddg}|d}g}g}|jD]Rtfd d |Dr\||} | dk r8| \} |d | fq8|s|r|rd d |} t j t d| dd|rddd |D} d| } t j t d| dd| dtj||d|_dS)Nru‘u’u“u”u—u–z--m-(\S+)-categoryc3s|]}|kVqdSN).0xZcommandr Flib/python3.8/site-packages/q2cli-2024.2.0-py3.8.egg/q2cli/commands.py %sz'RootCommand.__init__..z --m-%s-columnzdError: Detected invalid character in: %s Verify the correct quotes or dashes (ASCII) are being used., errorTerr css|]\}}d||fVqdS)zInstead of %s, trying using %sNr )r oldnewr r r r5sz|Error: The following options no longer exist because metadata *categories* are now called metadata *columns* in QIIME 2. %s)resyscompileargvanyappend fullmatchgroupsjoinclickechor cfg_styleexitsuper__init___plugins) selfargskwargsrrZunicodesZcategory_regexZ invalid_charsZ categoriesmatchZ param_namemsgZold_to_new_names __class__r r r%s<       zRootCommand.__init__cCs\ddl}|jdkr(ddl}|jjjj|_i}|jD] \}}|dr6|||j |<q6|S)Nractions) q2cli.utilr&Zq2cli.core.cachecorecacheZCACHEpluginsitemsutil to_cli_name)r'q2cliZname_mapnamepluginr r r _plugin_lookupDs zRootCommand._plugin_lookupcCs$ddl}|j}t|j}|||S)Nr) itertools_builtin_commandssortedr9chain)r'ctxr:builtinsr2r r r list_commandsUs zRootCommand.list_commandsc Cs||jkr|j|Sz|j|}Wntk rddlm}|||j}t|dkrdd|d}n|rxdd|}nd}tjt dd ||d d | d YnXt ||S) Nrget_close_matches Did you mean %r? (Possible commands: %s)rrz.Error: QIIME 2 has no plugin/command named %r.Tr) r;r9KeyErrorr/rBlenrr r!rr"r# PluginCommand)r'r>r7r8rB possibilitieshintr r r get_command_s,     zRootCommand.get_command)__name__ __module__ __qualname____doc__r6Zbuiltinrrrr;r%propertyr9r@rM __classcell__r r r,r rs )  rcsPeZdZdZfddZddZddZdd Zd d Zd d Z ddZ Z S)rJz2Provides ActionCommands based on available Actionsc sddl}||_i|_i|_|dD]8\}}|drL||j|j|<q&||j|j|<q&d|d}d|d} d|d } d | | |g} t j d d d d |j dd|j |j|j|jg} |jikr| t j dd d d |jddtj|f||d| | d|dS)Nrr._zGetting user support: %sZuser_support_textzPlugin website: %swebsitezDescription: %s description )z --versionTFzShow the version and exit.)is_flagZ expose_valueZis_eagercallbackhelp)z--show-hidden-actionszThis plugin has hidden actions with names starting with '_'. These are generally called internally by pipelines. Passing this flag will display those actions.Zshort_description) short_helprZparams)r/_plugin_action_lookup_hidden_actionsr3 startswithr4Zhidden_to_cli_namer5rr Option _get_versionexample_data_option _get_plugincitations_option_get_citation_recordsr_get_hidden_actionsr$r%) r'r8r7r(r)r6idaZsupportrUrVhelp_r\r,r r r%{sJ         zPluginCommand.__init__c Cs|r |jrdSddl}|j}|jD]$}|jd|jkr*|j}|j }qXq*d}}t d|jd|jd||f| dS)Nrr7z [UNKNOWN]z=QIIME 2 Plugin '%s' version %s (from package '%s' version %s)version) resilient_parsingr/r4get_plugin_managerr2valuesr]r7Z project_namerkr r!r#) r'r>paramvaluer6pmr8pkg_nameZ pkg_versionr r r rbs&  zPluginCommand._get_versioncCs$ddl}|j}|j|jdjSNrr7)r/r4rmr2r] citationsr'r6rqr r r rfs z#PluginCommand._get_citation_recordscCsF|r |jrdSddlm}|j|j|||jd|dS)z0Add actions that start with _ back to the lookupNr)r!)color) rlZ click.utilsr!r^updater_Zget_helprvr#)r'r>rorpr!r r r rgs  z!PluginCommand._get_hidden_actionscCs"ddl}|j}|j|jdSrs)r/r4rmr2r]rur r r rds zPluginCommand._get_plugincCs t|jSr)r<r^r'r>r r r r@szPluginCommand.list_commandsc Csz|j|j|j|}Wntk rddlm}|||j}t|dkr^d|d}n|rrdd|}nd}tj t dd |j d |f|d d | d YnXt||j |S)NrrArCrDrErrFrz*Error: QIIME 2 plugin %r has no action %r.r7TrrG)r^rwr_rHr/rBrIrr r!rr"r]r# ActionCommand)r'r>r7actionrBrKrLr r r rMs*    zPluginCommand.get_command) rNrOrPrQr%rbrfrgrdr@rMrSr r r,r rJys +rJcsjeZdZdZfddZddZddZdd Zd d Zd d Z ddZ dddZ ddZ ddZ ZS)ryzGA click manifestation of a QIIME 2 API Action (Method/Visualizer) c sxddl}ddl}||_||_|\|_|_|_tj dg|jj ddtj dgddddg|_ | }|j d kr|j tj d gdtd d tj d gddddtj dgddddtj dgdtjddddd tj dgdtjddddd g|j |j|j|jd|j|jg|j|j|j|j }|dg}|jdrT|tddtj||||dd|d dS)!Nrz --output-dirz)Output unspecified results to a directory)typerZz--verbose / --quietFzDisplay verbose output to stdout and/or stderr during execution of this action. Or silence output if execution is successful (silence is golden).)defaultrequiredrZpipelinez--recycle-poolawUse a cache pool for pipeline resumption. QIIME 2 will cache your results in this pool for reuse by future invocations. These pool are retained until deleted by the user. If not provided, QIIME 2 will create a pool which is automatically reused by invocations of the same action and removed if the action is successful. Note: these pools are local to the cache you are using.)r}r{rZz --no-recycleTzrDo not recycle results from a previous failed pipeline run or save the results from this run for future recycling.)rXr}rZz --parallelzQExecute your action in parallel. This flag will use your default parallel config.z--parallel-config)existsZdir_okayzEExecute your action in parallel using a config at the indicated path.z --use-cache)rZ file_okayaLSpecify the cache to be used for the intermediate work of this pipeline. If not provided, the default cache under $TMP/qiime2/ will be used. IMPORTANT FOR HPC USERS: If you are on an HPC system and are using parallel execution it is important to set this to a location that is globally accessible to all nodes in the cluster.rhrV deprecatedwarningz\WARNING: This command is deprecated and will be removed in a future version of this plugin.r7rW)r\rYr[rZ)r/Zq2cli.click.typer8rz_build_generated_options_inputs_params_outputsr rar{Z OutDirType_misc _get_actionextendstrPathr4rcrdrerfrrr"r$r%r)r'r7r8rzr6Z action_objZoptionsrjr,r r r%sl           $    zActionCommand.__init__c Csddl}g}g}g}|jdD]\}|}|d}|dkrB|}n|dkrP|}n|}|jjjfd|di|}||q|||fS)Nr signaturer{inputZ parameterprefix)Zq2cli.click.optionrzcopypopr ZoptionZGeneratedOptionr) r'r6Zinputsr\outputsitemr{ZstorageZoptr r r rFs  z&ActionCommand._build_generated_optionscCs"|j|j|j|j||gdS)N)ZInputsZ ParametersZOutputsZ Miscellaneous)rrrrZget_help_optionrxr r r get_opt_groups]s zActionCommand.get_opt_groupscCs |jSr)rrt)r'r r r rfesz#ActionCommand._get_citation_recordscCs"ddl}|j}|j|jdSrs)r/r4rmr2r8rur r r rdhs zActionCommand._get_plugincCs|}|j|jdS)Nrh)rdr.rz)r'r8r r r rmszActionCommand._get_actionc0 sddl}ddl}ddl}ddlm}m}m}ddlm}ddl m } | d} | r| ddd} | r|j | r|| rtd | d | d d} | d d } | dk r| rtd| dd}|dk r||std|d| dd }| dd}|dk r d}| d}|dkr*d }d }n|r6d }nd}||di}i}|D]$\}}|d^}}d|}|dkr|dkr|j | |}|||<n|dkr|||dtd <n|dkrp|dk rp|}t|trfdd|D}njt|tst|| r,fdd|D}n:t|trPtfd d|D}n|dk rfj|}|||<n|||<qT||}|}||jd|j}| s|jd!kr| dkr|} | dk r| |kr| krd"| j f}|!t"#d#|d}|s*ddl$}|j%d$d%d d&d'}|j&r`d(t'j()|j*d)|j+f}|!t"#d#|d } zz|j(j.||d*||rdd+l/m0}!m1}"|j2}|dkr|"}#n|!|\}$}%|"|$|%}#|#|3||| }&W5QRXn|3||| }&W5QRXWnVt4k rR}'z6d,t'j()|j*d)}(|r0d-}t'j(j5|'|(|d.W5d}'~'XYnXd} W5|r| r|j |j+r|,|-|j+X| dk r|6| t7|&|D]\})}*t|*t8rt|*dkr|*d}*||*r0| dkr0||*\}+}||+},t|)| r|,9|)||*}-n|,|)||*}-n |)|*}-|s||*rdd/|)jd0|+d1|}.nsz*ActionCommand.__call__..csi|]\}}|j|qSr r)r krrr r sz*ActionCommand.__call__..csg|]}j|qSr rrrr r rsr~zFThe pool '%s' does not exist on the cache at '%s'. It will be created.rzqiime2-q2cli-err-z.logw)rsuffixdeletemodezaPlugin warning from %s: %s is deprecated and will be removed in a future version of this plugin.r7)stdoutstderr)get_config_from_fileParallelConfigzPlugin error from %s:r)header tracebackzAdded z to cache: z as: z Collection[]zSaved z to: Zsuccess);osr Z qiime2.utilr/rrrZqiime2.core.cacherZ qiime2.sdkrrrsplitrrZis_cache ValueErrorr3splitrrI isinstancelistdictsetrr_order_outputsrZ plugin_idrhr{Z get_poolsr!rr"tempfileZNamedTemporaryFilerr6r4r5r8r7closeremoveZredirected_stdioZqiime2.sdk.parallel_configrrr_execute_action ExceptionZexit_with_errormakedirsziptupleZsave_collectionrn)0r'r)rr Zqiime2rrrrrrZpotential_cacherrZ used_cacherZparallel_config_fprquiet argumentsZ init_outputskeyrprpartsZvalue_rrzZ default_poolr+logrZcleanup_logfilerrrZconfigmappingresultserresultoutput cache_pathZ output_cachermessager{r rr __call__qs*                              $         zActionCommand.__call__Nc Cs`|R|dkr"|f|}|}n0|j|dd}||f|}|}W5QRXW5QRX|S)NT)rZreuse)Z_resultZ create_pool)r'rzrr1rrZpoolr r r rBs   zActionCommand._execute_actioncCs6g}|jdD]"}|ddkr|||dq|S)Nrr{rr7)rzr)r'rZorderedrr r r rTs  zActionCommand._order_outputsc Csb|jdr^|tjddd:|jdD](}|d|j|||dq*W5QRXdS)NZepilogZExamplesT)Zbold r)rzZsectionr ZstylewriteZcurrent_indent)r'r>Z formatterliner r r format_epilog[s   zActionCommand.format_epilog)N)rNrOrPrQr%rrrfrdrrrrrrSr r r,r rys KR ry)r Zq2cli.builtin.devr6Zq2cli.builtin.infoZq2cli.builtin.toolsZq2cli.click.commandrZq2cli.core.configrZ MultiCommandrrJZCommandryr r r r  s  f~