diff extract_features.py @ 5:bcfe8e0731f8 draft

Uploaded

--- a/extract_features.py	Fri Jun 07 02:30:21 2013 -0400
+++ b/extract_features.py	Sun Jun 09 09:53:26 2013 -0400
@@ -47,16 +47,18 @@
 
             if args.protein and line.startswith('protein sequence = ['):
                 if line.endswith(']'):
-                    line = line[20:-1]
-                    protein_seq = line
+                    protein_seq = line[20:-1]
+                    po.write( '>%s\n%s\n' % (gene_name, '\n'.join( textwrap.wrap( protein_seq, 80 ) ) ) )
+                    protein_seq = ''
                 else:
                     line = line[20:]
                     protein_seq = line
 
             if args.codingseq and line.startswith('coding sequence = ['):
                 if line.endswith(']'):
-                    line = line[19:-1]
-                    coding_seq = line
+                    coding_seq = line[19:-1]
+                    co.write( '>%s\n%s\n' % (gene_name, '\n'.join( textwrap.wrap( coding_seq, 80 ) ) ) )
+                    coding_seq = ''
                 else:
                     line = line[19:]
                     coding_seq = line