def create_get_site_response(self, siteArg=None):
if siteArg == None or siteArg == '':
siteResultArr = self.dao.get_all_sites()
else:
siteCodesArr = siteArg.split(',')
siteCodesArr = [
s.replace(self.network + ':', '') for s in siteCodesArr
]
siteResultArr = self.dao.get_sites_by_codes(siteCodesArr)
if len(siteResultArr) == 0:
return None
siteInfoResponse = WaterML.SiteInfoResponseType()
queryInfo = WaterML.QueryInfoType()
#TODO: check on how this should be done for multiple sites
criteria = WaterML.criteria(locationParam=siteArg)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
siteInfoResponse.set_queryInfo(queryInfo)
for siteResult in siteResultArr:
s = self.create_site_element(siteResult)
siteInfoResponse.add_site(s)
return siteInfoResponse
def create_get_site_info_response(self, siteArg, varArg=None):
siteCode = siteArg.replace(self.network + ':', '')
siteResult = self.dao.get_site_by_code(siteCode)
if (varArg == None or varArg == ''):
seriesResultArr = self.dao.get_series_by_sitecode(siteCode)
else:
varCode = varArg.replace(self.vocabulary + ':', '')
seriesResultArr = self.dao.get_series_by_sitecode_and_varcode(
siteCode, varCode)
if len(seriesResultArr) == 0:
return None
siteInfoResponse = WaterML.SiteInfoResponseType()
queryInfo = WaterML.QueryInfoType()
criteria = WaterML.criteria(locationParam=siteArg,
variableParam=varArg)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
siteInfoResponse.set_queryInfo(queryInfo)
s = self.create_site_element(siteResult, seriesResultArr)
siteInfoResponse.add_site(s)
return siteInfoResponse
def create_get_site_info_response(self, siteArg, varArg=None):
siteCode = siteArg.replace(self.network + ':', '')
siteResult = self.dao.get_site_by_code(siteCode)
if (varArg == None or varArg == ''):
seriesResultArr = self.dao.get_series_by_sitecode(siteCode)
else:
varCode = varArg.replace(self.vocabulary + ':', '')
seriesResultArr = self.dao.get_series_by_sitecode_and_varcode(
siteCode, varCode)
if len(seriesResultArr) == 0:
return None
siteInfoResponse = WaterML.SiteInfoResponseType()
queryInfo = WaterML.QueryInfoType()
criteria = WaterML.criteria(locationParam=siteArg,
variableParam=varArg)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
siteInfoResponse.set_queryInfo(queryInfo)
s = self.create_site_element(siteResult, seriesResultArr)
siteInfoResponse.add_site(s)
return siteInfoResponse
def create_get_site_response(self, siteArg=None):
if siteArg == None or siteArg == '':
siteResultArr = self.dao.get_all_sites()
else:
siteCodesArr = siteArg.split(',')
siteCodesArr = [s.replace(self.network + ':', '')
for s in siteCodesArr]
siteResultArr = self.dao.get_sites_by_codes(siteCodesArr)
if len(siteResultArr) == 0:
return None
siteInfoResponse = WaterML.SiteInfoResponseType()
queryInfo = WaterML.QueryInfoType()
#TODO: check on how this should be done for multiple sites
criteria = WaterML.criteria(locationParam=siteArg)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
siteInfoResponse.set_queryInfo(queryInfo)
for siteResult in siteResultArr:
s = self.create_site_element(siteResult)
siteInfoResponse.add_site(s)
return siteInfoResponse
def create_get_site_response(self, siteArg=None):
if siteArg == None or siteArg == '':
siteResultArr = self.dao.get_all_sites()
else:
siteCodesArr = siteArg.split(',')
siteCodesArr = [self.get_site_code(s)
for s in siteCodesArr]
siteResultArr = self.dao.get_sites_by_codes(siteCodesArr)
#if len(siteResultArr) == 0:
# return None
siteInfoResponse = WaterML.SiteInfoResponseType()
queryInfo = WaterML.QueryInfoType()
#TODO: check on how this should be done for multiple sites
criteria = WaterML.criteria(locationParam=siteArg)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
siteInfoResponse.set_queryInfo(queryInfo)
if siteResultArr:
for siteResult in siteResultArr:
s = self.create_site_element(siteResult)
siteInfoResponse.add_site(s)
else:
#site = WaterML.site()
#siteInfoResponse.add_site(site)
raise Exception("Site,'%s', Not Found" % siteArg)
return siteInfoResponse
def create_get_site_info_response(self, siteArg, varArg=None):
if siteArg is None:
raise Exception("Site Not Found")
siteCode = self.get_site_code(siteArg)
siteResult = self.dao.get_site_by_code(siteCode)
if (varArg == None or varArg == ''):
seriesResultArr = self.dao.get_series_by_sitecode(siteCode)
else:
varCode = self.get_variable_code(varArg)
seriesResultArr = self.dao.get_series_by_sitecode_and_varcode(
siteCode, varCode)
if seriesResultArr is None or len(seriesResultArr) == 0:
raise Exception("Site,'%s', Not Found" % siteArg)
siteInfoResponse = WaterML.SiteInfoResponseType()
queryInfo = WaterML.QueryInfoType()
criteria = WaterML.criteria(locationParam=siteArg,
variableParam=varArg)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
siteInfoResponse.set_queryInfo(queryInfo)
if siteResult:
s = self.create_site_element(siteResult, seriesResultArr)
siteInfoResponse.add_site(s)
else:
#site = WaterML.site()
#siteInfoResponse.add_site(site)
raise Exception("Site,'%s', Not Found" % siteArg)
return siteInfoResponse
def create_get_values_response(self,
siteArg,
varArg,
startDateTime=None,
endDateTime=None):
#TODO: Tim thinks the DAO should handle network and vocab parsing,
# not WOF
siteCode = siteArg.replace(self.network + ':', '')
varCode = varArg.replace(self.vocabulary + ':', '')
valueResultArr = self.dao.get_datavalues(siteCode, varCode,
startDateTime, endDateTime)
if not valueResultArr:
raise Exception(
"ERROR: No data found for %s:%s for dates %s - %s" %
(siteCode, varCode, startDateTime, endDateTime))
timeSeriesResponse = WaterML.TimeSeriesResponseType()
queryInfo = WaterML.QueryInfoType()
timeParam = WaterML.timeParam(beginDateTime=startDateTime,
endDateTime=endDateTime)
criteria = WaterML.criteria(locationParam=siteArg,
variableParam=varArg,
timeParam=timeParam)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
timeSeriesResponse.set_queryInfo(queryInfo)
timeSeries = WaterML.TimeSeriesType()
#sourceInfo (which is a siteInfo) element
siteResult = self.dao.get_site_by_code(siteCode)
#TODO: Exception?
if not siteResult:
pass
sourceInfo = self.create_site_info_element(siteResult)
timeSeries.sourceInfo = sourceInfo
#variable element
varResult = self.dao.get_variable_by_code(varCode)
#TODO: Exception?
if not varResult:
pass
variable = self.create_variable_element(varResult)
timeSeries.variable = variable
#TODO: fill in some more of the attributes in this element
values = WaterML.TsValuesSingleVariableType()
values.count = len(valueResultArr)
if varResult.VariableUnits:
values.unitsAbbreviation = varResult.VariableUnits.UnitsAbbreviation
values.unitsCode = varResult.VariableUnits.UnitsID
#Need to keep track of unique methodIDs and sourceIDs
methodIdSet = set()
sourceIdSet = set()
qualifierIdSet = set()
offsetTypeIdSet = set()
for valueResult in valueResultArr:
v = self.create_value_element(valueResult)
values.add_value(v)
if valueResult.MethodID:
methodIdSet.add(valueResult.MethodID)
if valueResult.SourceID:
sourceIdSet.add(valueResult.SourceID)
if valueResult.QualifierID:
qualifierIdSet.add(valueResult.QualifierID)
if valueResult.OffsetTypeID:
offsetTypeIdSet.add(valueResult.OffsetTypeID)
#Add method elements for each unique methodID
if methodIdSet:
methodIdArr = list(methodIdSet)
methodResultArr = self.dao.get_methods_by_ids(methodIdArr)
for methodResult in methodResultArr:
method = self.create_method_element(methodResult)
values.add_method(method)
#Add source elements for each unique sourceID
if sourceIdSet:
sourceIdArr = list(sourceIdSet)
sourceResultArr = self.dao.get_sources_by_ids(sourceIdArr)
for sourceResult in sourceResultArr:
source = self.create_source_element(sourceResult)
values.add_source(source)
#Add qualifier elements
if qualifierIdSet:
qualIdArr = list(qualifierIdSet)
qualResultArr = self.dao.get_qualifiers_by_ids(qualIdArr)
for qualifierResult in qualResultArr:
q = WaterML.qualifier(
qualifierID=qualifierResult.QualifierID,
default=None,
network=self.network,
vocabulary=self.vocabulary,
qualifierCode=qualifierResult.QualifierCode)
values.add_qualifier(q)
#Add offset elements
if offsetTypeIdSet:
offsetTypeIdArr = list(offsetTypeIdSet)
offsetTypeResultArr = self.dao.get_offsettypes_by_ids(
offsetTypeIdArr)
for offsetTypeResult in offsetTypeResultArr:
offset = self.create_offset_element(offsetTypeResult)
values.add_offset(offset)
timeSeries.values = values
timeSeriesResponse.set_timeSeries(timeSeries)
return timeSeriesResponse
def create_get_values_response(self, siteArg, varArg, startDateTime=None,
endDateTime=None):
#TODO: Tim thinks the DAO should handle network and vocab parsing,
# not WOF
siteCode = siteArg.replace(self.network + ':', '')
varCode = varArg.replace(self.vocabulary + ':', '')
valueResultArr = self.dao.get_datavalues(siteCode, varCode,
startDateTime, endDateTime)
if not valueResultArr:
raise Exception("ERROR: No data found for %s:%s for dates %s - %s" % (
siteCode, varCode, startDateTime, endDateTime))
timeSeriesResponse = WaterML.TimeSeriesResponseType()
queryInfo = WaterML.QueryInfoType()
timeParam = WaterML.timeParam(
beginDateTime=startDateTime, endDateTime=endDateTime)
criteria = WaterML.criteria(
locationParam=siteArg, variableParam=varArg, timeParam=timeParam)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
timeSeriesResponse.set_queryInfo(queryInfo)
timeSeries = WaterML.TimeSeriesType()
#sourceInfo (which is a siteInfo) element
siteResult = self.dao.get_site_by_code(siteCode)
#TODO: Exception?
if not siteResult:
pass
sourceInfo = self.create_site_info_element(siteResult)
timeSeries.sourceInfo = sourceInfo
#variable element
varResult = self.dao.get_variable_by_code(varCode)
#TODO: Exception?
if not varResult:
pass
variable = self.create_variable_element(varResult)
timeSeries.variable = variable
#TODO: fill in some more of the attributes in this element
values = WaterML.TsValuesSingleVariableType()
values.count = len(valueResultArr)
if varResult.VariableUnits:
values.unitsAbbreviation = varResult.VariableUnits.UnitsAbbreviation
values.unitsCode = varResult.VariableUnits.UnitsID
#Need to keep track of unique methodIDs and sourceIDs
methodIdSet = set()
sourceIdSet = set()
qualifierIdSet = set()
offsetTypeIdSet = set()
for valueResult in valueResultArr:
v = self.create_value_element(valueResult)
values.add_value(v)
if valueResult.MethodID:
methodIdSet.add(valueResult.MethodID)
if valueResult.SourceID:
sourceIdSet.add(valueResult.SourceID)
if valueResult.QualifierID:
qualifierIdSet.add(valueResult.QualifierID)
if valueResult.OffsetTypeID:
offsetTypeIdSet.add(valueResult.OffsetTypeID)
#Add method elements for each unique methodID
if methodIdSet:
methodIdArr = list(methodIdSet)
methodResultArr = self.dao.get_methods_by_ids(methodIdArr)
for methodResult in methodResultArr:
method = self.create_method_element(methodResult)
values.add_method(method)
#Add source elements for each unique sourceID
if sourceIdSet:
sourceIdArr = list(sourceIdSet)
sourceResultArr = self.dao.get_sources_by_ids(sourceIdArr)
for sourceResult in sourceResultArr:
source = self.create_source_element(sourceResult)
values.add_source(source)
#Add qualifier elements
if qualifierIdSet:
qualIdArr = list(qualifierIdSet)
qualResultArr = self.dao.get_qualifiers_by_ids(qualIdArr)
for qualifierResult in qualResultArr:
q = WaterML.qualifier(
qualifierID=qualifierResult.QualifierID,
default=None,
network=self.network,
vocabulary=self.vocabulary,
qualifierCode=qualifierResult.QualifierCode)
values.add_qualifier(q)
#Add offset elements
if offsetTypeIdSet:
offsetTypeIdArr = list(offsetTypeIdSet)
offsetTypeResultArr = self.dao.get_offsettypes_by_ids(
offsetTypeIdArr)
for offsetTypeResult in offsetTypeResultArr:
offset = self.create_offset_element(offsetTypeResult)
values.add_offset(offset)
timeSeries.values = values
timeSeriesResponse.set_timeSeries(timeSeries)
return timeSeriesResponse
def create_get_values_response(self, siteArg, varArg, startDateTime=None,
endDateTime=None):
#TODO: Tim thinks the DAO should handle network and vocab parsing,
# not WOF
siteCode = self.get_site_code(siteArg)
varCode = self.get_variable_code(varArg)
valueResultArr = self.dao.get_datavalues(siteCode, varCode,
startDateTime, endDateTime)
timeSeriesResponse = WaterML.TimeSeriesResponseType()
queryInfo = WaterML.QueryInfoType()
timeParam = WaterML.timeParam(
beginDateTime=startDateTime, endDateTime=endDateTime)
criteria = WaterML.criteria(
locationParam=siteArg, variableParam=varArg, timeParam=timeParam)
queryInfo.set_criteria(criteria)
queryInfoNote = WaterML.NoteType()
queryInfo.add_note(queryInfoNote)
queryInfo.set_extension('')
timeSeriesResponse.set_queryInfo(queryInfo)
#if not valueResultArr:
# timeSeries = WaterML.TimeSeriesType()
# timeSeriesResponse.set_timeSeries(timeSeries)
# return timeSeriesResponse
if not valueResultArr:
raise Exception("Values Not Found for %s:%s for dates %s - %s" % (
siteCode, varCode, startDateTime, endDateTime))
if isinstance(valueResultArr,dict):
for key in valueResultArr.keys():
valueResultArr = valueResultArr[key]
break
timeSeries = WaterML.TimeSeriesType()
#sourceInfo (which is a siteInfo) element
siteResult = self.dao.get_site_by_code(siteCode)
#TODO: Exception?
if not siteResult:
pass
sourceInfo = self.create_site_info_element(siteResult)
timeSeries.sourceInfo = sourceInfo
#variable element
varResult = self.dao.get_variable_by_code(varCode)
#TODO: Exception?
if not varResult:
pass
variable = self.create_variable_element(varResult)
timeSeries.variable = variable
#TODO: fill in some more of the attributes in this element
values = WaterML.TsValuesSingleVariableType()
values.count = len(valueResultArr)
if varResult.VariableUnits:
values.unitsAbbreviation = varResult.VariableUnits.UnitsAbbreviation
values.unitsCode = varResult.VariableUnits.UnitsID
#Need to keep track of unique methodIDs and sourceIDs
methodIdSet = set()
sourceIdSet = set()
qualifierIdSet = set()
offsetTypeIdSet = set()
qualitycontrollevelIdSet = set()
for valueResult in valueResultArr:
if valueResult.QualityControlLevelID is not None:
qualitycontrollevelIdSet.add(valueResult.QualityControlLevelID)
qlevelResult = self.dao.get_qualcontrollvl_by_id(valueResult.QualityControlLevelID)
if hasattr(qlevelResult,'Definition'):
valueResult.QualityControlLevel = qlevelResult.Definition
# else:
# if hasattr(valueResult,'QualityControlLevel'):
# valueResult.QualityControlLevel = qlevelResult.QualityControlLevelCode
v = self.create_value_element(valueResult)
values.add_value(v)
if valueResult.MethodID is not None:
methodIdSet.add(valueResult.MethodID)
if valueResult.SourceID is not None:
sourceIdSet.add(valueResult.SourceID)
if valueResult.QualifierID is not None:
qualifierIdSet.add(valueResult.QualifierID)
if valueResult.OffsetTypeID is not None:
offsetTypeIdSet.add(valueResult.OffsetTypeID)
if valueResult.QualityControlLevelID is not None:
qualitycontrollevelIdSet.add(valueResult.QualityControlLevelID)
#Add method elements for each unique methodID
if methodIdSet:
methodIdArr = list(methodIdSet)
methodResultArr = self.dao.get_methods_by_ids(methodIdArr)
for methodResult in methodResultArr:
method = self.create_method_element(methodResult)
values.add_method(method)
#Add source elements for each unique sourceID
if sourceIdSet:
sourceIdArr = list(sourceIdSet)
sourceResultArr = self.dao.get_sources_by_ids(sourceIdArr)
for sourceResult in sourceResultArr:
source = self.create_source_element(sourceResult)
values.add_source(source)
#Add qualifier elements
if qualifierIdSet:
qualIdArr = list(qualifierIdSet)
qualResultArr = self.dao.get_qualifiers_by_ids(qualIdArr)
for qualifierResult in qualResultArr:
q = WaterML.qualifier(
qualifierID=qualifierResult.QualifierID,
default=None,
network=self.network,
vocabulary=self.vocabulary,
qualifierCode=qualifierResult.QualifierCode)
values.add_qualifier(q)
#Add offset elements
if offsetTypeIdSet:
offsetTypeIdArr = list(offsetTypeIdSet)
offsetTypeResultArr = self.dao.get_offsettypes_by_ids(
offsetTypeIdArr)
for offsetTypeResult in offsetTypeResultArr:
offset = self.create_offset_element(offsetTypeResult)
values.add_offset(offset)
#Add qualitycontrollevel elements
if qualitycontrollevelIdSet:
qlevelIdIdArr = list(qualitycontrollevelIdSet)
try:
qlevelResultArr = self.dao.get_qualcontrollvls_by_ids(qlevelIdIdArr)
for qlevelResult in qlevelResultArr:
qlevel = self.create_qlevel_element(qlevelResult)
values.add_qualityControlLevel(qlevel)
except:
logging.warn("WofPy: DOA has no get_qualcontrollvls_by_ids method (added for 2.x)")
for qlevelID in qlevelIdIdArr:
qlevel = WaterML.QualityControlLevelType(
qualityControlLevelID=qlevelID )
values.add_qualityControlLevel(qlevel)
timeSeries.values = values
timeSeriesResponse.set_timeSeries(timeSeries)
return timeSeriesResponse
