annotate cpt_fasta_remove_id/fasta_remove_id.py @ 1:582e45ac71d6 draft default tip

Uploaded
author cpt_testbed
date Fri, 06 May 2022 08:10:16 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
1 #!/usr/bin/env python
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
2 import sys
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
3 import argparse
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
4 import logging
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
5 from Bio import SeqIO
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
6
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
7 logging.basicConfig(level=logging.INFO)
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
8 log = logging.getLogger()
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
9
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
10
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
11 def drop_id(fasta_file=None):
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
12 for rec in SeqIO.parse(fasta_file, "fasta"):
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
13 rec.description = ""
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
14 ind = str(rec.seq).find("##")
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
15 if (
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
16 ind != -1
582e45ac71d6 Uploaded
cpt_testbed
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
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
18 rec.seq = rec.seq[0:ind]
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
19 yield rec
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
20
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
21
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
22 if __name__ == "__main__":
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
23 parser = argparse.ArgumentParser(description="Identify shine-dalgarno sequences")
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
24 parser.add_argument("fasta_file", type=argparse.FileType("r"), help="Genbank file")
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
25
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
26 args = parser.parse_args()
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
27 for rec in drop_id(**vars(args)):
582e45ac71d6 Uploaded
cpt_testbed
parents:
diff changeset
28 SeqIO.write([rec], sys.stdout, "fasta")