annotate cummerbund_wrapper.py @ 12:e132e60f95b1

Fix cuffdata datatype init
author Jim Johnson <jj@umn.edu>
date Sun, 21 Oct 2012 18:17:51 -0500
parents b0d11fcbc3ac
children 2d7eee38ab5b
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
1 #!/usr/bin/env python
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
2
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
3 ### Runs "r_script" and generates a HTML report
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
4 ### Inspired on cuffdiff_wrapper.py and gatk_wrapper.py
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
5 ### Carlos Borroto <carlos.borroto@gmail.com>
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
6
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
7 import optparse, os, shutil, subprocess, sys, tempfile
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
8
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
9 def stop_err( msg ):
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
10 sys.stderr.write( "%s\n" % msg )
8
b0d11fcbc3ac cummerbund add MDS and PCA plots, handle errors from R
Jim Johnson <jj@umn.edu>
parents: 1
diff changeset
11 sys.exit(1)
0
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
12
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
13 def html_report_from_directory( html_out, dir ):
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
14 html_out.write( '<html>\n<head>\n<title>Galaxy - cummeRbund Output</title>\n</head>\n<body>\n<p/>\n<ul>\n' )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
15 for fname in sorted( os.listdir( dir ) ):
1
6a9bc26ab8d9 Added cuffdata and cuffdatadb datatypes, cuffdiff multiselect outputs
Jim Johnson <jj@umn.edu>
parents: 0
diff changeset
16 # html_out.write( '<li><a href="%s">%s</a></li>\n' % ( fname, fname ) )
8
b0d11fcbc3ac cummerbund add MDS and PCA plots, handle errors from R
Jim Johnson <jj@umn.edu>
parents: 1
diff changeset
17 html_out.write( '<li><a href="%s"><img src="%s" alt="" height="80" width="80">%s</a></li>\n' % ( fname, fname , fname ) )
0
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
18 html_out.write( '</ul>\n</body>\n</html>\n' )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
19
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
20 def __main__():
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
21 #Parse Command Line
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
22 parser = optparse.OptionParser()
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
23
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
24 # wrapper options
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
25 parser.add_option('', '--r-script', dest='r_script', help='R script')
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
26 parser.add_option('', '--html-report-from-directory', dest='html_report_from_directory', type="string", nargs=2, help='"Target HTML File" "Directory"')
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
27
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
28 (options, args) = parser.parse_args()
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
29
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
30 (html_filename, html_dir) = options.html_report_from_directory
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
31
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
32 # Make html report directory for output.
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
33 os.mkdir( html_dir )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
34
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
35 # Make a tmp dir
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
36 tmp_dir = tempfile.mkdtemp( prefix='tmp-cummeRbund-' )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
37
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
38 # Build command.
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
39 cmd = ( "Rscript --vanilla %s" % options.r_script )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
40
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
41 # Debugging.
8
b0d11fcbc3ac cummerbund add MDS and PCA plots, handle errors from R
Jim Johnson <jj@umn.edu>
parents: 1
diff changeset
42 # print cmd
0
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
43
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
44 #liubo added, for test, look at the generated R script
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
45 # shutil.copy2(options.r_script, '/nfs/software/galaxy/r_script')
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
46
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
47
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
48 # Run command.
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
49 try:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
50 tmp_name = tempfile.NamedTemporaryFile( dir=tmp_dir ).name
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
51 tmp_stderr = open( tmp_name, 'wb' )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
52 proc = subprocess.Popen( args=cmd, shell=True, cwd=html_dir, stderr=tmp_stderr.fileno() )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
53 returncode = proc.wait()
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
54 tmp_stderr.close()
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
55
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
56 # Get stderr, allowing for case where it's very large.
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
57 tmp_stderr = open( tmp_name, 'rb' )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
58 stderr = ''
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
59 buffsize = 1048576
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
60 try:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
61 while True:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
62 stderr += tmp_stderr.read( buffsize )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
63 if not stderr or len( stderr ) % buffsize != 0:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
64 break
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
65 except OverflowError:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
66 pass
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
67 tmp_stderr.close()
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
68
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
69 # Error checking.
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
70 if returncode != 0:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
71 raise Exception, stderr
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
72 except Exception, e:
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
73 stop_err( 'Error running R script. ' + str( e ) )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
74
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
75 # write the html report
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
76 html_report_from_directory( open( html_filename, 'wb' ), html_dir )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
77
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
78 # Clean up temp dirs
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
79 if os.path.exists( tmp_dir ):
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
80 shutil.rmtree( tmp_dir )
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
81
8ba5f0460b1e Uploaded
jjohnson
parents:
diff changeset
82 if __name__=="__main__": __main__()