ceps 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8b3a7b380c76920d10a307a1e23f28bdf5d2a407
-  data.tar.gz: 70d587c1ed83c16db19eafb6977d0a636ac8690e
+  metadata.gz: 4a8733b593bf45104acf53df0259b11255014a96
+  data.tar.gz: 79641e0bd8e6f86a3baabf597905dce309e6936d
 SHA512:
-  metadata.gz: c8a6379551b631c0c6b74b2f0538ac7ce329f195042fd510a3a82cbfa490f1be4d2243bc00cd0b757f11dff46a0dbbaa98b67030492029745026aad5e50df97f
-  data.tar.gz: e8c19492b98dc4c912ff11df30b04493bc00c0c04030d22c945e5b86af6ca661988faf771fbdb94fd85d426244e2052163447d7aca79fc90dfdde79fd1df9de6
+  metadata.gz: 046101f7a4f959c219334bc345a1181478d44d97bedbc0adcbcd66b1f10a7750556a41b78ce5c6e5ed1a27c2dbbe2df90259229c7d13c34182607802495b65ca
+  data.tar.gz: c3efb9e8bf6255af3f72156de8e62eac21dda0a7a72248784aab7f5c574bb1cc1a64166ee5a52b7b970d273417f7cd4e5f4a30a15fa12630fa195b95fdee9994

data/Gemfile

@@ -2,8 +2,8 @@ source 'https://rubygems.org'
 gem 'rake'
 gem 'mdb'
 gem 'ruby-progressbar'
-# gem 'geocoder'
-# gem 'retryable'
+gem 'geocoder'
+gem 'retryable'
 
 # Specify your gem's dependencies in countries.gemspec
 gemspec

data/README.md

