72 lines
No EOL
2.3 KiB
Python
72 lines
No EOL
2.3 KiB
Python
#
|
|
# NcModelCycleQuery
|
|
#
|
|
# This code has been developed by the SIB for use in the AWIPS2 system.
|
|
# Scans dataLocation directory and returns all the available cycles.
|
|
#
|
|
# Usage:
|
|
# import NcModelCycleQuery
|
|
# query = NcModelCycleQuery.NcModelCycleQuery()
|
|
# query.setDataLocation("$MODEL/gfs")
|
|
# return query.execute()
|
|
#
|
|
# SOFTWARE HISTORY
|
|
#
|
|
# Date Ticket# Engineer Description
|
|
# ------------ ---------- ----------- --------------------------
|
|
# 10/22/10 ?? mgamazaychikov Initial Creation
|
|
#
|
|
from java.util import ArrayList
|
|
from com.raytheon.uf.common.message.response import ResponseMessageGeneric
|
|
from gov.noaa.nws.ncep.edex.uengine.utility import GempakConvert
|
|
import os
|
|
|
|
|
|
class NcModelCycleQuery():
|
|
|
|
def __init__(self):
|
|
self.convert = GempakConvert()
|
|
|
|
#
|
|
# Set parameters of the metadata catalog query
|
|
#
|
|
def setDataLocation(self, aDataLocation):
|
|
dirLevels = aDataLocation.split("/")
|
|
numberDirLevel = len(dirLevels)
|
|
if aDataLocation.find("$") == 0:
|
|
envName = dirLevels[0].split("$")[1]
|
|
location = os.environ [envName]
|
|
for ii in range (1,numberDirLevel -1):
|
|
location = location + "/" + dirLevels[ii]
|
|
model = dirLevels[numberDirLevel-1].lower()
|
|
else:
|
|
location = "/".join(dirLevels[0:numberDirLevel-1])
|
|
model = dirLevels[numberDirLevel-1].lower()
|
|
self.dataDir = location + "/" + model
|
|
|
|
#
|
|
# Executes the query and calls appropriate response functions
|
|
#
|
|
def execute(self):
|
|
cycles=[]
|
|
for fn in os.listdir (self.dataDir):
|
|
cycle = fn.split('_')[1].split('f')[0][2:8] + "_" + fn.split('_')[1].split('f')[0][8:]+ '00'
|
|
if cycle not in cycles:
|
|
cycles.append(cycle)
|
|
cycles.sort()
|
|
if cycles is None:
|
|
return self.makeNullResponse("No cycle times available")
|
|
else:
|
|
return self.__makeResponse(cycles)
|
|
#
|
|
# Generates a list of responses in XML format
|
|
#
|
|
def __makeResponse(self, cycles):
|
|
retStr = "|".join(cycles)
|
|
return ResponseMessageGeneric(retStr)
|
|
|
|
#
|
|
# Returns a string with a response message
|
|
#
|
|
def __makeNullResponse(self, aMessage="Database Query returned no results"):
|
|
return ResponseMessageGeneric(aMessage) |