Mercurial > repos > bgruening > augustus

annotate extract_features.py @ 10:cb47e789ccaa draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
author bgruening
date Fri, 10 May 2019 08:25:57 -0400
parents bcfe8e0731f8
children 66c8e9d8d1c4
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
1 #!/usr/bin/env python
3c5116448979 Uploaded
bgruening
parents:
diff changeset
2
10
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
3 import argparse
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
4 import sys
3c5116448979 Uploaded
bgruening
parents:
diff changeset
5 import textwrap
3c5116448979 Uploaded
bgruening
parents:
diff changeset
6
10
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
7
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
8 def main( args ):
3c5116448979 Uploaded
bgruening
parents:
diff changeset
9 """
3c5116448979 Uploaded
bgruening
parents:
diff changeset
10 Extract the protein and coding section from an augustus gff, gtf file
3c5116448979 Uploaded
bgruening
parents:
diff changeset
11 Example file:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
12 HS04636 AUGUSTUS stop_codon 6901 6903 . + 0 Parent=g1.t1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
13 HS04636 AUGUSTUS transcription_end_site 8857 8857 . + . Parent=g1.t1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
14 # protein sequence = [MLARALLLCAVLALSHTANPCCSHPCQNRGVCMSVGFDQYKCDCTRTGFYGENCSTPEFLTRIKLFLKPTPNTVHYIL
3c5116448979 Uploaded
bgruening
parents:
diff changeset
15 # THFKGFWNVVNNIPFLRNAIMSYVLTSRSHLIDSPPTYNADYGYKSWEAFSNLSYYTRALPPVPDDCPTPLGVKGKKQLPDSNEIVEKLLLRRKFIPD
3c5116448979 Uploaded
bgruening
parents:
diff changeset
16 # PQGSNMMFAFFAQHFTHQFFKTDHKRGPAFTNGLGHGVDLNHIYGETLARQRKLRLFKDGKMKYQIIDGEMYPPTVKDTQAEMIYPPQVPEHLRFAVG
3c5116448979 Uploaded
bgruening
parents:
diff changeset
17 # QEVFGLVPGLMMYATIWLREHNRVCDVLKQEHPEWGDEQLFQTSRLILIGETIKIVIEDYVQHLSGYHFKLKFDPELLFNKQFQYQNRIAAEFNTLYH
3c5116448979 Uploaded
bgruening
parents:
diff changeset
18 # WHPLLPDTFQIHDQKYNYQQFIYNNSILLEHGITQFVESFTRQIAGRVAGGRNVPPAVQKVSQASIDQSRQMKYQSFNEYRKRFMLKPYESFEELTGE
3c5116448979 Uploaded
bgruening
parents:
diff changeset
19 # KEMSAELEALYGDIDAVELYPALLVEKPRPDAIFGETMVEVGAPFSLKGLMGNVICSPAYWKPSTFGGEVGFQIINTASIQSLICNNVKGCPFTSFSV
3c5116448979 Uploaded
bgruening
parents:
diff changeset
20 # PDPELIKTVTINASSSRSGLDDINPTVLLKERSTEL]
3c5116448979 Uploaded
bgruening
parents:
diff changeset
21 # end gene g1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
22 ###
3c5116448979 Uploaded
bgruening
parents:
diff changeset
23 #
3c5116448979 Uploaded
bgruening
parents:
diff changeset
24 # ----- prediction on sequence number 2 (length = 2344, name = HS08198) -----
3c5116448979 Uploaded
bgruening
parents:
diff changeset
25 #
3c5116448979 Uploaded
bgruening
parents:
diff changeset
26 # Predicted genes for sequence number 2 on both strands
3c5116448979 Uploaded
bgruening
parents:
diff changeset
27 # start gene g2
3c5116448979 Uploaded
bgruening
parents:
diff changeset
28 HS08198 AUGUSTUS gene 86 2344 1 + . ID=g2
3c5116448979 Uploaded
bgruening
parents:
diff changeset
29 HS08198 AUGUSTUS transcript 86 2344 . + . ID=g2.t1;Parent=g2
3c5116448979 Uploaded
bgruening
parents:
diff changeset
30 HS08198 AUGUSTUS transcription_start_site 86 86 . + . Parent=g2.t1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
31 HS08198 AUGUSTUS exon 86 582 . + . Parent=g2.t1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
32 HS08198 AUGUSTUS start_codon 445 447 . + 0 Parent=g2.t1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
33 """
3c5116448979 Uploaded
bgruening
parents:
diff changeset
34 protein_seq = ''
3c5116448979 Uploaded
bgruening
parents:
diff changeset
35 coding_seq = ''
3c5116448979 Uploaded
bgruening
parents:
diff changeset
36 if args.protein:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
37 po = open( args.protein, 'w+' )
3c5116448979 Uploaded
bgruening
parents:
diff changeset
38 if args.codingseq:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
39 co = open( args.codingseq, 'w+' )
3c5116448979 Uploaded
bgruening
parents:
diff changeset
40
3c5116448979 Uploaded
bgruening
parents:
diff changeset
41 for line in sys.stdin:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
42 # protein- and coding-sequence are stored as comments
3c5116448979 Uploaded
bgruening
parents:
diff changeset
43 if line.startswith('#'):
3c5116448979 Uploaded
bgruening
parents:
diff changeset
44 line = line[2:].strip()
3c5116448979 Uploaded
bgruening
parents:
diff changeset
45 if line.startswith('start gene'):
3c5116448979 Uploaded
bgruening
parents:
diff changeset
46 gene_name = line[11:].strip()
3c5116448979 Uploaded
bgruening
parents:
diff changeset
47
10
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
48 if protein_seq:
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
49 if line.endswith(']'):
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
50 protein_seq += line[:-1]
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
51 po.write( '>%s\n%s\n' % (gene_name, '\n'.join( textwrap.wrap( protein_seq, 80 ) ) ) )
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
52 protein_seq = ''
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
53 else:
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
54 protein_seq += line
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
55
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
56 if coding_seq:
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
57 if line.endswith(']'):
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
58 coding_seq += line[:-1]
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
59 co.write( '>%s\n%s\n' % (gene_name, '\n'.join( textwrap.wrap( coding_seq, 80 ) ) ) )
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
60 coding_seq = ''
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
61 else:
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
62 coding_seq += line
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
63
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
64 if args.protein and line.startswith('protein sequence = ['):
3c5116448979 Uploaded
bgruening
parents:
diff changeset
65 if line.endswith(']'):
5
bcfe8e0731f8 Uploaded
bgruening
parents: 3
diff changeset
66 protein_seq = line[20:-1]
bcfe8e0731f8 Uploaded
bgruening
parents: 3
diff changeset
67 po.write( '>%s\n%s\n' % (gene_name, '\n'.join( textwrap.wrap( protein_seq, 80 ) ) ) )
bcfe8e0731f8 Uploaded
bgruening
parents: 3
diff changeset
68 protein_seq = ''
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
69 else:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
70 line = line[20:]
3c5116448979 Uploaded
bgruening
parents:
diff changeset
71 protein_seq = line
3c5116448979 Uploaded
bgruening
parents:
diff changeset
72
3c5116448979 Uploaded
bgruening
parents:
diff changeset
73 if args.codingseq and line.startswith('coding sequence = ['):
3c5116448979 Uploaded
bgruening
parents:
diff changeset
74 if line.endswith(']'):
5
bcfe8e0731f8 Uploaded
bgruening
parents: 3
diff changeset
75 coding_seq = line[19:-1]
bcfe8e0731f8 Uploaded
bgruening
parents: 3
diff changeset
76 co.write( '>%s\n%s\n' % (gene_name, '\n'.join( textwrap.wrap( coding_seq, 80 ) ) ) )
bcfe8e0731f8 Uploaded
bgruening
parents: 3
diff changeset
77 coding_seq = ''
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
78 else:
3
702d9e042295 Uploaded
bgruening
parents: 1
diff changeset
79 line = line[19:]
702d9e042295 Uploaded
bgruening
parents: 1
diff changeset
80 coding_seq = line
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
81
3c5116448979 Uploaded
bgruening
parents:
diff changeset
82 if args.codingseq:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
83 co.close()
3c5116448979 Uploaded
bgruening
parents:
diff changeset
84 if args.protein:
3c5116448979 Uploaded
bgruening
parents:
diff changeset
85 po.close()
3c5116448979 Uploaded
bgruening
parents:
diff changeset
86
10
cb47e789ccaa planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/augustus commit 4a8f640dec78899be470ddd1c436fc0d158c80f3
bgruening
parents: 5
diff changeset
87
1
3c5116448979 Uploaded
bgruening
parents:
diff changeset
88 if __name__ == '__main__':
3c5116448979 Uploaded
bgruening
parents:
diff changeset
89 parser = argparse.ArgumentParser()
3c5116448979 Uploaded
bgruening
parents:
diff changeset
90 parser.add_argument('-p', '--protein', help='Path to the protein file.')
3c5116448979 Uploaded
bgruening
parents:
diff changeset
91 parser.add_argument('-c', '--codingseq', help='Path to the coding file.')
3c5116448979 Uploaded
bgruening
parents:
diff changeset
92
3c5116448979 Uploaded
bgruening
parents:
diff changeset
93 args = parser.parse_args()
3c5116448979 Uploaded
bgruening
parents:
diff changeset
94 main( args )