source: cows_wps/trunk/process_modules/get_weather_stations.py @ 7018

"""
get_weather_stations.py
===================

Process get_weather_stations that holds the GetWeatherStations class.

"""

import os, stat, time
import sys
import logging

from cows_wps.process_handler.fileset import FileSet, FLAG
import cows_wps.process_handler.process_support as process_support
from cows_wps.process_handler.context.process_status import STATUS

import process_modules.lib_get_weather_stations.utils as gws_utils


log = logging.getLogger(__name__)
log.setLevel(logging.DEBUG)


class GetWeatherStations(object):
    def __call__(self, context):
        
        startTime = time.time()
        jobId = os.path.basename(context.processDir)

        # Parse the inputs
        Counties = context.inputs.get('Counties', [])
        if Counties == []:
            BBox = context.inputs.get("BBox", None)
            if BBox == None:
                raise Exception("Invalid arguments provided. Must provide either a geographical bounding box or a list of counties.")

        DataTypes = context.inputs.get("DataTypes", [])
        StartDateTime = context.inputs['StartDateTime']
        StartDateTime = gws_utils.revertDateTimeToLongString(StartDateTime)
        log.warn("DATE TIME: %s, type = %s" % (StartDateTime, type(StartDateTime)))
        EndDateTime = context.inputs['EndDateTime']
        EndDateTime = gws_utils.revertDateTimeToLongString(EndDateTime)
        
        context.setStatus(STATUS.STARTED, 'Job is now running', 0)

        # Always need a FileSet, even if empty
        self.fileSet = context.outputs['FileSet'] = FileSet()

        # Make path to output file
        WFile = 'weather_stations.txt'
        MyFilePath = context.processDir + '/outputs/' + WFile

        # Call code to get Weather Stations
        station_list = gws_utils.getStationList(Counties, BBox, DataTypes, StartDateTime,
                           EndDateTime, MyFilePath)

        context.outputs['ProcessSpecificContent'] = {
                                    "WeatherStations": " ".join(station_list)} 
        context.outputs['job_details']['job_capabilities'] = "send_to_extract_weather_data"

        context.log.info('Written output file: %s' % WFile)
        filesize = os.stat(MyFilePath)[stat.ST_SIZE]

        fileSet.contents.append(FileSet(FLAG.DATA, WFile, filesize, 'Weather Stations File'))

#        context.setStatus(STATUS.COMPLETED, 'The End', 100)
#        completionTime = time.time()
#        process_support.updateJobDetailsAfterCompletion(context, startTime, completionTime)

        # Set keep = True so that weather station files are accessible to downstream process
        # without unzipping. This is fine as files are small.
        process_support.finishProcess(context, self.fileSet, startTime, keep = True)


    def dryRun(self, context):

        self.fileSet = context.outputs['FileSet'] = FileSet()
        self.fileSet.contents.append(FileSet(FLAG.DATA, "testfile.txt", 30000, "Some file"))
        
        process_support.finishDryRun(context, [("meta", "data")], self.fileSet,
                                         60, acceptedMessage='Dry run complete')