@@ -4,6 +4,8 @@ CEPs
 Ceps is a collection of all sorts of useful information about every brazilian CEP packaged as pretty little cep objects.
 
 [![Build Status](https://travis-ci.org/theprogramer/ceps.png)](https://travis-ci.org/theprogramer/ceps)
+[![Code Climate](https://codeclimate.com/github/theprogramer/ceps/badges/gpa.svg)](https://codeclimate.com/github/theprogramer/ceps)
+[![Test Coverage](https://codeclimate.com/github/theprogramer/ceps/badges/coverage.svg)](https://codeclimate.com/github/theprogramer/ceps/coverage)
 
 Installation
 ------------

data/Rakefile

@@ -4,52 +4,11 @@ require 'bundler/gem_tasks'
 require 'rake'
 require 'rspec/core/rake_task'
 
+Dir.glob('lib/tasks/*.rake').each { |r| load r}
+
 desc 'Run all examples'
 RSpec::Core::RakeTask.new(:spec) do |t|
   t.rspec_opts = %w(--color --warnings)
 end
 
 task default: [:spec]
-
-
-desc 'Import MDB file'
-task :import do
-
-  require 'mdb'
-  require_relative 'lib/ceps/mdb'
-  require 'ruby-progressbar'
-
-  db = Mdb.open ARGV[1]
-  puts "Loading routes"
-  puts "This should take some time!"
-  routes = db["LOG_LOGRADOURO"]
-  puts "Loading neighborhoods"
-  neighborhoods = db["LOG_BAIRRO"]
-  puts "Loading cities"
-  cities = db["LOG_LOCALIDADE"]
-  puts "Loading states"
-  states = db["LOG_FAIXA_UF"]
-  
-  ceps = {}
-  puts "Generating new hash"
-  progressbar = ProgressBar.create title: "Progress",
-                                   starting_at: 0,
-                                   total: routes.count,
-                                   format: "%c from %C |%b>>%i| %a (%e) %p%%"
-  routes.each do |route|
-    cep = Correios::Mdb.decript route[:LOG_KEY_DNE]
-    ceps[cep.to_sym] = {
-      location: route[:LOG_NO],
-      type: route[:LOG_TIPO_LOGRADOURO],
-      city: cities.select{|c| c[:LOC_NU_SEQUENCIAL] == route[:LOC_NU_SEQUENCIAL].to_s}.first[:LOC_NO],
-      state: states.select{|s| s[:UFE_SG] == route[:UFE_SG]}.first[:UFE_NO],
-      neighborhood: neighborhoods.select{|n| n[:BAI_NU_SEQUENCIAL] == route[:BAI_NU_SEQUENCIAL_INI].to_s}.first[:BAI_NO]
-    }
-    progressbar.increment
-    progressbar.refresh
-  end
-  puts "Saving new ceps.yaml"
-  File.open('ceps.yml','w') do |h| 
-    h.write ceps.to_yaml
-  end
-end

data/lib/ceps/railtie.rb

@@ -0,0 +1,16 @@
+require 'cep'
+require 'rails'
+
+module Ceps
+  class Railtie < Rails::Railtie
+    
+    railtie_name :ceps
+
+    rake_tasks do
+      rake_tasks do
+        Dir[File.join(File.dirname(__FILE__),'tasks/*.rake')].each { |f| load f }
+      end
+    end
+
+  end
+end

data/lib/ceps/version.rb

@@ -1,3 +1,3 @@
 module Ceps
-  VERSION = '0.0.2'
+  VERSION = '0.0.3'
 end

data/lib/ceps.rb

@@ -1,7 +1,40 @@
 require 'ceps/version'
 
 require 'correios'
+require 'ceps/railtie' if defined?(Rails)
 
 class Cep < Correios::Cep
 
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.configure
+    self.configuration ||= Configuration.new
+  end
+
+end
+
+class Configuration
+ 
+  attr_accessor :data_file_name
+
+  attr_accessor :data_file_path
+
+  attr_accessor :data_file
+
+  def initialize
+    @data_file_name = 'ceps.yaml'
+    if defined?(Rails)
+      @data_file_path = File.join(Rails.root, "lib", "data")
+    else
+      @data_file_path = File.join( 
+                          File.dirname(
+    	                    File.dirname(__FILE__)),
+    	                    "lib", "data")
+    end
+
+    @data_file = File.join @data_file_path, @data_file_name
+
+  end
 end

data/lib/tasks/geolocalize.rake

@@ -0,0 +1,41 @@
+namespace :ceps do
+
+  desc "Geolocalize ceps.yml file"
+  task :geolocalize do
+  	require 'geocoder'
+    require 'retryable'
+    require 'yaml'
+    require 'ceps'
+
+    # raise on geocoding errors such as query limit exceeded
+    Geocoder.configure(always_raise: :all)
+    puts "Loading routes"
+    puts "This should take some time!"
+    ceps = YAML.load_file(ARGV[1] || Cep.configure.data_file) || {}
+    puts "Geolocalize new hash"
+    ceps.each do |cep, value|
+      query = "#{ceps[cep][:type]} #{ceps[cep][:location]}, #{ceps[cep][:neighborhood]} - #{ceps[cep][:city]} - #{ceps[cep][:state]} - cep: #{cep}" 
+      begin 
+        Retryable.retryable(tries: 5, sleep: lambda { |n| 2**n }) do
+          geodata = Geocoder.search(query).first
+          if geodata.empty?
+            raise "Empty geodata"
+          else
+            ceps[cep][:latitude] = geodata["location"]["lat"]
+            ceps[cep][:longitude] = geodata["location"]["lng"]
+            puts geodata["location"]
+          end 
+        end
+      rescue => e
+        warn "Attempts exceeded for query #{query}, last error was #{e.message}"
+      end
+    end
+
+    puts "Saving new ceps.yaml"
+    File.open(Cep.configure.data_file,'w') do |h| 
+      h.write ceps.to_yaml
+    end
+
+  end
+
+end

data/lib/tasks/import.rake

@@ -0,0 +1,46 @@
+namespace :ceps do
+
+  desc 'Import MDB file'
+  task :import do
+
+    require 'mdb'
+    require 'yaml'
+    require_relative 'lib/ceps/mdb'
+    require 'ruby-progressbar'
+
+    db = Mdb.open ARGV[1]
+    puts "Loading routes"
+    puts "This should take some time!"
+    routes = db["LOG_LOGRADOURO"]
+    puts "Loading neighborhoods"
+    neighborhoods = db["LOG_BAIRRO"]
+    puts "Loading cities"
+    cities = db["LOG_LOCALIDADE"]
+    puts "Loading states"
+    states = db["LOG_FAIXA_UF"]
+  
+    ceps = {}
+    puts "Generating new hash"
+    progressbar = ProgressBar.create title: "Progress",
+                                     starting_at: 0,
+                                     total: routes.count,
+                                     format: "%c from %C |%b>>%i| %a (%e) %p%%"
+    routes.each do |route|
+      cep = Correios::Mdb.decript route[:LOG_KEY_DNE]
+      ceps[cep.to_sym] = {
+        location: route[:LOG_NO],
+        type: route[:LOG_TIPO_LOGRADOURO],
+        city: cities.select{|c| c[:LOC_NU_SEQUENCIAL] == route[:LOC_NU_SEQUENCIAL].to_s}.first[:LOC_NO],
+        state: states.select{|s| s[:UFE_SG] == route[:UFE_SG]}.first[:UFE_NO],
+        neighborhood: neighborhoods.select{|n| n[:BAI_NU_SEQUENCIAL] == route[:BAI_NU_SEQUENCIAL_INI].to_s}.first[:BAI_NO]
+      }
+      progressbar.increment
+      progressbar.refresh
+    end
+    puts "Saving new ceps.yaml"
+    File.open(Cep.configure.data_file,'w') do |h| 
+      h.write ceps.to_yaml
+    end
+  end
+
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ceps
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Thiago Miranda
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-25 00:00:00.000000000 Z
+date: 2015-07-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -42,10 +42,13 @@ files:
 - lib/ceps.rb
 - lib/ceps/cep.rb
 - lib/ceps/mdb.rb
+- lib/ceps/railtie.rb
 - lib/ceps/setup.rb
 - lib/ceps/version.rb
 - lib/correios.rb
 - lib/data/ceps.yaml
+- lib/tasks/geolocalize.rake
+- lib/tasks/import.rake
 - spec/cep_spec.rb
 - spec/setup_spec.rb
 - spec/spec_helper.rb