Mercurial > repos > trinity_ctat > align_and_estimate_abundance
annotate trinityToolWrapper.py @ 9:b6152cc26ffd draft
Trying to fix a syntax error.
| author | trinity_ctat |
|---|---|
| date | Tue, 26 Sep 2017 15:14:48 -0400 |
| parents | 28f6d3b8e931 |
| children | a8fa525a2f30 |
| rev | line source |
|---|---|
| 0 | 1 #!/usr/bin/env python |
| 2 | |
| 3 | |
| 4 # borrowed from: http://wiki.g2.bx.psu.edu/Future/Job%20Failure%20When%20stderr and modified for use with Trinity tools. | |
| 5 | |
| 6 """ | |
| 7 Wrapper that execute a program and its arguments but reports standard error | |
| 8 messages only if the program exit status was not 0 | |
| 9 Example: ./stderr_wrapper.py myprog arg1 -f arg2 | |
| 10 """ | |
| 11 | |
| 12 import sys, subprocess, os | |
| 13 | |
| 14 assert sys.version_info[:2] >= ( 2, 4 ) | |
| 15 | |
| 16 TRINITY_BASE_DIR = "" | |
| 17 if os.environ.has_key('TRINITY_HOME'): | |
| 18 TRINITY_BASE_DIR = os.environ['TRINITY_HOME']; | |
|
5
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
19 elif hasattr(os, 'symlink'): # symlink was implemented to always return false when it was not implemented in earlier versions. |
|
2
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
20 # 2017-09-26 |
|
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
21 # Cicada Dennis added looking for the location of the Trinity program using the Unix "which" utility. |
| 0 | 22 # I tried using "command -v Trinity" but for some reason, I was getting a OS permission error with that. |
|
2
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
23 # I just found distutils.spawn.find_executable() which might work, but already implemented the below. |
| 0 | 24 try: |
| 25 pipe1 = subprocess.Popen(["which", "Trinity"], stdout=subprocess.PIPE) | |
| 26 except: | |
| 8 | 27 msg = "You must set the environmental variable TRINITY_HOME to the base installation directory of Trinity before running {:s}.".format(sys.argv[0]) |
| 28 sys.stderr.write(msg) | |
| 9 | 29 # t, v, tb = sys.exc_info() |
| 30 # raise t, v, tb | |
| 31 # For some reason the above was giving a syntax error. | |
| 32 # A simple raise shoudl reraise the existing exception. | |
| 33 raise | |
| 0 | 34 else: |
|
2
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
35 TrinityPath, err_info = pipe1.communicate() |
|
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
36 # FIX - probably should be checking err_info for errors... |
|
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
37 # Determine the TRINITY_BASE_DIR from output1. |
|
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
38 # If TrinityPath is a link, we need to dereference the link. |
|
5
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
39 TrinityPath = TrinityPath.rstrip() # Need to strip off a newline. |
|
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
40 # print "Trinity that was found is: {:s}".format(repr(TrinityPath)) |
|
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
41 # print os.path.islink(TrinityPath) |
|
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
42 TrinityPath = os.path.abspath(TrinityPath) |
|
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
43 print "The Absolute Trinity path that was found is: {:s}".format(TrinityPath) |
|
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
44 # print os.path.islink(TrinityPath) |
| 4 | 45 while os.path.islink(TrinityPath): |
|
5
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
46 print "That path is a link." |
|
2
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
47 TrinityPath = os.path.join(os.path.dirname(TrinityPath),os.readlink(TrinityPath)) |
|
5
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
48 print "The new path is: {:s}".format(TrinityPath) |
| 0 | 49 # Take off the last part of the path (which is the Trinity command) |
|
2
9e5c3f162eca
Needed to dereference links when searching for TRINITY_BASE_DIR.
trinity_ctat
parents:
0
diff
changeset
|
50 TRINITY_BASE_DIR = "/".join(TrinityPath.split("/")[0:-1]) |
|
6
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
51 else: |
|
7
78d12a725443
Adding additional requirements for align_and_estimate_abundance.
trinity_ctat
parents:
6
diff
changeset
|
52 sys.stderr.write("Either set TRINITY_HOME to the trinity base directory, or ensure that directory is in the PATH before running.") |
|
78d12a725443
Adding additional requirements for align_and_estimate_abundance.
trinity_ctat
parents:
6
diff
changeset
|
53 sys.exit() |
|
78d12a725443
Adding additional requirements for align_and_estimate_abundance.
trinity_ctat
parents:
6
diff
changeset
|
54 |
| 0 | 55 |
| 56 # get bindir | |
| 57 bindir = sys.argv[0] | |
| 58 bindir = bindir.split("/") | |
| 59 if len(bindir) > 1: | |
| 60 bindir.pop() | |
| 61 bindir = "/".join(bindir) | |
| 62 else: | |
| 63 bindir = "." | |
| 64 | |
| 65 | |
| 66 ## add locations of tools to path setting. | |
|
6
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
67 #TOOL_PATHS_FILE = bindir + "/__add_to_PATH_setting.txt"; |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
68 #for line in open(TOOL_PATHS_FILE): |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
69 # line = line.rstrip() |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
70 # os.environ['PATH'] += ":" + line |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
71 |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
72 # Add TrinityPath and its utils to the PATH environment variable. |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
73 # print "Initially the PATH env variable is:\n\t{:s}".format(os.environ['PATH']) |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
74 os.environ['PATH'] = os.environ['PATH'] + ":{:s}:{:s}".format(TRINITY_BASE_DIR,TRINITY_BASE_DIR+"/util") |
|
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
75 # print "Now the PATH env variable is:\n\t{:s}".format(os.environ['PATH']) |
| 0 | 76 |
| 77 | |
| 78 def stop_err( msg ): | |
| 79 sys.stderr.write( "%s\n" % msg ) | |
| 80 sys.exit() | |
| 81 | |
| 82 def __main__(): | |
| 83 # Get command-line arguments | |
| 84 args = sys.argv | |
| 85 # Remove name of calling program, i.e. ./stderr_wrapper.py | |
| 86 args.pop(0) | |
| 87 # If there are no arguments left, we're done | |
| 88 if len(args) == 0: | |
| 89 return | |
| 90 | |
| 91 # If one needs to silence stdout | |
| 92 #args.append( ">" ) | |
| 93 #args.append( "/dev/null" ) | |
|
5
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
94 print "The TRINITY_BASE_DIR is:\n\t{:s}".format(TRINITY_BASE_DIR) |
|
6
12bc09b4a26d
Adding TrinityPath/util to the PATH so other utilities can be found.
trinity_ctat
parents:
5
diff
changeset
|
95 print "The PATH env variable is:\n\t{:s}".format(os.environ['PATH']) |
| 0 | 96 |
| 97 args[0] = "".join([TRINITY_BASE_DIR, '/', args[0]]); | |
| 98 | |
| 99 cmdline = " ".join(args) | |
| 100 | |
|
5
98d99fed9d35
Strip eol off of TrinityPath, so os.path.islink() works correctly.
trinity_ctat
parents:
4
diff
changeset
|
101 print "The command being invoked is:\n\t{:s}".format(cmdline) |
| 0 | 102 |
| 103 | |
| 104 try: | |
| 105 # Run program | |
| 106 err_capture = open("stderr.txt", 'w') | |
| 107 proc = subprocess.Popen( args=cmdline, shell=True, stderr=err_capture, stdout=sys.stdout ) | |
| 108 returncode = proc.wait() | |
| 109 err_capture.close() | |
| 110 | |
| 111 | |
| 112 if returncode != 0: | |
| 113 raise Exception | |
| 114 | |
| 115 except Exception: | |
| 116 # Running Grinder failed: write error message to stderr | |
| 117 err_text = open("stderr.txt").readlines() | |
| 118 stop_err( "ERROR:\n" + "\n".join(err_text)) | |
| 119 | |
| 120 | |
| 121 if __name__ == "__main__": __main__() |