def testGetCatchmentFeaturesForStreamflowGage(self):
goodReachcode = "02060003000745"
goodMeasure = 75.06239
badReachcode = "klsdklasd"
badMeasure = "sdfklfkl"
# Bad reachcode
exceptionThrown = False
try:
(shapefileName, url) = getCatchmentFeaturesForStreamflowGage(
self.context.config, self.projectDir, "catchment", badReachcode, goodMeasure
)
except WebserviceError:
exceptionThrown = True
except Exception:
raise
self.assertTrue(exceptionThrown)
# Bad measure
exceptionThrown = False
try:
(shapefileName, url) = getCatchmentFeaturesForStreamflowGage(
self.context.config, self.projectDir, "catchment", goodReachcode, badMeasure
)
except WebserviceError:
exceptionThrown = True
except Exception:
raise
self.assertTrue(exceptionThrown)
# Good case
exceptionThrown = False
try:
(shapefileName, url) = getCatchmentFeaturesForStreamflowGage(
self.context.config, self.projectDir, "catchment", goodReachcode, goodMeasure
)
except:
traceback.print_exc(file=sys.stdout)
exceptionThrown = True
self.assertTrue(not exceptionThrown)
self.assertTrue(shapefileName == "catchment.shp")
self.assertTrue(os.path.exists(os.path.join(self.projectDir, shapefileName)))
def testGetCatchmentFeaturesForStreamflowGage(self):
goodReachcode = "02060003000745"
goodMeasure = 75.06239
badReachcode = "klsdklasd"
badMeasure = "sdfklfkl"
# Bad reachcode
exceptionThrown = False
try:
(shapefileName, url) = getCatchmentFeaturesForStreamflowGage(self.context.config, self.projectDir,
'catchment', badReachcode, goodMeasure)
except WebserviceError:
exceptionThrown = True
except Exception:
raise
self.assertTrue( exceptionThrown )
# Bad measure
exceptionThrown = False
try:
(shapefileName, url) = getCatchmentFeaturesForStreamflowGage(self.context.config, self.projectDir,
'catchment', goodReachcode, badMeasure)
except WebserviceError:
exceptionThrown = True
except Exception:
raise
self.assertTrue( exceptionThrown )
# Good case
exceptionThrown = False
try:
(shapefileName, url) = getCatchmentFeaturesForStreamflowGage(self.context.config, self.projectDir,
'catchment', goodReachcode, goodMeasure)
except:
traceback.print_exc(file=sys.stdout)
exceptionThrown = True
self.assertTrue( not exceptionThrown )
self.assertTrue( shapefileName == 'catchment.shp' )
self.assertTrue( os.path.exists( os.path.join( self.projectDir, shapefileName ) ) )
sys.exit("Config file %s does not define option %s in section %s" % \
(args.configfile, 'NHDPLUS2', 'PATH_OF_NHDPLUS2_CATCHMENT'))
shapeFilename = getCatchmentFeaturesForGage(context.config, context.projectDir, outfile,
reachcode, measure,
format=OGR_SHAPEFILE_DRIVER_NAME)
source = 'http://www.horizon-systems.com/NHDPlus/NHDPlusV2_home.php'
writeMetadata = True
sys.stdout.write('done\n')
else:
sys.stdout.write('Geting catchment area draining through gage using NHDPlus webservice...')
sys.stdout.flush()
try:
(shapeFilename, source) = getCatchmentFeaturesForStreamflowGage(context.config, context.projectDir,
outfile, reachcode, measure,
format=OGR_SHAPEFILE_DRIVER_NAME)
writeMetadata = True
except Exception as e:
sys.exit( str(e) )
sys.stdout.write('done\n')
if writeMetadata:
# Write provenance
asset = AssetProvenance(GenericMetadata.MANIFEST_SECTION)
asset.name = 'study_area_shapefile'
asset.dcIdentifier = shapeFilename
asset.dcSource = source
asset.dcTitle = 'Study area shapefile'
asset.dcPublisher = 'USGS'
measure,
format=OGR_SHAPEFILE_DRIVER_NAME)
source = 'http://www.horizon-systems.com/NHDPlus/NHDPlusV2_home.php'
writeMetadata = True
sys.stdout.write('done\n')
else:
sys.stdout.write(
'Geting catchment area draining through gage using NHDPlus webservice...'
)
sys.stdout.flush()
try:
(shapeFilename, source) = getCatchmentFeaturesForStreamflowGage(
context.config,
context.projectDir,
outfile,
reachcode,
measure,
format=OGR_SHAPEFILE_DRIVER_NAME)
writeMetadata = True
except Exception as e:
sys.exit(str(e))
sys.stdout.write('done\n')
if writeMetadata:
# Write provenance
asset = AssetProvenance(GenericMetadata.MANIFEST_SECTION)
asset.name = 'study_area_shapefile'
asset.dcIdentifier = shapeFilename
asset.dcSource = source
