annotate stderr_wrapper.py @ 2:93451f832736 draft

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