ADLongwell-usgs-waterdata 0.1.0 → 0.2.0

data/lib/usgs/gauge.rb

@@ -2,6 +2,7 @@ module USGS
 
   require 'open-uri'
   require 'date'
+  require 'rexml/document'
 
   class Gauge
 
@@ -69,6 +70,18 @@ module USGS
       @daily_mean_flows
     end
 
+    # Return the latitude and longitude for this gauge
+    def get_lat_lon
+      populate_site_data if @lat_lon.nil? or @lat_lon.empty?
+      @lat_lon
+    end
+
+    # Return the site name for this gauge
+    def get_site_name
+      populate_site_data if @site_name.nil? or @site_name.empty?
+      @site_name
+    end
+
     private
 
     # Populate daily data for a gauge from the USGS site
@@ -89,7 +102,6 @@ module USGS
       begin_dt, end_dt = end_dt, begin_dt if end_dt < begin_dt
 
       url = "http://waterdata.usgs.gov/nwis/dv?site_no=#{@site_number}&cb_00060=on&begin_date=#{begin_dt.strftime("%Y-%m-%d")}&end_date=#{end_dt.strftime("%Y-%m-%d")}&format=rdb"
-      puts url
       mean_flows_hash = {}  
       open(url).each do |line|
         next if line =~ /^#/
@@ -165,6 +177,19 @@ module USGS
       @statistical_percentile20_flows = percentile20_flows_hash.sort
       @statistical_percentile80_flows = percentile80_flows_hash.sort
     end
+
+    def populate_site_data
+      
+      url = "http://waterservices.usgs.gov/NWISQuery/GetDV1?SiteNum=#{@site_number}&ParameterCode=00060&StatisticCode=00003&StartDate=2000-11-16&EndDate=2000-11-17&action=Submit"
+      doc = REXML::Document.new(open(url))
+
+      @lat_lon = [0.00, 0.00]
+      REXML::XPath.each(doc, "//latitude") {|el| @lat_lon[0] = el.text.to_f}
+      REXML::XPath.each(doc, "//longitude") {|el| @lat_lon[1] = el.text.to_f}
+
+      @site_name = ""
+      REXML::XPath.each(doc, "//siteName") {|el| @site_name = el.text}
+    end
   end
 end
 

data/spec/gauge_spec.rb

@@ -42,7 +42,7 @@ describe USGS::Gauge do
     gauge = USGS::Gauge.new(13336500)
     puts "Latest flow for the Selway River is #{gauge.latest_flow} CFS"
   end
-  
+
   it 'should get daily mean flows for a river' do
     gauge = USGS::Gauge.new(13336500)
     daily_mean_flows = gauge.get_daily_mean_flows
@@ -106,7 +106,7 @@ describe USGS::Gauge do
     end
   end
 
-   it 'should handle a date that has not yet occured' do
+  it 'should handle a date that has not yet occured' do
     gauge = USGS::Gauge.new(13336500)
     gauge.should_not be_nil
     begin_dt = Date.new(Date.today.year, Date.today.month + 1, Date.today.day)
@@ -119,7 +119,7 @@ describe USGS::Gauge do
     end
   end
 
-     it 'should handle a out-of-range, switched data ' do
+  it 'should handle out-of-range, switched data ' do
     gauge = USGS::Gauge.new(13336500)
     gauge.should_not be_nil
     begin_dt = Date.new(Date.today.year, Date.today.month + 1, Date.today.day)
@@ -133,5 +133,12 @@ describe USGS::Gauge do
     end
   end
 
+  it 'should obtain site information' do
+    gauge = USGS::Gauge.new(13336500)
+    latlon = gauge.get_lat_lon
+    site_name = gauge.get_site_name
+    puts "The latitude and longitude for the gauge at #{site_name} is #{latlon[0]}, #{latlon[1]}"
+  end
+
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: ADLongwell-usgs-waterdata
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Aaron Longwell