|
17 | 17 |
|
18 | 18 | from Bio import SeqIO |
19 | 19 |
|
20 | | -parser = ArgumentParser(description='Simple script to reverse complement a FASTA file') |
21 | | - |
22 | | -parser.add_argument('-i', '--input', help='FASTA file to reverse complement') |
23 | | -parser.add_argument('-o', '--output', help='FASTA file to write to') |
24 | | - |
25 | | -args = parser.parse_args() |
26 | | - |
27 | | -if args.input is None or args.output is None: |
28 | | - raise Exception('Both -i and -o need to be set to run this') |
29 | | - |
30 | 20 | def reverse_complement(seq): |
31 | 21 | revcomp = seq.reverse_complement() |
32 | 22 | revcomp.id = seq.id |
33 | 23 | revcomp.description = seq.description |
34 | 24 |
|
35 | 25 | return revcomp |
36 | 26 |
|
37 | | -seqs = SeqIO.parse(args.input, 'fasta') |
38 | | -seqs = map(reverse_complement, seqs) |
| 27 | +def main(): |
| 28 | + parser = ArgumentParser(description='Simple script to reverse complement a FASTA file') |
| 29 | + |
| 30 | + parser.add_argument('-i', '--input', help='FASTA file to reverse complement') |
| 31 | + parser.add_argument('-o', '--output', help='FASTA file to write to') |
| 32 | + |
| 33 | + args = parser.parse_args() |
| 34 | + |
| 35 | + if args.input is None or args.output is None: |
| 36 | + raise Exception('Both -i and -o need to be set to run this') |
| 37 | + |
| 38 | + seqs = SeqIO.parse(args.input, 'fasta') |
| 39 | + seqs = map(reverse_complement, seqs) |
39 | 40 |
|
40 | | -SeqIO.write(seqs, args.output, 'fasta') |
| 41 | + SeqIO.write(seqs, args.output, 'fasta') |
0 commit comments