U eA@sGdddeZddZddZddZdd Zd d Zd d Zd;ddZddZ ddZ GdddZ ddZ ddZ dc@s eZdZdS) OutOfDiskN)__name__ __module__ __qualname__rrBlib/python3.8/site-packages/q2cli-2024.2.0-py3.8.egg/q2cli/util.pyr srcCsZddl}|jd}|dk r@|||j|jBr@|j|ddSddl}|j dddSdS)NZ CONDA_PREFIXvarZq2cliF)Zroaming) osenvirongetaccessW_OKX_OKpathjoinclick get_app_dir)r Z conda_prefixrrrrrs  rcCsddl}|jtdS)Nrcache)os.pathrrrr rrr get_cache_dirsrcCsddl}|jtdS)Nrz completion.sh)rrrrrrrrget_completion_path"srcCs0|dstd|dt|}|dddS)N_z The name 'zn' does not start with '_' meaning it is not a hidden action and this method should not have been called on it.-) startswith ValueError to_cli_namereplacenamerrrhidden_to_cli_name's r!cCs |ddS)Nrrrrrrrr4srcCs |ddS)Nrrr"rrrr to_snake_case8sr#An error has been encountered:stderrrc Cs ddl}ddl}ddl}ddl}ddlm}g} d} |dkrJ|j} dg} n|dk rb|} d| jg} |t |d} || g| } |dk r|j t |||j | d| d|j|d d | d d | s|jd d z||Wn tk r||YnXdS) NrCONFIGr%zSee above for debug info.zDebug info has been saved to %sz )file errorz Terr)sys tracebacktextwraprq2cli.core.configr'r%r indentstrprint_exceptiontype __traceback__writeecho cfg_stylerZget_current_contextexit RuntimeError) eheaderr.statusr-tbr/rr'ZfooterZtb_filer*Zsegmentsrrrexit_with_error<s2     r?c Csddlm}d|krdSt|\}}z.||rP|sJtd|dnWdSWn4tk r}zdt|krrn|W5d }~XYnXdS) zRDetermines if an output path follows the format /path_to_extant_cache:key rCache:FzKey 'z' is not a valid Python identifier. Keys must be valid Python identifiers. Python identifier rules may be found here https://www.askpython.com/python/python-identifiers-rules-best-practicesTzNo such file or directoryN)qiime2.core.cacherA_get_cache_path_and_keyis_cache isidentifierrFileNotFoundErrorr2)fprA cache_pathkeyr;rrroutput_in_cache`s       rKcs`ddl}fdd|D}|s6|j|dd}|t|dkr\|dddg}|S) Nrcsg|]}|r|qSr)r).0mrrr s z%get_close_matches..g?)cutoffz...)diffliblowerget_close_matchessortlen)r Z possibilitiesrRZmatchesrrrrTs rTc@s.eZdZd ddZddZdd Zd d Zd S)pretty_failurer$r%rcCs||_||_||_dSN)r<r.r=)selfr<r.r=rrr__init__szpretty_failure.__init__csfdd}j|_|S)Nc s. ||diW5QRSQRXdS)NZfailurer)argskwargsfunctionrYrrwrappedsz(pretty_failure.__call__..wrapped)r)rYr^r_rr]r__call__szpretty_failure.__call__cCs|SrXr)rYrrr __enter__szpretty_failure.__enter__cCs,|dk r(t|dkr(t||j|j|jdS)N1F)r2r?r<r.r=)rYexc_typeZexc_valZexc_tbrrr__exit__szpretty_failure.__exit__N)r$r%r)rrrrZr`rardrrrrrWs  rWcCsddl}tttttd}|d}|r|ddkrJ|ddkrJ||dS|ddkr|dd kr|d d}|d d }|dk r|d ds|d 7}|dk r|d d s|d 8}|||S|ddkr|dd kr|d Stn ||dSdS)Nr)IntStrFloatZColorZBool predicater ZChoicesrfchoicesZRangererangerZ inclusiverg)rintr2floatboolZChoiceZIntRangeNotImplementedError)Zastrmappingrhstartendrrrconvert_primitives.    rrcs<ddlddlmfdd}jdgddd|dd S) Nrr&c s|r |jrdS}|rddl}ddl}|jddt|D}|,}|d||j | ddW5QRX| n j ddd d | d dS) NrcSsg|]\}}d||fqS)zkey%dr)rLirrrrrNsz6citations_option..callback..zE% use `qiime tools citations` on a QIIME 2 result for complete list F)nlZproblemzNo citations found.Tr+r) resilient_parsingio qiime2.sdksdkZ Citations enumerateStringIOr6Zsaver7getvaluer9r8)ctxparamvalueZrecordsrwqiime2Z citationsZfhr'rget_citation_recordsrrcallbacks&      z"citations_option..callbackz --citationsTFzShow citations and exit.)Zis_flag expose_valueis_eagerrhelp)rr0r'Option)rrrrrcitations_options  rNcs>ddlddlm}fdd}jdg|dd|dd S) Nr) OutDirTypec s|r |jrdSddlmm}}dk rF|j}|||}n |||}d}|D]"\}} jd|| fddd}qZ|r|njdddd |d dS) NrFzSaved %s to: %sZgreen)fgTzNo example data found.Zyellow)rr,r) rvZq2cli.core.usagecoreusageZactionsZwrite_example_dataZwrite_plugin_example_dataZsechor9) r}r~rrZpluginaction generatorZranZhintr action_namer get_pluginrrrs      z%example_data_option..callbackz--example-dataTFzWrite example data and exit.)r4rrrr)rZq2cli.click.typerr)rrrrrrrexample_data_options rcCsddl}z|jjWS|jjk r|ddl}d|jkrnddlm}|}|jjdd}| ||YS|jYSXdS)NrZ MYSTERY_STEW) create_pluginF)Z add_plugins) rxryZ PluginManagerZreuse_existingZUninitializedPluginManagerErrorr r Zq2_mystery_stew.plugin_setuprZ add_plugin)rr rZthe_stewZpmrrrget_plugin_managers   rc Csddl}ddl}t|\}}|dkrz|j|}WnZtk r}z<|rVd|krV|}d|}d|jkrldnd}t|||dW5d}~XYnX|S)NrrBz8There was an issue with loading the file %s as metadata: --verboser%r<r.)rr-_load_metadata_artifactMetadataload Exceptionargvr?)rHrr-Zmetadatar*r;r<r>rrr load_metadata&s   rc Csddl}ddl}t|\}}|d}t|tr2|d}|rzHd}t||jrTtd|sntd|jdd}| |j dfWStk r}z0d|d}d |j krdn|}t |||d W5d}~XYqXnd|fSdS) Nrrr%z4Visualizations cannot be viewed as QIIME 2 metadata.zArtifacts with type z& cannot be viewed as QIIME 2 metadata.z-There was an issue with viewing the artifact z as QIIME 2 Metadata:rr) rr- _load_input isinstancerZ VisualizationrZ has_metadatar4viewrrr?) rHrr-artifactr*Z default_tbr;r<r>rrrr9s2     "rFc Cs0ddl}d}|st}|j|rRt||dkrDtd|dt|\}}nd|krt|\}}|dkrt |\}}t|\}}|dkrd}t |\}}|rt |\}}t |\}}|dk rd}n t|\}}t |t r$|j dkr$ddlm}t|j} dtd| d|d | d fS||f|fS) NrzProvided directory 'z ' is empty.rB) get_cachez#There was not enough space left on z to use the artifact zQ. (Try setting $TMPDIR to a directory with more space, or increasing the size of ))r rrisdirrVlistdirr_load_collection_load_input_file_get_path_and_collection_key_load_input_cacherOSErrorerrnorCrr2r) rHrr rJrrr*Znew_fprrrrrr_s6           rcCs`ddl}ddl}|j|d}|j|r:t||\}}n|d|dt|\}}||fS)Nrz.orderzThe directory zy does not contain a .order file. The files will be read into the collection in the order the filesystem provides them in.)r warningsrrisfile_load_ordered_collectionwarn_load_unordered_collection)rHr rorder_fp artifactsr*rrrrs  rc Cszddl}i}t|X}|D]D}|j||d}t|\||<}|r"d|fW5QRSq"W5QRX|dfS)Nrz.qza)r openread splitlinesrrr)rHrr rZorder_fhrJZ artifact_pathr*rrrrs $rcCsRddl}i}||D]2}|j||}t|\||<}|rd|fSq|dfS)Nr)r rrrr)rHr rrZ artifact_fpr*rrrrsrc CsDd}}z t|}Wn&tk r:}z|}W5d}~XYnX||fSrX)try_as_cache_inputr)rHrr*r;rrrrs  rc Csddl}ddlm}||}d}z|jj|}Wn<tk rn}zt|trR|d|fWYSd}~XYnXz|jj |}|dfWStk r}z4|rt d|j d|j |n|}d|fWYSd}~XYnXdS)Nr) expanduserziIt looks like you have an Artifact but are missing the plugin(s) necessary to load it. Artifact has type z and format ) rxrrryZResultpeekrr SyntaxErrorrrr4format)rHrrrr*rr;rrrrs,   rcCsXddl}ddlm}t|\}}|j|r6||sFtd|d||}||S)zM Determine if an input is in a cache and load it from the cache if it is rNr@z The path z is not a valid cache.) r rrArDrexistsrErr)rHr rArIrJrrrrrs  rcCs |ddSNrBr)rsplitrHrrrrD srDcCs |ddSr)splitrrrrrsrcCs*ddlm}d|d||dS)Nr)sha1Zrecycle_rzutf-8)ZhashlibrencodeZ hexdigest)Z plugin_actionrrrrget_default_recycle_pools r)r$r%r)N)F)rrrrrr!rr#r?rKrTrWrrrrrrrrrrrrrrrDrrrrrr s:  $!# !& ? $