Mercurial > repos > devteam > fastqc
comparison rgFastQC.py @ 8:5b995cef2fbb draft
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/fastqc commit bff04a5409662a8968c50501487042a3a0c93702
| author | iuc |
|---|---|
| date | Tue, 31 Jan 2017 07:22:22 -0500 |
| parents | ec73b7c83b2c |
| children | 0a7c65540937 |
comparison
equal
deleted
inserted
replaced
| 7:ec73b7c83b2c | 8:5b995cef2fbb |
|---|---|
| 13 | 13 |
| 14 EXAMPLE (generated by Galaxy) | 14 EXAMPLE (generated by Galaxy) |
| 15 | 15 |
| 16 rgFastQC.py -i path/dataset_1.dat -j 1000gsample.fastq -o path/dataset_3.dat -d path/job_working_directory/subfolder | 16 rgFastQC.py -i path/dataset_1.dat -j 1000gsample.fastq -o path/dataset_3.dat -d path/job_working_directory/subfolder |
| 17 -f fastq -n FastQC -c path/dataset_2.dat -e fastqc | 17 -f fastq -n FastQC -c path/dataset_2.dat -e fastqc |
| 18 | |
| 19 """ | 18 """ |
| 20 | 19 import bz2 |
| 20 import glob | |
| 21 import gzip | |
| 22 import mimetypes | |
| 23 import optparse | |
| 24 import os | |
| 21 import re | 25 import re |
| 22 import os | |
| 23 import shutil | 26 import shutil |
| 24 import subprocess | 27 import subprocess |
| 25 import optparse | |
| 26 import tempfile | 28 import tempfile |
| 27 import glob | |
| 28 import gzip | |
| 29 import bz2 | |
| 30 import zipfile | 29 import zipfile |
| 31 import mimetypes | 30 |
| 32 | 31 |
| 33 class FastQCRunner(object): | 32 class FastQCRunner(object): |
| 34 | 33 def __init__(self, opts=None): |
| 35 def __init__(self,opts=None): | |
| 36 ''' | 34 ''' |
| 37 Initializes an object to run FastQC in Galaxy. To start the process, use the function run_fastqc() | 35 Initializes an object to run FastQC in Galaxy. To start the process, use the function run_fastqc() |
| 38 ''' | 36 ''' |
| 39 | 37 |
| 40 # Check whether the options are specified and saves them into the object | 38 # Check whether the options are specified and saves them into the object |
| 41 assert opts != None | 39 assert opts is not None |
| 42 self.opts = opts | 40 self.opts = opts |
| 43 | 41 |
| 44 def prepare_command_line(self): | 42 def prepare_command_line(self): |
| 45 ''' | 43 ''' |
| 46 Develops the Commandline to run FastQC in Galaxy | 44 Develops the Commandline to run FastQC in Galaxy |
| 60 f.readline() | 58 f.readline() |
| 61 except: | 59 except: |
| 62 trimext = True | 60 trimext = True |
| 63 f.close() | 61 f.close() |
| 64 elif linf.endswith('bz2'): | 62 elif linf.endswith('bz2'): |
| 65 f = bz2.open(self.opts.input,'rb') | 63 f = bz2.BZ2File(self.opts.input, 'r') |
| 66 try: | 64 try: |
| 67 f.readline() | 65 f.readline() |
| 68 except: | 66 except: |
| 69 trimext = True | 67 trimext = True |
| 70 f.close() | 68 f.close() |
| 71 elif linf.endswith('.zip'): | 69 elif linf.endswith('.zip'): |
| 72 if not zipfile.is_zipfile(self.opts.input): | 70 if not zipfile.is_zipfile(self.opts.input): |
| 73 trimext = True | 71 trimext = True |
| 74 if trimext: | 72 if trimext: |
| 75 f = open(self.opts.input) | 73 f = open(self.opts.input) |
| 76 try: | 74 try: |
| 77 f.readline() | 75 f.readline() |
| 78 except: | 76 except: |
| 79 raise Exception("Input file corruption, could not identify the filetype") | 77 raise Exception("Input file corruption, could not identify the filetype") |
| 80 infname = os.path.splitext(infname)[0] | 78 infname = os.path.splitext(infname)[0] |
| 81 | 79 |
| 82 # Replace unwanted or problematic charaters in the input file name | 80 # Replace unwanted or problematic charaters in the input file name |
| 83 self.fastqinfilename = re.sub(ur'[^a-zA-Z0-9_\-\.]', '_', os.path.basename(infname)) | 81 self.fastqinfilename = re.sub(r'[^a-zA-Z0-9_\-\.]', '_', os.path.basename(infname)) |
| 84 # check that the symbolic link gets a proper ending, fastqc seems to ignore the given format otherwise | 82 # check that the symbolic link gets a proper ending, fastqc seems to ignore the given format otherwise |
| 85 if 'fastq' in opts.informat: | 83 if 'fastq' in self.opts.informat: |
| 86 # with fastq the .ext is ignored, but when a format is actually passed it must comply with fastqc's | 84 # with fastq the .ext is ignored, but when a format is actually passed it must comply with fastqc's |
| 87 # accepted formats.. | 85 # accepted formats.. |
| 88 opts.informat = 'fastq' | 86 self.opts.informat = 'fastq' |
| 89 elif not self.fastqinfilename.endswith(opts.informat): | 87 elif not self.fastqinfilename.endswith(self.opts.informat): |
| 90 self.fastqinfilename += '.%s' % opts.informat | 88 self.fastqinfilename += '.%s' % self.opts.informat |
| 91 | 89 |
| 92 # Build the Commandline from the given parameters | 90 # Build the Commandline from the given parameters |
| 93 command_line = [opts.executable, '--outdir %s' % opts.outputdir] | 91 command_line = [opts.executable, '--outdir %s' % self.opts.outputdir] |
| 94 if opts.contaminants != None: | 92 if self.opts.contaminants is not None: |
| 95 command_line.append('--contaminants %s' % opts.contaminants) | 93 command_line.append('--contaminants %s' % self.opts.contaminants) |
| 96 if opts.limits != None: | 94 if self.opts.limits is not None: |
| 97 command_line.append('--limits %s' % opts.limits) | 95 command_line.append('--limits %s' % self.opts.limits) |
| 98 command_line.append('--quiet') | 96 command_line.append('--quiet') |
| 99 command_line.append('--extract') # to access the output text file | 97 command_line.append('--extract') # to access the output text file |
| 100 if type[-1] != "gzip": | 98 if type[-1] != "gzip": |
| 101 command_line.append('-f %s' % opts.informat) | 99 command_line.append('-f %s' % self.opts.informat) |
| 102 else: | 100 else: |
| 103 self.fastqinfilename += ".gz" | 101 self.fastqinfilename += ".gz" |
| 104 command_line.append(self.fastqinfilename) | 102 command_line.append(self.fastqinfilename) |
| 105 self.command_line = ' '.join(command_line) | 103 self.command_line = ' '.join(command_line) |
| 106 | 104 |
| 107 def copy_output_file_to_dataset(self): | 105 def copy_output_file_to_dataset(self): |
| 108 ''' | 106 ''' |
| 109 Retrieves the output html and text files from the output directory and copies them to the Galaxy output files | 107 Retrieves the output html and text files from the output directory and copies them to the Galaxy output files |
| 110 ''' | 108 ''' |
| 111 | 109 |
| 112 # retrieve html file | 110 # retrieve html file |
| 113 result_file = glob.glob(opts.outputdir + '/*html') | 111 result_file = glob.glob(self.opts.outputdir + '/*html') |
| 114 with open(result_file[0], 'rb') as fsrc: | 112 with open(result_file[0], 'rb') as fsrc: |
| 115 with open(self.opts.htmloutput, 'wb') as fdest: | 113 with open(self.opts.htmloutput, 'wb') as fdest: |
| 116 shutil.copyfileobj(fsrc, fdest) | 114 shutil.copyfileobj(fsrc, fdest) |
| 117 | 115 |
| 118 # retrieve text file | 116 # retrieve text file |
| 119 text_file = glob.glob(opts.outputdir + '/*/fastqc_data.txt') | 117 text_file = glob.glob(self.opts.outputdir + '/*/fastqc_data.txt') |
| 120 with open(text_file[0], 'rb') as fsrc: | 118 with open(text_file[0], 'rb') as fsrc: |
| 121 with open(self.opts.textoutput, 'wb') as fdest: | 119 with open(self.opts.textoutput, 'wb') as fdest: |
| 122 shutil.copyfileobj(fsrc, fdest) | 120 shutil.copyfileobj(fsrc, fdest) |
| 123 | 121 |
| 124 def run_fastqc(self): | 122 def run_fastqc(self): |
| 125 ''' | 123 ''' |
| 126 Executes FastQC. Make sure the mandatory import parameters input, inputfilename, outputdir and htmloutput have been specified in the options (opts) | 124 Executes FastQC. Make sure the mandatory import parameters input, inputfilename, outputdir and htmloutput have been specified in the options |
| 127 ''' | 125 ''' |
| 128 | 126 |
| 129 # Create a log file | 127 # Create a log file |
| 130 dummy,tlog = tempfile.mkstemp(prefix='rgFastQC',suffix=".log",dir=self.opts.outputdir) | 128 dummy, tlog = tempfile.mkstemp(prefix='rgFastQC', suffix=".log", dir=self.opts.outputdir) |
| 131 sout = open(tlog, 'w') | 129 sout = open(tlog, 'w') |
| 132 | 130 |
| 133 self.prepare_command_line() | 131 self.prepare_command_line() |
| 134 sout.write(self.command_line) | 132 sout.write(self.command_line) |
| 135 sout.write('\n') | 133 sout.write('\n') |
| 136 sout.write("Creating symlink\n") # between the input (.dat) file and the given input file name | 134 sout.write("Creating symlink\n") # between the input (.dat) file and the given input file name |
| 137 os.symlink(self.opts.input, self.fastqinfilename) | 135 os.symlink(self.opts.input, self.fastqinfilename) |
| 138 sout.write("check_call\n") | 136 sout.write("check_call\n") |
| 139 subprocess.check_call(self.command_line, shell=True) | 137 subprocess.check_call(self.command_line, shell=True) |
| 140 sout.write("Copying working %s file to %s \n" % (self.fastqinfilename, self.opts.htmloutput)) | 138 sout.write("Copying working %s file to %s \n" % (self.fastqinfilename, self.opts.htmloutput)) |
| 141 self.copy_output_file_to_dataset() | 139 self.copy_output_file_to_dataset() |
| 142 sout.write("Finished") | 140 sout.write("Finished") |
| 143 sout.close() | 141 sout.close() |
| 142 | |
| 144 | 143 |
| 145 if __name__ == '__main__': | 144 if __name__ == '__main__': |
| 146 op = optparse.OptionParser() | 145 op = optparse.OptionParser() |
| 147 op.add_option('-i', '--input', default=None) | 146 op.add_option('-i', '--input', default=None) |
| 148 op.add_option('-j', '--inputfilename', default=None) | 147 op.add_option('-j', '--inputfilename', default=None) |
| 154 op.add_option('-c', '--contaminants', default=None) | 153 op.add_option('-c', '--contaminants', default=None) |
| 155 op.add_option('-l', '--limits', default=None) | 154 op.add_option('-l', '--limits', default=None) |
| 156 op.add_option('-e', '--executable', default='fastqc') | 155 op.add_option('-e', '--executable', default='fastqc') |
| 157 opts, args = op.parse_args() | 156 opts, args = op.parse_args() |
| 158 | 157 |
| 159 assert opts.input != None | 158 assert opts.input is not None |
| 160 assert opts.inputfilename != None | 159 assert opts.inputfilename is not None |
| 161 assert opts.htmloutput != None | 160 assert opts.htmloutput is not None |
| 162 if not os.path.exists(opts.outputdir): | 161 if not os.path.exists(opts.outputdir): |
| 163 os.makedirs(opts.outputdir) | 162 os.makedirs(opts.outputdir) |
| 164 | 163 |
| 165 fastqc_runner = FastQCRunner(opts) | 164 fastqc_runner = FastQCRunner(opts) |
| 166 fastqc_runner.run_fastqc() | 165 fastqc_runner.run_fastqc() |
