U ld>@sGdddeZddZddZddZdd Zd d Zd9ddZddZddZ GdddZ ddZ ddZ d:ddZ ddZd d!Zd"d#Zd;d%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8ZdS)_get_cache_path_and_keyis_cache isidentifier ValueErrorFileNotFoundErrorr/)fpr> cache_pathkeyr8rrroutput_in_cacheSs       rIcs`ddl}fdd|D}|s6|j|dd}|t|dkr\|dddg}|S) Nrcsg|]}|r|qSr) startswith).0mrrr ys z%get_close_matches..g?)cutoffz...)diffliblowerget_close_matchessortlen)rZ possibilitiesrQZmatchesrrrrSts rSc@s.eZdZd ddZddZdd Zd d Zd S)pretty_failurer r!r"cCs||_||_||_dSN)r9r+r:)selfr9r+r:rrr__init__szpretty_failure.__init__csfdd}j|_|S)Nc s. ||diW5QRSQRXdS)NZfailurer)argskwargsfunctionrXrrwrappedsz(pretty_failure.__call__..wrapped)r)rXr]r^rr\r__call__szpretty_failure.__call__cCs|SrWr)rXrrr __enter__szpretty_failure.__enter__cCs,|dk r(t|dkr(t||j|j|jdS)N1F)r/r<r9r+r:)rXexc_typeZexc_valZexc_tbrrr__exit__szpretty_failure.__exit__N)r r!r")rrrrYr_r`rcrrrrrVs  rVcCsddl}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 predicaterZChoicesrechoicesZRangerdranger"Z inclusiverf)rintr/floatboolZChoiceZIntRangeNotImplementedError)Zastrmappingrgstartendrrrconvert_primitives.    rqcs<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)rKirrrrrMsz6citations_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 enumerateStringIOr3Zsaver4getvaluer6r5)ctxparamvalueZrecordsrvqiime2Z citationsZfhr$rget_citation_recordsrrcallbacks&      z"citations_option..callbackz --citationsTFzShow citations and exit.)Zis_flag expose_valueis_eagerrhelp)rr-r$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") ruZq2cli.core.usagecoreusageZactionsZwrite_example_dataZwrite_plugin_example_dataZsechor6) r|r}r~rZpluginaction generatorZranZhintr action_namer get_pluginrrrs      z%example_data_option..callbackz--example-dataTFzWrite example data and exit.)r1rrrr)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) rwrxZ 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)Nrr?z8There was an issue with loading the file %s as metadata: --verboser!r9r+)rr*_load_metadata_artifactMetadataload Exceptionargvr<)rFrr*Zmetadatar'r8r9r;rrr load_metadatas   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) Nrr"r!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_metadatar1viewrrr<) rFrr*artifactr'Z default_tbr8r9r;rrrr,s2     "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.r?) 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 rrisdirrUlistdirrD_load_collection_load_input_file_get_path_and_collection_key_load_input_cacherOSErrorerrnor@rr/r) rFrr rHrrr'Znew_fprrrrrrRs6           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)rFr rorder_fp artifactsr'rrrrs  rc Cszddl}i}t|X}|D]D}|j||d}t|\||<}|r"d|fW5QRSq"W5QRX|dfS)Nrz.qza)r openread splitlinesrrr)rFrr rZorder_fhrHZ artifact_pathr'rrrrs $rcCsRddl}i}||D]2}|j||}t|\||<}|rd|fSq|dfS)Nr)r rrrr)rFr rrZ artifact_fpr'rrrrsrc CsDd}}z t|}Wn&tk r:}z|}W5d}~XYnX||fSrW)try_as_cache_inputr)rFrr'r8rrrrs  rc Csddl}d}z|jj|}Wn<tk rZ}zt|tr>|d|fWYSd}~XYnXz|jj|}|dfWStk r}z4|rtd|j d|j |n|}d|fWYSd}~XYnXdS)NrziIt looks like you have an Artifact but are missing the plugin(s) necessary to load it. Artifact has type z and format ) rwrxZResultpeekrr SyntaxErrorrrDr1format)rFrrr'rr8rrrrs(  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 rr>rArexistsrBrDr)rFr r>rGrHrrrrrs  rcCs |ddSNr?r")rsplitrFrrrrAsrAcCs |ddSr)splitrrrrrsrcCs*ddlm}d|d||dS)Nr)sha1Zrecycle_rzutf-8)ZhashlibrencodeZ hexdigest)Z plugin_actionrrrrget_default_recycle_pools r)r r!r")N)F)rrrrrrrr<rIrSrVrqrrrrrrrrrrrrrArrrrrr s8 $!# !& ?