anp_prices 1.0 → 1.01

data/VERSION

@@ -1 +1 @@
-1.0
+1.01

data/anp_prices.gemspec

@@ -1,7 +1,7 @@
 # -*- encoding: utf-8 -*-
 Gem::Specification.new do |s|
   s.authors = ["Rogério Zambon"]
-  s.date = "2013-01-10"
+  s.date = "2013-01-12"
   s.description = "Get fuel prices of the ANP - Agência Nacional do Petróleo, Gás Natural e Biocombustíveis"
   s.email = "rogeriozambon@gmail.com"
   s.add_dependency "rest-client", ">= 1.6.7"
@@ -18,5 +18,5 @@ Gem::Specification.new do |s|
   s.require_paths = ["lib"]
   s.required_rubygems_version = Gem::Requirement.new(">= 1.8.23")
   s.summary = s.description
-  s.version = "1.0"
+  s.version = "1.01"
 end

data/lib/anp_prices.rb

@@ -7,7 +7,7 @@ class ANP
   def initialize(city = nil, fuel = nil)
     raise ArgumentError if city.nil? or fuel.nil?
 
-    @city = city
+    @city = find_code_of(city)
     @fuel = fuel
   end
 
@@ -50,6 +50,22 @@ class ANP
     Nokogiri::HTML(req.body).xpath "//div[@class='multi_box3']/table[@class='table_padrao scrollable_table']"
   end
 
+  def request_city(name)
+    req = RestClient.post "http://www.anp.gov.br/preco/prc/Resumo_Ultimas_Coletas_Index.asp", :txtMunicipio => name
+
+    Nokogiri::HTML(req.body).xpath "//div[@id='divMunicipios']/select[@name='selMunicipio']/option"
+  end
+
+  def find_code_of(name)
+    request_city(name).each do |node|
+      node.each_with_index do |code, index|
+        next if index == 0
+
+        return code[1]
+      end
+    end
+  end
+
   def push_values_with(lines)
     entry = []
     entries = []

data/spec/anp_prices_spec.rb

@@ -5,12 +5,26 @@ describe ANP do
     expect { anp = ANP.new }.to raise_error(ArgumentError)
   end
 
+  context "Find city code" do
+    before do
+      fixture = File.open("spec/fixture/city_sao_paulo.html").read
+      parser = Nokogiri::HTML(fixture).xpath "//div[@id='divMunicipios']/select[@name='selMunicipio']/option"
+
+      @anp = ANP.new "Sao Paulo", :gasolina
+      @anp.stub(:request_city).and_return(parser)
+    end
+
+    it "check city code" do
+      @anp.city.should == "9668*SAO@PAULO"
+    end
+  end
+
   context "Informations" do
     before do
       fixture = File.open("spec/fixture/anp_sao_paulo.html").read
       parser = Nokogiri::HTML(fixture).xpath("//div[@class='multi_box3']/table[@class='table_padrao scrollable_table']")
 
-      @anp = ANP.new "9668*SAO@PAULO", :gasolina
+      @anp = ANP.new "Sao Paulo", :gasolina
       @anp.stub(:request_page).and_return(parser)
     end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: anp_prices
 version: !ruby/object:Gem::Version
-  version: '1.0'
+  version: '1.01'
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-10 00:00:00.000000000 Z
+date: 2013-01-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rest-client