mirror of
https://github.com/Unidata/python-awips.git
synced 2025-02-23 22:57:56 -05:00
82 lines
3 KiB
Python
82 lines
3 KiB
Python
# #
|
|
# #
|
|
|
|
#
|
|
# Method for performing a DAF time query where all parameter/level/location
|
|
# combinations must be available at the same time.
|
|
#
|
|
#
|
|
#
|
|
# SOFTWARE HISTORY
|
|
#
|
|
# Date Ticket# Engineer Description
|
|
# ------------ ---------- ----------- --------------------------
|
|
# 06/22/16 #5591 bsteffen Initial Creation.
|
|
#
|
|
|
|
from awips.dataaccess import DataAccessLayer
|
|
|
|
def getAvailableTimes(request, refTimeOnly=False):
|
|
return __getAvailableTimesForEachParameter(request, refTimeOnly)
|
|
|
|
def __getAvailableTimesForEachParameter(request, refTimeOnly=False):
|
|
parameters = request.getParameters()
|
|
if parameters:
|
|
times = None
|
|
for parameter in parameters:
|
|
specificRequest = __cloneRequest(request)
|
|
specificRequest.setParameters(parameter)
|
|
specificTimes = __getAvailableTimesForEachLevel(specificRequest, refTimeOnly)
|
|
if times is None:
|
|
times = specificTimes
|
|
else:
|
|
times.intersection_update(specificTimes)
|
|
if not times:
|
|
break
|
|
return times
|
|
else:
|
|
return __getAvailableTimesForEachLevel(request, refTimeOnly)
|
|
|
|
def __getAvailableTimesForEachLevel(request, refTimeOnly=False):
|
|
levels = request.getLevels()
|
|
if levels:
|
|
times = None
|
|
for level in levels:
|
|
specificRequest = __cloneRequest(request)
|
|
specificRequest.setLevels(level)
|
|
specificTimes = __getAvailableTimesForEachLocation(specificRequest, refTimeOnly)
|
|
if times is None:
|
|
times = specificTimes
|
|
else:
|
|
times.intersection_update(specificTimes)
|
|
if not times:
|
|
break
|
|
return times
|
|
else:
|
|
return __getAvailableTimesForEachLocation(request, refTimeOnly)
|
|
|
|
def __getAvailableTimesForEachLocation(request, refTimeOnly=False):
|
|
locations = request.getLocationNames()
|
|
if locations:
|
|
times = None
|
|
for location in locations:
|
|
specificRequest = __cloneRequest(request)
|
|
specificRequest.setLocationNames(location)
|
|
specificTimes = DataAccessLayer.getAvailableTimes(specificRequest, refTimeOnly)
|
|
if times is None:
|
|
times = set(specificTimes)
|
|
else:
|
|
times.intersection_update(specificTimes)
|
|
if not times:
|
|
break
|
|
return times
|
|
else:
|
|
return DataAccessLayer.getAvailableTimes(request, refTimeOnly)
|
|
|
|
def __cloneRequest(request):
|
|
return DataAccessLayer.newDataRequest(datatype = request.getDatatype(),
|
|
parameters = request.getParameters(),
|
|
levels = request.getLevels(),
|
|
locationNames = request.getLocationNames(),
|
|
envelope = request.getEnvelope(),
|
|
**request.getIdentifiers())
|