@@ -180,6 +180,17 @@ def bunzip2(fq_pair):
180180
181181 return bunzipped
182182
183+ def get_new_filename (fastq_old , rg_id_fn , r1_r2 , outdir ):
184+ if fastq_old .endswith ('fq' ) or fastq_old .endswith ('fastq' ):
185+ fastq_new = os .path .join (os .getcwd (), outdir , f'{ rg_id_fn } _{ r1_r2 } .fq' )
186+ elif fastq_old .endswith ('fq.gz' ) or fastq_old .endswith ('fastq.gz' ):
187+ fastq_new = os .path .join (os .getcwd (), outdir , f'{ rg_id_fn } _{ r1_r2 } .fq.gz' )
188+ else :
189+ sys .exit ("Unsupported file format: %s." % fastq_old )
190+
191+ return fastq_new
192+
193+
183194
184195def main ():
185196 """
@@ -228,14 +239,16 @@ def main():
228239 fq_pair = filename_to_file (fp , args .seq_files )
229240 fastq_pair = bunzip2 (fq_pair )
230241 rg = filepair_map_to_readgroup [fp ]['read_groups' ][0 ]
231- file_r1_new = os .path .join (os .getcwd (), args .outdir , os .path .basename (fastq_pair [0 ]))
242+ rg_id = rg ['submitter_read_group_id' ]
243+ rg_id_fn = readgroup_id_to_fname (rg_id , '' , study_id , donor_id , sample_id )
244+ file_r1_new = get_new_filename (fastq_pair [0 ], rg_id_fn , "R1" , args .outdir )
232245 os .symlink (os .path .abspath (fastq_pair [0 ]), file_r1_new )
233246 if fastq_pair [1 ]:
234- file_r2_new = os . path . join ( os . getcwd (), args . outdir , os . path . basename ( fastq_pair [ 1 ]) )
235- os .symlink (os .path .abspath (fastq_pair [0 ]), file_r2_new )
247+ file_r2_new = get_new_filename ( fastq_pair [ 1 ], rg_id_fn , "R2" , args . outdir )
248+ os .symlink (os .path .abspath (fastq_pair [1 ]), file_r2_new )
236249 else :
237250 file_r2_new = 'No_File'
238- rgs_file_pair_map [rg [ 'submitter_read_group_id' ] ] = {
251+ rgs_file_pair_map [rg_id ] = {
239252 'file_r1' : file_r1_new ,
240253 'file_r2' : file_r2_new
241254 }
0 commit comments