Mercurial > repos > peterjc > effectivet3
diff tools/effectiveT3/effectiveT3.py @ 8:60a9b3f760cc draft
v0.0.17 Used cached URL, python style updates
author | peterjc |
---|---|
date | Wed, 01 Feb 2017 09:22:21 -0500 |
parents | 5f85301d50bf |
children | 512530020360 |
line wrap: on
line diff
--- a/tools/effectiveT3/effectiveT3.py Mon Sep 21 05:52:29 2015 -0400 +++ b/tools/effectiveT3/effectiveT3.py Wed Feb 01 09:22:21 2017 -0500 @@ -16,30 +16,26 @@ import subprocess # The Galaxy auto-install via tool_dependencies.xml will set this environment variable -effectiveT3_dir = os.environ.get("EFFECTIVET3", "/opt/EffectiveT3/") -effectiveT3_jar = os.path.join(effectiveT3_dir, "TTSS_GUI-1.0.1.jar") +effective_t3_dir = os.environ.get("EFFECTIVET3", "/opt/EffectiveT3/") +effective_t3_jar = os.path.join(effective_t3_dir, "TTSS_GUI-1.0.1.jar") if "-v" in sys.argv or "--version" in sys.argv: # TODO - Get version of the JAR file dynamically? - print("Wrapper v0.0.16, TTSS_GUI-1.0.1.jar") + print("Wrapper v0.0.17, TTSS_GUI-1.0.1.jar") sys.exit(0) -def sys_exit(msg, error_level=1): - """Print error message to stdout and quit with given error level.""" - sys.stderr.write("%s\n" % msg) - sys.exit(error_level) - if len(sys.argv) != 5: - sys_exit("Require four arguments: model, threshold, input protein FASTA file & output tabular file") + sys.exit("Require four arguments: model, threshold, input protein FASTA file & output tabular file") model, threshold, fasta_file, tabular_file = sys.argv[1:] if not os.path.isfile(fasta_file): - sys_exit("Input FASTA file not found: %s" % fasta_file) + sys.exit("Input FASTA file not found: %s" % fasta_file) if threshold not in ["selective", "sensitive"] \ -and not threshold.startswith("cutoff="): - sys_exit("Threshold should be selective, sensitive, or cutoff=..., not %r" % threshold) + and not threshold.startswith("cutoff="): + sys.exit("Threshold should be selective, sensitive, or cutoff=..., not %r" % threshold) + def clean_tabular(raw_handle, out_handle): """Clean up Effective T3 output to make it tabular.""" @@ -48,65 +44,66 @@ errors = 0 for line in raw_handle: if not line or line.startswith("#") \ - or line.startswith("Id; Description; Score;"): + or line.startswith("Id; Description; Score;"): continue assert line.count(";") >= 3, repr(line) # Normally there will just be three semi-colons, however the # original FASTA file's ID or description might have had # semi-colons in it as well, hence the following hackery: try: - id_descr, score, effective = line.rstrip("\r\n").rsplit(";",2) + id_descr, score, effective = line.rstrip("\r\n").rsplit(";", 2) # Cope when there was no FASTA description if "; " not in id_descr and id_descr.endswith(";"): id = id_descr[:-1] descr = "" else: - id, descr = id_descr.split("; ",1) + id, descr = id_descr.split("; ", 1) except ValueError: - sys_exit("Problem parsing line:\n%s\n" % line) + sys.exit("Problem parsing line:\n%s\n" % line) parts = [s.strip() for s in [id, descr, score, effective]] out_handle.write("\t".join(parts) + "\n") count += 1 if float(score) < 0: - errors += 1 + errors += 1 if effective.lower() == "true": - positive += 1 + positive += 1 return count, positive, errors + def run(cmd): # Avoid using shell=True when we call subprocess to ensure if the Python # script is killed, so too is the child process. try: child = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) except Exception, err: - sys_exit("Error invoking command:\n%s\n\n%s\n" % (" ".join(cmd), err)) + sys.exit("Error invoking command:\n%s\n\n%s\n" % (" ".join(cmd), err)) # Use .communicate as can get deadlocks with .wait(), stdout, stderr = child.communicate() return_code = child.returncode if return_code or stderr.startswith("Exception in thread"): - cmd_str= " ".join(cmd) # doesn't quote spaces etc + cmd_str = " ".join(cmd) # doesn't quote spaces etc if stderr and stdout: - sys_exit("Return code %i from command:\n%s\n\n%s\n\n%s" % (return_code, cmd_str, stdout, stderr)) + sys.exit("Return code %i from command:\n%s\n\n%s\n\n%s" % (return_code, cmd_str, stdout, stderr)) else: - sys_exit("Return code %i from command:\n%s\n%s" % (return_code, cmd_str, stderr)) + sys.exit("Return code %i from command:\n%s\n%s" % (return_code, cmd_str, stderr)) -if not os.path.isdir(effectiveT3_dir): - sys_exit("Effective T3 folder not found: %r" % effectiveT3_dir) +if not os.path.isdir(effective_t3_dir): + sys.exit("Effective T3 folder not found: %r" % effective_t3_dir) -if not os.path.isfile(effectiveT3_jar): - sys_exit("Effective T3 JAR file not found: %r" % effectiveT3_jar) +if not os.path.isfile(effective_t3_jar): + sys.exit("Effective T3 JAR file not found: %r" % effective_t3_jar) -if not os.path.isdir(os.path.join(effectiveT3_dir, "module")): - sys_exit("Effective T3 module folder not found: %r" % os.path.join(effectiveT3_dir, "module")) +if not os.path.isdir(os.path.join(effective_t3_dir, "module")): + sys.exit("Effective T3 module folder not found: %r" % os.path.join(effective_t3_dir, "module")) -effectiveT3_model = os.path.join(effectiveT3_dir, "module", model) -if not os.path.isfile(effectiveT3_model): +effective_t3_model = os.path.join(effective_t3_dir, "module", model) +if not os.path.isfile(effective_t3_model): sys.stderr.write("Contents of %r is %s\n" - % (os.path.join(effectiveT3_dir, "module"), - ", ".join(repr(p) for p in os.listdir(os.path.join(effectiveT3_dir, "module"))))) - sys.stderr.write("Main JAR was found: %r\n" % effectiveT3_jar) - sys_exit("Effective T3 model JAR file not found: %r" % effectiveT3_model) + % (os.path.join(effective_t3_dir, "module"), + ", ".join(repr(p) for p in os.listdir(os.path.join(effective_t3_dir, "module"))))) + sys.stderr.write("Main JAR was found: %r\n" % effective_t3_jar) + sys.exit("Effective T3 model JAR file not found: %r" % effective_t3_model) # We will have write access whereever the output should be, temp_file = os.path.abspath(tabular_file + ".tmp") @@ -115,7 +112,7 @@ tabular_file = os.path.abspath(tabular_file) fasta_file = os.path.abspath(fasta_file) -cmd = ["java", "-jar", effectiveT3_jar, +cmd = ["java", "-jar", effective_t3_jar, "-f", fasta_file, "-m", model, "-t", threshold, @@ -124,14 +121,14 @@ try: # Must run from directory above the module subfolder: - os.chdir(effectiveT3_dir) -except: - sys_exit("Could not change to Effective T3 folder: %s" % effectiveT3_dir) + os.chdir(effective_t3_dir) +except Exception: + sys.exit("Could not change to Effective T3 folder: %s" % effective_t3_dir) run(cmd) if not os.path.isfile(temp_file): - sys_exit("ERROR - No output file from Effective T3") + sys.exit("ERROR - No output file from Effective T3") out_handle = open(tabular_file, "w") out_handle.write("#ID\tDescription\tScore\tEffective\n") @@ -143,11 +140,11 @@ os.remove(temp_file) if errors: - print("%i sequences, %i positive, %i errors" - % (count, positive, errors)) + print("%i sequences, %i positive, %i errors" + % (count, positive, errors)) else: - print("%i/%i sequences positive" % (positive, count)) + print("%i/%i sequences positive" % (positive, count)) -if count and count==errors: - # Galaxy will still allow them to see the output file - sys_exit("All your sequences gave an error code") +if count and count == errors: + # Galaxy will still allow them to see the output file + sys.exit("All your sequences gave an error code")