U ldX@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-2023.5.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 )  rcsHeZdZdZfddZddZddZdd Zd d Zd d Z Z S)rJz2Provides ActionCommands based on available Actionsc sddl||_fdd|dD|_d|d}d|d}d |d }d |||g}tjd d dd |jddj |j j |j g} t j|f||d|| d|dS)Nrcsi|]\}}j||qSr )r4r5)r idar6r r s z*PluginCommand.__init__..r.zGetting user support: %sZuser_support_textzPlugin website: %swebsitezDescription: %s description )z --versionTFzShow the version and exit.)is_flagZ expose_valueZis_eagercallbackhelpZshort_description) short_helpr]params)r/_pluginr3_action_lookuprr Option _get_versionr4example_data_option _get_plugincitations_option_get_citation_recordsr$r%) r'r8r7r(r)ZsupportrXrYhelp_r_r,rVr r%{s.        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) Zresilient_parsingr/r4get_plugin_managerr2valuesr`r7Z project_namerir r!r#) r'r>Zparamvaluer6pmr8pkg_nameZ pkg_versionr r r rcs&  zPluginCommand._get_versioncCs$ddl}|j}|j|jdjSNrr7)r/r4rjr2r` citationsr'r6rmr r r rgs z#PluginCommand._get_citation_recordscCs"ddl}|j}|j|jdSro)r/r4rjr2r`rqr r r res zPluginCommand._get_plugincCs t|jSr)r<rar'r>r r r r@szPluginCommand.list_commandsc Csz|j|}Wntk rddlm}|||j}t|dkrPd|d}n|rddd|}nd}tjt dd |j d |f|d d | d YnXt ||j |S)NrrArCrDrErrFrz*Error: QIIME 2 plugin %r has no action %r.r7TrrG) rarHr/rBrIrr r!rr"r`r# ActionCommand)r'r>r7actionrBrKrLr r r rMs(    zPluginCommand.get_command) rNrOrPrQr%rcrgrer@rMrSr r r,r rJys rJcs`eZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ Z S)rszGA 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)typer]z--verbose / --quietFzDisplay verbose output to stdout and/or stderr during execution of this action. Or silence output if execution is successful (silence is golden).)defaultrequiredr]pipelinez--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.)rwrur]z --no-recycleTzrDo not recycle results from a previous failed pipeline run or save the results from this run for future recycling.)r[rwr]z --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)ryZ 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.rTrY deprecatedwarningz\WARNING: This command is deprecated and will be removed in a future version of this plugin.r7rZ)r_r\r^r])r/Zq2cli.click.typer8rt_build_generated_options_inputs_params_outputsr rbruZ OutDirType_misc _get_actionextendstrPathr4rdrerfrgrrr"r$r%r)r'r7r8rtr6Z action_objZoptionsrhr,r r r%sl           $    zActionCommand.__init__c Csddl}g}g}g}|jdD]\}|}|d}|dkrB|}n|dkrP|}n|}|jjjfd|di|}||q|||fS)Nr signatureruinputZ parameterprefix)Zq2cli.click.optionrtcopypopr ZoptionZGeneratedOptionr) r'r6Zinputsr_outputsitemruZstorageZoptr r r r|s  z&ActionCommand._build_generated_optionscCs"|j|j|j|j||gdS)N)ZInputsZ ParametersZOutputsZ Miscellaneous)r}r~rrZget_help_optionrrr r r get_opt_groups3s zActionCommand.get_opt_groupscCs |jSr)rrp)r'r r r rg;sz#ActionCommand._get_citation_recordscCs"ddl}|j}|j|jdSro)r/r4rjr2r8rqr r r re>s zActionCommand._get_plugincCs|}|j|jdS)NrT)rer.rt)r'r8r r r rCszActionCommand._get_actionc, 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 ddlm}d}||| d}|dkr>d }d }n|rJd }nd}||di}i}|D]$\}}|d^}}d|}|dkr|dkr|j | |}|||<n|dkr|||dtd <n|dkr|dk r|}t|trfdd|D}njt|ts(t|| r@fdd |D}n:t|trdtfd!d|D}n|dk rzj|}|||<n|||<qh||}|}||jd|j }| s|j!d"kr| dkr|} | dk r| |kr| "krd#| j f}|#t$%d$|d}|s>ddl&} | j'd%d&d d'd(}|j(rtd)t)j*+|j,d*|j-f}|#t$%d$|d }!zzz|j*j0||d+`|r|j1}| dkr|f|}"|"2}"n0j3| dd,}#|#|f|}"|"2}"W5QRXW5QRXWnVt4k rJ}$z6d-t)j*+|j,d*}%|r(d.}t)j*j5|$|%|d/W5d}$~$XYnXd}!W5|r|!r|j |j-r|.|/|j-X| dk r|6| t7|"|D]\}&}'t|'t8rt|'dkr|'d}'||'r(| dkr(||'\}(}||(t|&| r9|&||'})n|&||'})n |&|'})|s||'r\d0|&j!d1|(d2|}*nsz*ActionCommand.__call__..csi|]\}}|j|qSr r)r krrr r rWsz*ActionCommand.__call__..csg|]}j|qSr rrrr r rsrxzFThe pool '%s' does not exist on the cache at '%s'. It will be created.r{zqiime2-q2cli-err-z.logw)rsuffixdeletemodezaPlugin warning from %s: %s is deprecated and will be removed in a future version of this plugin.r7)stdoutstderr)keyZreusezPlugin 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.sdkrrrsplitrryZis_cache ValueErrorZqiime2.sdk.parallel_configrr3splitrrI isinstancelistdictsetrr_order_outputsrZ plugin_idrTruZ get_poolsr!rr"tempfileZNamedTemporaryFilerzr6r4r5r8r7closeremoveZredirected_stdiorZ_resultZ create_pool ExceptionZexit_with_errormakedirsziptupleZsave_collectionrk),r'r)rr Zqiime2rrrrrrZpotential_cacherrZ used_cacherZparallel_config_fprrquietZ argumentsZ init_outputsrrlrpartsZvalue_rrtZ default_poolr+logrZcleanup_logfileZresultsZpoolerresultoutput cache_pathrmessagerur rr __call__Gs                                  $         zActionCommand.__call__cCs6g}|jdD]"}|ddkr|||dq|S)Nrrurr7)rtr)r'rZorderedrr r r rs  zActionCommand._order_outputsc Csb|jdr^|tjddd:|jdD](}|d|j|||dq*W5QRXdS)NZepilogZExamplesT)Zbold r)rtZsectionr ZstylewriteZcurrent_indent)r'r>Z formatterliner r r format_epilogs   zActionCommand.format_epilog)rNrOrPrQr%r|rrgrerrrrrSr r r,r rss KRrs)r Zq2cli.builtin.devr6Zq2cli.builtin.infoZq2cli.builtin.toolsZq2cli.click.commandrZq2cli.core.configrZ MultiCommandrrJZCommandrsr r r r  s  fT