annotate rest_tool_search.py @ 8:3c1e862e8cd6 draft

Uploaded
author bernhardlutz
date Fri, 02 May 2014 05:40:28 -0400
parents 9ee84d9fd3a7
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
1 #!/usr/bin/env python
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
2
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
3 import sys, os
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
4 import argparse
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
5 import readfile
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
6
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
7 def main(args):
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
8 #search for acitivity or target
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
9 url="http://pubchem.ncbi.nlm.nih.gov/rest/pug/"+args.type+"/name/"+args.name
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
10 if args.type == "assay":
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
11 url+="/aids"
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
12 elif args.type == "compound":
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
13 url+="/cids"
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
14 else:
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
15 url+="/sids"
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
16 url+="/txt"
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
17 #print("url: "+url)
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
18 data=readfile.getresult(url)
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
19 args.outfile.write(data)
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
20 args.outfile.close()
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
21
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
22
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
23 if __name__ == "__main__":
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
24 parser = argparse.ArgumentParser()
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
25 parser.add_argument('--name', type=str, required=True,
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
26 help="Enter the name")
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
27 parser.add_argument('--type', type=str, required=True,
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
28 help="What you want to search for")
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
29 parser.add_argument('--outfile', type=argparse.FileType('w'), required=True,
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
30 help="Specify output file")
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
31 if len(sys.argv) < 2:
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
32 print "Too few arguments..."
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
33 parser.print_help()
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
34 exit(1)
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
35 args = parser.parse_args()
9ee84d9fd3a7 Uploaded
bernhardlutz
parents:
diff changeset
36 main( args )