view qualimap_bamqc.py @ 6:a2438e0f1bbf draft default tip

planemo upload for repository https://github.com/refinery-platform/qualimap2 commit 40f39b0f1c81cae4606d15d1c0bd318b1e4c572d

#!/usr/bin/env python
from __future__ import print_function
import argparse
import os
from subprocess import check_call, CalledProcessError
import shutil
import sys

QUALIMAP_OUPUT_DIR = "qualimap_results"


def qualimap_bamqc(bam_filename, jv_mem_size):
    qualimap_command = [
        "qualimap", "bamqc",
        "-bam " + bam_filename,
        "-outdir " + QUALIMAP_OUPUT_DIR,
        "--java-mem-size=" + jv_mem_size
    ]

    try:
        check_call(qualimap_command)
    except CalledProcessError:
        print("Error running the qualimap bamqc", file=sys.stderr)


def main():
    parser = argparse.ArgumentParser(
        description="Generate Bam Quality Statistics"
    )
    parser.add_argument('--input_file')
    parser.add_argument('--out_dir')
    parser.add_argument('--out_results')
    parser.add_argument('--out_zip')
    parser.add_argument('--java_mem_size')

    args = parser.parse_args()
    
    # Run qualimap
    qualimap_bamqc(args.input_file, args.java_mem_size)
    
    # Create .zip archive containing the raw_data_qualimapReport files
    shutil.make_archive(
        'raw_data_qualimapReport',
        'zip',
        os.path.join(QUALIMAP_OUPUT_DIR, 'raw_data_qualimapReport')
    )
    
    # Move newly created .zip to it's proper Galaxy output file 
    shutil.move("raw_data_qualimapReport.zip", args.out_zip)
    
    # Move genome_results.txt to it's proper Galaxy output file 
    shutil.move(
        os.path.join(QUALIMAP_OUPUT_DIR, "genome_results.txt"),
        args.out_results
    )

if __name__ == "__main__":
    main()