diff pyrocleaner_galaxy_tool_V1.2/pyrocleaner_wrapper.py @ 0:ef5dd11c01e6 default tip

pyrocleaner v1.2
author g2cmnty@test-web1.g2.bx.psu.edu
date Thu, 09 Jun 2011 06:09:09 -0400
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pyrocleaner_galaxy_tool_V1.2/pyrocleaner_wrapper.py	Thu Jun 09 06:09:09 2011 -0400
@@ -0,0 +1,88 @@
+#
+# Pyrocleaner wrapper for galaxy
+# Copyright (C) 2009 INRA
+# 
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+#
+
+__author__ = 'Plateforme bioinformatique Midi Pyrenees'
+__copyright__ = 'Copyright (C) 2009 INRA'
+__license__ = 'GNU General Public License'
+__version__ = '1.2'
+__email__ = 'support.genopole@toulouse.inra.fr'
+__status__ = 'beta'
+
+import optparse, os, shutil, subprocess, sys, tempfile, re, string
+    
+if __name__ == "__main__":
+    
+    parser = optparse.OptionParser()
+    
+    parser.add_option( '-s', '--options',      dest='options',      help='The pyrocleaner options' )
+    parser.add_option( '-l', '--log',          dest='log',          help='Path to the pyrocleaner log' )
+    parser.add_option( '-f', '--format',       dest='format',       help='The input file format (sff|fastq)' )
+    parser.add_option( '-o', '--output',       dest='output',       help='The output file path' )
+    parser.add_option( '-a', '--out-pe-qual',  dest='out_pe_qual',  help='The output pairends qual file path' )
+    parser.add_option( '-b', '--out-pe-fasta', dest='out_pe_fasta', help='The output pairends fasta file path' )
+    parser.add_option( '-d', '--out-dir',      dest='out_dir',      help='The output dir where to process the data' )
+    
+    (options, args) = parser.parse_args()
+    
+    buffsize = 1048576
+    tmp_dir = tempfile.mkdtemp()
+    
+    try:
+    
+        if options.format == "sff": format = "sff"
+        else : format = "fastq"
+        
+        tmp_stderr_name = tempfile.NamedTemporaryFile( dir=tmp_dir,suffix='.err' ).name
+        tmp_stderr = open( tmp_stderr_name, 'wb' )
+        tmp_stdout_name = tempfile.NamedTemporaryFile( dir=tmp_dir,suffix='.out' ).name
+        tmp_stdout = open( tmp_stdout_name, 'wb' )
+        cmd = 'pyrocleaner --format %s %s --out %s' % (format, options.options, options.out_dir)
+        proc = subprocess.Popen( args=cmd, shell=True, cwd=tmp_dir, stderr=tmp_stderr.fileno(), stdout=tmp_stdout.fileno() )
+        returncode = proc.wait()
+        tmp_stderr.close()
+        # get stderr, allowing for case where it's very large
+        tmp_stderr = open( tmp_stderr_name, 'rb' )
+        stderr = ''
+        try:
+            while True:
+                stderr += tmp_stderr.read( buffsize )
+                if not stderr or len( stderr ) % buffsize != 0:
+                    break
+        except OverflowError:
+            pass
+        tmp_stderr.close()
+        if returncode != 0:
+            raise Exception, stderr
+        
+        # Move the log file to the right place
+        shutil.move(os.path.join(options.out_dir, "pyrocleaner.log"), options.log)
+        
+        for file in os.listdir(options.out_dir):
+            # If the file has the input file format and is not a shotgun file
+            if file.endswith(format):
+                shutil.move(os.path.join(options.out_dir, file), options.output)
+            # If it's a quality file from shotgun
+            if file.endswith(".qual") :
+                shutil.move(os.path.join(options.out_dir, file), options.out_pe_qual)
+            # If it's a fasta file from shotgun
+            elif file.endswith(".fasta") :
+                shutil.move(os.path.join(options.out_dir, file), options.out_pe_fasta)
+
+    except Exception:
+        raise Exception, 'Error executing pyrocleaner.'