annotate fasta_remove_id.py @ 0:2b42545705fa draft

Uploaded Python Script
author curtisross
date Thu, 23 Sep 2021 16:24:09 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
1 #!/usr/bin/env python
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
2 import sys
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
3 import argparse
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
4 import logging
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
5 from Bio import SeqIO
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
6
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
7 logging.basicConfig(level=logging.INFO)
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
8 log = logging.getLogger()
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
9
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
10
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
11 def drop_id(fasta_file=None):
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
12 for rec in SeqIO.parse(fasta_file, "fasta"):
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
13 rec.description = ""
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
14 ind = str(rec.seq).find("##")
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
15 if (
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
16 ind != -1
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
17 ): # This method causes mid-file comments (such as from Apollo sequences) to be appended to the end of the previous sequence
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
18 rec.seq = rec.seq[0:ind]
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
19 yield rec
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
20
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
21
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
22 if __name__ == "__main__":
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
23 parser = argparse.ArgumentParser(description="Identify shine-dalgarno sequences")
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
24 parser.add_argument("fasta_file", type=argparse.FileType("r"), help="Genbank file")
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
25
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
26 args = parser.parse_args()
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
27 for rec in drop_id(**vars(args)):
2b42545705fa Uploaded Python Script
curtisross
parents:
diff changeset
28 SeqIO.write([rec], sys.stdout, "fasta")