U adF @s<ddlmZddlZddlZddlZddlZzddlmZWn ek r\ddl mZYnXz ddl Z Wn ek rddl m Z YnXej ddkrddl Z ddlmZddlmZeejZdd Zd d Zd d ZddZd&ddZddZddZddZddZddZddZd'd d!Zd"d#Z d$d%Z!dS)()with_statementN) OrderedDict)bz2)qconfig) get_loggercCsd}tj|\}}t|tjs6tjd|ddd|dkrPtj|dd}n@|dkrrt j |d d}t |}n|d krPzt j |d d}Wn8tk rt\}}}tjd t|dd YnX|}t|d krtjd|dd t|dkrtd|d z||d }t |}Wn$tk rLtjddd YnXn@z t|}Wn2tk rt\}}}tj|ddYnX|S)NzPermission denied accessing T)Z to_stderrexit_with_code)z.gzz.gzipZrt)mode)z.bz2z.bzip2r)z.zipzCan't open zip file: )r rz Reading %s: zip archive is emptyz3Zip archive must contain exactly one file. Using %sz=Use python 2.6 or newer to work with contigs directly in zip.)Z exit_code)ospathsplitextaccessR_OKloggererrorgzipopenrZBZ2File_read_compressed_filezipfileZZipFile Exceptionsysexc_infostrZnamelistlenZwarningAttributeErrorIOErrorZ exception)fpath fasta_file_ZextZzfileexc_type exc_valuenamesr%5lib/python3.8/site-packages/quast_libs/fastaparser.py_get_fasta_file_handler!s>     r'cCs&tjddkr"tt|S|S)Nrr)r version_infoio TextIOWrapperBytesIOread)Zcompressed_filer%r%r&rMsrcCs2z|dddWStk r,YdSXdS)zk Extracts name from fasta entry line: ">chr1 length=100500; coverage=15;" ---> "chr1" rNr)split IndexError)liner%r%r&__get_entry_nameSsr1cCst}d}d}t|}|D]j}|ddkr8|d}n|g}|D]@}|sLqB|ddkrr|rh|||<d}t|}qB|t|7}qBq|||<||S)ze Takes filename of FASTA-file Returns list of lengths of sequences in FASTA-file rN >)rr'findr.r1rstripclose)rZ chr_lengthslchr_namer raw_linelinesr0r%r%r&get_chr_lengths_from_fastafile^s(   r<Fc Cszd}i}i}t|D]Z\}}|d}t|}||7}tddt|D||<|rf|t||8}|||<q|||fS)Nrcss"|]\}}|dkr|dVqdS)NrNr%).0xsr%r%r& sz#get_genome_stats..) read_fastar.rset enumerate) fasta_fpathZskip_nsZ genome_sizeZreference_chromosomesZns_by_chromosomesnameseqr9Zchr_lenr%r%r&get_genome_stats|s  rHc CsHd}d}d}d}|d}g}t|}|D]}|ddkrH|d} n|g} | D]} | s\qR| ddkr|r||||t| t| g||7}d}d}t| }|t| 7}qR|s| } |t| 7}|t| 7}qRq*W5QRX||||t| t| gt|d0} |D]$} | ddd | Dd qW5QRXdS) Nrz.fair2r3r4w cSsg|] }t|qSr%)r)r>Zfsr%r%r& sz#create_fai_file.. ) rr5r.appendrr6r1writejoin)rEr8Z total_offsetZ chr_offsetr9Z fai_fpathZ fai_fieldsZin_fr:r;r0Zchr_lineZout_fZfieldsr%r%r&create_fai_files<      rPcCsztj|st|d}t|D]F}|ddkrZ|r>|ttj|t|dd}|r"||q"|rv|dS)z 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)s.fasta" original.fasta Nrr4z.farI) r risdirmkdirrr7rOr1rN)rZoutput_dirpathZoutFiler0r%r%r& split_fastas     rSccsd}g}d}t|}|D]^}|d}|D]J}|s4q*|ddkrf|sT|d|fVd}t|}g}q*||q*q|s|r|d|fV|dS)zH Generator that returns FASTA entries in tuples (name, seq) Tr-r2rr4FN)r'r.rOr1rMr6r7)rfirstrGrFr r:r;r0r%r%r&rBs&  rBcCs(g}t|D]\}}|||fq |S)zB Returns list of FASTA entries (in tuples: name, seq) )rBrM)rlist_seqrFrGr%r%r&read_fasta_one_timesrVcCs`t|}g}|D]8}|d}|D]$}|s,q"|ddkr"||q"q|d|}|S)z Returns string r2rr4r-)r'r.rMr6r7rO)rr rUr:r;r0Z fasta_strr%r%r&read_fasta_strs   rWcCsH|D]>\}}td|tdt|dD]}t|||dq(qdS)Nz>%sr<)printranger)fastarFrGir%r%r& print_fastas  r]rIcCsbt||}|D]F\}}|d|tdt|dD]}||||ddq4q|dS)Nz>%s rrXrL)rrNrZrr7)rr[r ZoutfilerFrGr\r%r%r& write_fastas   r^cCsdddddd|S)NTAGCr=)r`r_rbrar=)upper)Zletterr%r%r&compsrdcs*ttdddfddt|DS)NZ ATCGNatcgnZ TAGCNtagcnr-c3s|]}|dVqdS)r-N)get)r>Z nucleotidecr%r&rAszrev_comp..)dictziprOreversed)rGr%rfr&rev_compsrk)F)rI)"Z __future__rr rrr collectionsr ImportErrorZ%quast_libs.site_packages.ordered_dictrZquast_libs.site_packagesr(r)Z quast_libsrZquast_libs.logrZLOGGER_DEFAULT_NAMErr'rr1r<rHrPrSrBrVrWr]r^rdrkr%r%r%r&s>     ,  #