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