gistfile1.txt

from galaxy.jobs import JobDestination
import os
import logging

def space_allocation(job):
    """Allocate dynamic space"""
    logging.basicConfig(filename='fastqc.log', level=logging.DEBUG)
    logging.debug("I'm working...")
    
    for da in job.input_datasets:
        debugStr = str(da.name) + str(da.dataset) + "\n"
        logging.debug(debugStr)
    
    inp_data = dict( [ ( da.name, da.dataset ) for da in job.input_datasets ] )
    inp_data.update( [ ( da.name, da.dataset ) for da in job.input_library_datasets ] )
    query_file = inp_data[ "input_file" ].file_name
    query_size = os.path.getsize( query_file )
    
    if query_size > 600000000:
        nativeSpec = "-q test.q -pe galaxy 4 -l h_vmem=2G"
    else:
        nativeSpec = "-q test.q"

    return JobDestination(runner="drmaa", params={"nativeSpecification": nativeSpec})