Mercurial > repos > iuc > concoct

annotate extract_fasta_bins.py @ 2:bd409bbd287f 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/concoct commit 77dea70f156d56c83b6851b61dc997d2b344bdc9
author iuc
date Fri, 01 Jul 2022 14:10:59 +0000
parents bf23c688912a
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
1 #!/usr/bin/env python
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
2
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
3 import argparse
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
4 import gzip
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
5 import os
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
6 import sys
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
7 from collections import defaultdict
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
8 from functools import partial
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
9
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
10 import pandas as pd
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
11 from Bio import SeqIO
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
12
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
13 parser = argparse.ArgumentParser(description=__doc__)
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
14
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
15 parser.add_argument('--gzipped', action='store_true', dest='gzipped', help='Input files are gzipped')
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
16 parser.add_argument("--input_fasta", action="store", dest="input_fasta", help="Input Fasta file")
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
17 parser.add_argument("--input_cluster", action="store", dest="input_cluster", help="Concoct output cluster file")
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
18 parser.add_argument("--output_path", help="Output directory")
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
19
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
20 args = parser.parse_args()
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
21
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
22 all_seqs = {}
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
23 if args.gzipped:
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
24 _open = partial(gzip.open, mode='rt')
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
25 else:
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
26 _open = open
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
27
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
28 with _open(args.input_fasta) as fh:
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
29 for seq in SeqIO.parse(fh, "fasta"):
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
30 all_seqs[seq.id] = seq
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
31
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
32 # Make sure we're reading the file as tabular!
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
33 df = pd.read_csv(args.input_cluster, sep='\t')
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
34 try:
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
35 assert df.columns[0] == 'contig_id'
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
36 assert df.columns[1] == 'cluster_id'
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
37 except AssertionError:
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
38 sys.stderr.write("ERROR! Header line was not 'contig_id, cluster_id', please adjust your input file. Exiting!\n")
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
39 sys.exit(-1)
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
40
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
41 cluster_to_contigs = defaultdict(list)
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
42 for i, row in df.iterrows():
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
43 cluster_to_contigs[row['cluster_id']].append(row['contig_id'])
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
44
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
45 for cluster_id, contig_ids in cluster_to_contigs.items():
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
46 output_file = os.path.join(args.output_path, "{0}.fa".format(cluster_id))
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
47 seqs = [all_seqs[contig_id] for contig_id in contig_ids]
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
48 with open(output_file, 'w') as ofh:
bf23c688912a "planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/concoct commit 49b42f61ff37c3c33dd15c195e5705e1db066c37"
iuc
parents:
diff changeset
49 SeqIO.write(seqs, ofh, 'fasta')