annotate sr_assembly/velveth_wrapper.py @ 0:fa1f3c7464a1 draft

Uploaded
author scisjnu123
date Tue, 09 Jul 2019 09:42:27 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
1 #!/usr/bin/env python
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
2
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
3 """
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
4 Classes encapsulating decypher tool.
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
5 James E Johnson - University of Minnesota
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
6 """
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
7 import pkg_resources
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
8 import logging, os, string, sys, tempfile, glob, shutil, types, urllib
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
9 import shlex, subprocess
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
10 from optparse import OptionParser, OptionGroup
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
11 from stat import *
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
12
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
13
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
14 log = logging.getLogger( __name__ )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
15
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
16 assert sys.version_info[:2] >= ( 2, 4 )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
17
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
18 def stop_err( msg ):
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
19 sys.stderr.write( "%s\n" % msg )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
20 sys.exit()
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
21
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
22 def __main__():
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
23 #Parse Command Line
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
24 s = 'velveth_wrapper.py: argv = %s\n' % (sys.argv)
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
25 argcnt = len(sys.argv)
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
26 html_file = sys.argv[1]
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
27 working_dir = sys.argv[2]
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
28 try: # for test - needs this done
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
29 os.makedirs(working_dir)
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
30 except Exception, e:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
31 stop_err( 'Error running velveth ' + str( e ) )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
32 hash_length = sys.argv[3]
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
33 inputs = string.join(sys.argv[4:],' ')
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
34 cmdline = 'velveth %s %s %s > /dev/null' % (working_dir, hash_length, inputs)
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
35 try:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
36 proc = subprocess.Popen( args=cmdline, shell=True, stderr=subprocess.PIPE )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
37 returncode = proc.wait()
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
38 # get stderr, allowing for case where it's very large
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
39 stderr = ''
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
40 buffsize = 1048576
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
41 try:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
42 while True:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
43 stderr += proc.stderr.read( buffsize )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
44 if not stderr or len( stderr ) % buffsize != 0:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
45 break
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
46 except OverflowError:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
47 pass
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
48 if returncode != 0:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
49 raise Exception, stderr
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
50 except Exception, e:
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
51 stop_err( 'Error running velveth ' + str( e ) )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
52 sequences_path = os.path.join(working_dir,'Sequences')
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
53 roadmaps_path = os.path.join(working_dir,'Roadmaps')
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
54 rval = ['<html><head><title>Velvet Galaxy Composite Dataset </title></head><p/>']
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
55 rval.append('<div>%s<p/></div>' % (cmdline) )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
56 rval.append('<div>This composite dataset is composed of the following files:<p/><ul>')
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
57 rval.append( '<li><a href="%s" type="text/plain">%s </a>%s</li>' % (sequences_path,'Sequences','Sequences' ) )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
58 rval.append( '<li><a href="%s" type="text/plain">%s </a>%s</li>' % (roadmaps_path,'Roadmaps','Roadmaps' ) )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
59 rval.append( '</ul></div></html>' )
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
60 f = file(html_file,'w')
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
61 f.write("\n".join( rval ))
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
62 f.write('\n')
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
63 f.close()
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
64
fa1f3c7464a1 Uploaded
scisjnu123
parents:
diff changeset
65 if __name__ == "__main__": __main__()