Mercurial > repos > dfornika > screen_abricate_report

annotate screen_abricate_report.py @ 0:2ffd23634c1e draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
author dfornika
date Thu, 26 Sep 2019 19:37:27 -0400
parents
children 40003338a8e8
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
1 #!/usr/bin/env python
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
2
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
3 from __future__ import print_function
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
4
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
5 import argparse
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
6 import os
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
7 import re
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
8 import sys
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
9 import csv
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
10 from pprint import pprint
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
11
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
12 def parse_screen_file(screen_file):
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
13 screen = []
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
14 with open(screen_file) as f:
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
15 reader = csv.DictReader(f, delimiter="\t", quotechar='"')
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
16 for row in reader:
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
17 screen.append(row)
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
18 return screen
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
19
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
20 def get_abricate_report_fieldnames(abricate_report):
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
21 with open(abricate_report) as f:
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
22 reader = csv.DictReader(f, delimiter="\t", quotechar='"')
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
23 row = next(reader)
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
24 fieldnames = row.keys()
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
25 return fieldnames
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
26
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
27 def main(args):
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
28 screen = parse_screen_file(args.screen)
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
29 abricate_report_fieldnames = get_abricate_report_fieldnames(args.abricate_report)
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
30 with open(args.abricate_report) as f:
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
31 reader = csv.DictReader(f, delimiter="\t", quotechar='"')
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
32 writer = csv.DictWriter(sys.stdout, delimiter="\t", quotechar='"', fieldnames=abricate_report_fieldnames)
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
33 writer.writeheader()
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
34 for row in reader:
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
35 for gene in screen:
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
36 if re.search(gene['regex'], row['GENE']):
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
37 writer.writerow(row)
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
38
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
39
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
40
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
41 if __name__ == '__main__':
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
42 parser = argparse.ArgumentParser()
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
43 parser.add_argument("--screen", help="TSV file defining genes to screen for")
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
44 parser.add_argument("abricate_report", help="Abricate output")
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
45 args = parser.parse_args()
2ffd23634c1e planemo upload for repository https://github.com/dfornika/galaxytools/blob/master/tools/screen_abricate_report commit d9732cd3279d03dcc498bf2eb903f9e6120a9d85-dirty
dfornika
parents:
diff changeset
46 main(args)