Mercurial > repos > eduardo > annotateviz

annotate annotateviz.py @ 3:3302f18f9e16 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/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
author eduardo
date Sat, 17 Jun 2017 15:10:53 -0400
parents 7537482eed36
children 92921dfea0b5
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
1 import sys
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
2 import os
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
3 import gffutils
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
4 import argparse
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
5 import json
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
6
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
7 def add_matches(matches,array):
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
8 for m in matches:
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
9 trans = {}
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
10 trans['mRNA'] = '.'.join(m.id.split('.',2)[:2]) #hack to recover mRNA from PFAM gff which doesn't include a query field
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
11 trans['Contig'] = m.seqid
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
12 trans['Source'] = m.source
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
13 trans['Score'] = m.score
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
14 trans['Prediction']=m["Name"]
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
15 array.append(trans)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
16 def main(argv, wayout):
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
17 if not len(argv):
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
18 argv.append("-h")
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
19 parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter, description=__doc__)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
20 parser.add_argument('-d','--database', help="gffutils sqlite database")
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
21 parser.add_argument('-j','--json', help="report in json format")
3
3302f18f9e16 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents: 0
diff changeset
22 args = parser.parse_args(argv)
0
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
23 gffutils.constants.always_return_list = False
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
24 db = gffutils.interface.FeatureDB(args.database)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
25 prediction_list = []
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
26 add_matches(db.features_of_type("protein_match"), prediction_list)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
27 add_matches(db.features_of_type("signalpep"), prediction_list)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
28 add_matches(db.features_of_type("trans_helix"), prediction_list)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
29 add_matches(db.features_of_type("PFAM"), prediction_list)
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
30 fout=open(args.json, 'w')
7537482eed36 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents:
diff changeset
31 json.dump(prediction_list,fout)
3
3302f18f9e16 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents: 0
diff changeset
32 if __name__ == "__main__":
3302f18f9e16 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/annotateviz commit 9bbaa3eacc76ff3bf2b6da313cc0d85705f15dd1-dirty
eduardo
parents: 0
diff changeset
33 main(sys.argv[1:],sys.stdout)