ó ›¢yZc@€s/ddlmZddlZddlZddlZddlZyddlZWn!ek rsddlmZnXej ddkr–ddl Z nddl m Z ddl mZee jƒZd„Zd „Zd „Zd „Zd „Zd „Zd„Zd„Zdd„Zd„Zd„ZdS(iÿÿÿÿ(twith_statementN(tbz2ii(tqconfig(t get_loggercC€sd}tjj|ƒ\}}tj|tjƒsStjd|dtddƒn|dkrwt j |ddƒ}n–|dkr§t j |dd ƒ}t |ƒ}nf|dkrÁytj|dd ƒ}WnCtk rtjƒ\}}}tjd t|ƒddƒq X|jƒ}t|ƒdkrJtjd|ddƒnt|ƒdkrttjd|dƒny#|j |dƒ}t |ƒ}Wq tk r½tjdddƒq XnLyt |ƒ}Wn9tk r tjƒ\}}}tj|ddƒnX|S(NsPermission denied accessing t to_stderrtexit_with_codeis.gzs.gziptmodetrts.bz2s.bzip2trs.zipsCan't open zip file: is Reading %s: zip archive is emptys3Zip archive must contain exactly one file. Using %ss=Use python 2.6 or newer to work with contigs directly in zip.it exit_code(s.gzs.gzip(s.bz2s.bzip2(s.zip(tNonetostpathtsplitexttaccesstR_OKtloggerterrortTruetgziptopenRtBZ2Filet_read_compressed_filetzipfiletZipFilet Exceptiontsystexc_infotstrtnamelisttlentwarningtAttributeErrortIOErrort exception(tfpatht fasta_filet_texttzfiletexc_typet exc_valuetnames((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyt_get_fasta_file_handlers>     !   cC€s3tjddkr/tjtj|jƒƒƒS|S(Nii(Rt version_infotiot TextIOWrappertBytesIOtread(tcompressed_file((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyRGscC€sétƒ}d}d}t|ƒ}x­|D]¥}|jdƒdkrU|jdƒ}n |g}xl|D]d}|swqen|ddkr³|r |||iN(tdictR R+tfindtsplittstripRtclose(R#t chr_lengthstltchr_nameR$traw_linetlinestline((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pytget_chr_lengths_from_fastafileMs(         c C€sÞd}d}d}d}|d}g}t|ƒ}x|D]}|jdƒdkrk|jdƒ} n |g} xÏ| D]Ç} | sq{n| ddkr |rê|j|||t| jƒƒt| ƒgƒ||7}d}d}n| djƒ}|t| ƒ7}q{|s| } n|t| jƒƒ7}|t| ƒ7}q{Wq>WWdQX|j|||t| jƒƒt| ƒgƒt|dƒK} xA|D]9} | jdjg| D]}t |ƒ^q°ƒd ƒq—WWdQXdS( Nis.fais iÿÿÿÿR2itws s ( R RR4R5tappendRR6twritetjoinR(t fasta_fpathR9t total_offsett chr_offsetR:t fai_fpatht fai_fieldstin_fR;R<R=tchr_linetout_ftfieldstfs((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pytcreate_fai_fileks<    .   . cC€s¾tjj|ƒs"tj|ƒnd}x|t|ƒD]n}|ddkr|r^|jƒnttjj||djƒdƒdƒ}n|r5|j |ƒq5q5W|rº|jƒndS(sð Takes filename of FASTA-file and directory to output Creates separate FASTA-files for each sequence in FASTA-file Returns nothing Oops, similar to: pyfasta split --header "%(seqid).fasta" original.fasta iR2is.faR?N( R R tisdirtmkdirR RR7RBR6RA(R#toutput_dirpathtoutFileR=((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyt split_fastaŽs /cc€sët}g}d}t|ƒ}x™|D]‘}|jdƒ}xy|D]q}|sSqAn|ddkrŸ|s€|dj|ƒfVnt}|jƒd}g}qA|j|jƒƒqAWq%W|sÆ|rÝ|dj|ƒfVn|jƒdS(sB Returns list of FASTA entries (in tuples: name, seq) ts iR2iN(RR+R5RBtFalseR6R@R7(R#tfirsttseqtnameR$R;R<R=((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyt read_fasta£s&     c C€st}g}d}t|ƒ}g}x¡|D]™}|jdƒ}x|D]y}|sYqGn|ddkr­|sŽ|j|dj|ƒfƒnt}|jƒd}g}qG|j|jƒƒqGWq+W|sÔ|ró|j|dj|ƒfƒn|jƒ|S(sB Returns list of FASTA entries (in tuples: name, seq) RSs iR2i(RR+R5R@RBRTR6R7( R#RURVRWR$tlist_seqR;R<R=((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pytread_fasta_one_timeÂs*      cC€sWxP|D]H\}}d|GHx0tdt|ƒdƒD]}|||d!GHq5WqWdS(Ns>%sii<(trangeR(tfastaRWRVti((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyt print_fastaãs R?cC€s„t||ƒ}xd|D]\\}}|jd|ƒx<tdt|ƒdƒD]"}|j|||d!dƒqLWqW|jƒdS(Ns>%s ii<s (RRAR[RR7(R#R\RtoutfileRWRVR]((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyt write_fastaês $cC€s1idd6dd6dd6dd6dd6|jƒS(NtTtAtGtCtN(tupper(tletter((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pytcompôsc€s8ttddƒƒ‰dj‡fd†t|ƒDƒƒS(Nt ATCGNatcgnt TAGCNtagcnRSc3€s!|]}ˆj|dƒVqdS(RSN(tget(t.0t nucleotide(tc(sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pys ús(R3tzipRBtreversed(RV((RnsE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pytrev_compøs(t __future__RR RRRRt ImportErrortquast_libs.site_packagesR,R-t quast_libsRtquast_libs.logRtLOGGER_DEFAULT_NAMERR+RR>RMRRRXRZR^R`RhRq(((sE/home/birch/BIRCH/local/install/quast-4.6.3/quast_libs/fastaparser.pyts0      ,   #   !