3 7] @s>ddlmZGdddZGdddeZGdddeZdS) )xopenc@s2eZdZedfddZddZddZdd ZdS) FileWriterNcCs2||_t|tr$||d|_d|_n t||_dS)NwbT)_file isinstancestr_close_on_exitbool)selffileopener _close_filer]/home/psgendb/BIRCHDEV/lib-linux-x86_64/python/lib64/python3.6/site-packages/dnaio/writers.py__init__s   zFileWriter.__init__cCs|jr|jjdS)N)rrclose)r rrrr szFileWriter.closecCs|jjrtd|S)NzI/O operation on closed file)rclosed ValueError)r rrr __enter__szFileWriter.__enter__cGs |jdS)N)r)r argsrrr__exit__szFileWriter.__exit__)__name__ __module__ __qualname__rrrrrrrrrrsrcs2eZdZdZdedffdd ZdddZZS) FastaWriterz4 Write FASTA-formatted sequences to a file. Ncs(tj|||d|dkr|nd|_dS)zm If line_length is not None, the lines will be wrapped after line_length characters. )r r rN)superr line_length)r r rr r ) __class__rrrszFastaWriter.__init__cCs|dkr|j}|j}n|}|jdk r|jjd|djdg}x6tdt||jD] }|j||||jdqVW|jjdj |jdn&d|d|d}|jj|jddS)aWrite an entry to the the FASTA file. If only one parameter (name_or_record) is given, it must have attributes .name and .sequence, which are then used. Otherwise, the first parameter must be the name and the second the sequence. The effect is that you can write this: writer.write("name", "ACCAT") or writer.write(Sequence("name", "ACCAT")) N> asciir) namesequencerrwriteencoderangelenappendjoin)r Zname_or_recordr#r"sirrrr$'s   zFastaWriter.write)N)rrr__doc__rrr$ __classcell__rr)rrrsrcs<eZdZdZdZdedffdd ZddZd d ZZ S) FastqWriterz Write sequences with qualities in FASTQ format. FASTQ files are formatted like this: @read name SEQUENCE + QUALITIS rFNcstj|||d||_dS)N)r r )rr _two_headers)r r Z two_headersr r )rrrrQszFastqWriter.__init__cCsP|jr |jnd}d|jd|jd|d|jd}|jj|jddS)z Write a Sequence record to the FASTQ file. The record object must have attributes .name, .sequence and .qualities. r!@rz +r N)r/r"r# qualitiesrr$r%)r recordZname2r*rrrr$Us*zFastqWriter.writecCs |jjdj|||jddS)Nz@{0:s} {1:s} + {2:s} r )rr$formatr%)r r"r#r1rrrwriteseq`s zFastqWriter.writeseq) rrrr,Z file_moderrr$r4r-rr)rrr.Es   r.N)rrrr.rrrrs +