annotate grinder-fa1fa683bcf1/stderr_wrapper.py @ 0:a41241d67693 default tip

Uploaded
author greg
date Wed, 18 Jan 2012 11:45:20 -0500
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
a41241d67693 Uploaded
greg
parents:
diff changeset
1 #!/usr/bin/env python
a41241d67693 Uploaded
greg
parents:
diff changeset
2
a41241d67693 Uploaded
greg
parents:
diff changeset
3 """
a41241d67693 Uploaded
greg
parents:
diff changeset
4 Wrapper that executes a program with its arguments but reports standard error
a41241d67693 Uploaded
greg
parents:
diff changeset
5 messages only if the program exit status was not 0. This is useful to prevent
a41241d67693 Uploaded
greg
parents:
diff changeset
6 Galaxy to interpret that there was an error if something was printed on stderr,
a41241d67693 Uploaded
greg
parents:
diff changeset
7 e.g. if this was simply a warning.
a41241d67693 Uploaded
greg
parents:
diff changeset
8 Example: ./stderr_wrapper.py myprog arg1 -f arg2
a41241d67693 Uploaded
greg
parents:
diff changeset
9 Author: Florent Angly
a41241d67693 Uploaded
greg
parents:
diff changeset
10 """
a41241d67693 Uploaded
greg
parents:
diff changeset
11
a41241d67693 Uploaded
greg
parents:
diff changeset
12 import sys, subprocess
a41241d67693 Uploaded
greg
parents:
diff changeset
13
a41241d67693 Uploaded
greg
parents:
diff changeset
14 assert sys.version_info[:2] >= ( 2, 4 )
a41241d67693 Uploaded
greg
parents:
diff changeset
15
a41241d67693 Uploaded
greg
parents:
diff changeset
16 def stop_err( msg ):
a41241d67693 Uploaded
greg
parents:
diff changeset
17 sys.stderr.write( "%s\n" % msg )
a41241d67693 Uploaded
greg
parents:
diff changeset
18 sys.exit()
a41241d67693 Uploaded
greg
parents:
diff changeset
19
a41241d67693 Uploaded
greg
parents:
diff changeset
20 def __main__():
a41241d67693 Uploaded
greg
parents:
diff changeset
21 # Get command-line arguments
a41241d67693 Uploaded
greg
parents:
diff changeset
22 args = sys.argv
a41241d67693 Uploaded
greg
parents:
diff changeset
23 # Remove name of calling program, i.e. ./stderr_wrapper.py
a41241d67693 Uploaded
greg
parents:
diff changeset
24 args.pop(0)
a41241d67693 Uploaded
greg
parents:
diff changeset
25 # If there are no arguments left, we're done
a41241d67693 Uploaded
greg
parents:
diff changeset
26 if len(args) == 0:
a41241d67693 Uploaded
greg
parents:
diff changeset
27 return
a41241d67693 Uploaded
greg
parents:
diff changeset
28
a41241d67693 Uploaded
greg
parents:
diff changeset
29 # If one needs to silence stdout
a41241d67693 Uploaded
greg
parents:
diff changeset
30 #args.append( ">" )
a41241d67693 Uploaded
greg
parents:
diff changeset
31 #args.append( "/dev/null" )
a41241d67693 Uploaded
greg
parents:
diff changeset
32
a41241d67693 Uploaded
greg
parents:
diff changeset
33 #cmdline = " ".join(args)
a41241d67693 Uploaded
greg
parents:
diff changeset
34 #print cmdline
a41241d67693 Uploaded
greg
parents:
diff changeset
35 try:
a41241d67693 Uploaded
greg
parents:
diff changeset
36 # Run program
a41241d67693 Uploaded
greg
parents:
diff changeset
37 proc = subprocess.Popen( args=args, shell=False, stderr=subprocess.PIPE )
a41241d67693 Uploaded
greg
parents:
diff changeset
38 returncode = proc.wait()
a41241d67693 Uploaded
greg
parents:
diff changeset
39 # Capture stderr, allowing for case where it's very large
a41241d67693 Uploaded
greg
parents:
diff changeset
40 stderr = ''
a41241d67693 Uploaded
greg
parents:
diff changeset
41 buffsize = 1048576
a41241d67693 Uploaded
greg
parents:
diff changeset
42 try:
a41241d67693 Uploaded
greg
parents:
diff changeset
43 while True:
a41241d67693 Uploaded
greg
parents:
diff changeset
44 stderr += proc.stderr.read( buffsize )
a41241d67693 Uploaded
greg
parents:
diff changeset
45 if not stderr or len( stderr ) % buffsize != 0:
a41241d67693 Uploaded
greg
parents:
diff changeset
46 break
a41241d67693 Uploaded
greg
parents:
diff changeset
47 except OverflowError:
a41241d67693 Uploaded
greg
parents:
diff changeset
48 pass
a41241d67693 Uploaded
greg
parents:
diff changeset
49 # Running Grinder failed: write error message to stderr
a41241d67693 Uploaded
greg
parents:
diff changeset
50 if returncode != 0:
a41241d67693 Uploaded
greg
parents:
diff changeset
51 raise Exception, stderr
a41241d67693 Uploaded
greg
parents:
diff changeset
52 except Exception, e:
a41241d67693 Uploaded
greg
parents:
diff changeset
53 # Running Grinder failed: write error message to stderr
a41241d67693 Uploaded
greg
parents:
diff changeset
54 stop_err( 'Error: ' + str( e ) )
a41241d67693 Uploaded
greg
parents:
diff changeset
55
a41241d67693 Uploaded
greg
parents:
diff changeset
56
a41241d67693 Uploaded
greg
parents:
diff changeset
57 if __name__ == "__main__": __main__()