mlightner-zip_codes 0.5.0

data/Rakefile

@@ -0,0 +1,100 @@
+require 'rake'
+require 'rake/testtask'
+require 'rake/rdoctask'
+require 'rubygems'
+require 'sqlite3'
+
+DIR = File.dirname(__FILE__) + '/data'
+system("mkdir -p #{DIR}")
+
+desc 'Test the library.'
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'lib'
+  t.pattern = 'test/**/*_test.rb'
+  t.verbose = true
+end
+
+desc "Update gem files list"
+task :gemfiles do
+  file_name = "zip_codes.gemspec"
+  files = `find ./ | grep -v \.git`.each_line.collect { |f| f.gsub(/^\.\//, '').strip }.reject { |e| e =~ /^\s*$/ }
+  file = File.new(file_name)
+  lines = file.readlines
+  file.close
+
+  newf = []
+  lines.each do |line|
+    if line =~ /^\s+s\.files\s*\=/
+      newf.push "  s.files = [\"#{files.join('","')}\"]\n"
+    else
+      newf.push(line.rstrip + "\n")
+    end
+  end
+
+  # Don't write the file if there are no changes
+  file = File.new(file_name,'w')
+  newf.each do |line|
+    file.write(line)
+  end
+  file.close
+end
+
+namespace :zip_codes do
+
+  desc "Reloads and refreshes db and data"
+  task :all => [ :clean, :download, :create_db, :load_data, :index ]
+
+  desc 'Downloads the zip code database'
+  task :download do
+    require 'open-uri'
+    chdir("#{DIR}")
+    system("/usr/bin/env wget http://www.populardata.com/downloads/zip_codes.zip")
+    system("/usr/bin/env unzip -p zip_codes.zip > zip_codes.csv")
+    puts "New zip code database downloaded."
+  end
+
+  desc 'Creates the test database'
+  task :create_db do
+    puts "Creating zip code database..."
+	  require 'csv'
+	  system("rm -f #{DIR}/zip_codes.db")
+	  csv = CSV.open("#{DIR}/zip_codes.csv", "r")
+	  db = SQLite3::Database.new( "#{DIR}/zip_codes.db" )
+	  db.execute("create table zip_codes (
+	    id        INTEGER     NOT NULL PRIMARY KEY,
+	    zip       VARCHAR(9)  NOT NULL,
+	    zip_class       VARCHAR(9)  ,
+	    city      TEXT        ,
+	    county      TEXT       ,
+	    state     VARCHAR(2)  ,
+	    lat       FLOAT       ,
+	    long      FLOAT       
+	  );")
+
+  desc "Loads zip code data"
+  task :load_data
+	  st = db.prepare("insert into zip_codes(zip, lat, long, city, state, county, zip_class) values(?, ?, ?, ?, ?, ?, ?)")
+	  csv = CSV.open("#{DIR}/zip_codes.csv", "r")
+	  csv.each_with_index do |r,i|
+	    puts "Creating zip code #{r[0]} (#{i})"
+	    st.execute(*r)
+	  end
+    puts "Database created."
+  end
+
+  desc 'Creates index on zip codes table'
+  task :index do
+	  db = SQLite3::Database.open( "#{DIR}/zip_codes.db" )
+	  db.execute("create unique index zip_index on zip_codes (zip);") rescue nil
+	  db.execute("create index zip_state_index on zip_coddes (state);") rescue nil
+  end
+
+  desc 'Cleans data dir'
+  task :clean do
+	  File.unlink("#{DIR}/zip_codes.csv") rescue nil
+	  File.unlink("#{DIR}/zip_codes.zip") rescue nil    
+  end
+end
+
+
+task :default => ['zip_codes:all']

data/bin/zipfind

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+require File.dirname(__FILE__) + '/../zip_codes.rb'
+
+zip = ARGV[0].strip
+
+raise "Syntax: $0 <zip_code>" unless zip
+
+if zip !~ /^([\w\-]+)$/
+  raise "Zip is invalid format"
+end
+
+z = ZipCode.new(zip)
+
+if z && z.valid?
+  puts "Results for zip code: #{zip}:"
+  ZipCode.columns.each do |c|
+    next if c == :id
+    next if c == :zip
+    puts "  - #{c.to_s}: #{z.record[c.to_sym]}"
+  end
+else
+  puts "No match for zip code: #{zip}:"
+end

data/lib/zip_code.rb

@@ -0,0 +1,41 @@
+class ZipCode
+
+  attr_accessor :record
+
+  ZIP_COLUMNS.each do |f|
+    attr_accessor f.to_sym
+  end
+
+  def self.db
+    raise "No database availablbe.  Please initialize it with Rake." unless ZIP_DB.kind_of?(SQLite3::Database)
+    ZIP_DB
+  end
+
+  def db
+    self.class.db
+  end
+
+  def self.columns
+    ZIP_COLUMNS
+  end
+
+  def columns
+    self.class.columns
+  end
+
+  def initialize(zipl)
+    row = db.execute( "select * from zip_codes where zip = '#{zipl.to_s}';" ).first
+    return nil unless (row.kind_of?(Array) && row[1].to_s == zipl.to_s)
+    @record = {}
+    columns.each_with_index do |f,i|
+      @record[f.to_sym] = row[i]
+      self.instance_variable_set(:"@#{f.to_s}", row[i])
+    end
+    
+  end
+
+  def valid?
+    record.kind_of?(Hash) && record.has_key?(:zip)
+  end
+
+end

data/test.rb

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'zip_codes'
+
+puts ZipCode.new('94949').inspect

data/zip_codes.gemspec

@@ -0,0 +1,17 @@
+Gem::Specification.new do |s|
+	s.name = %q{zip_codes}
+	s.version = "0.5.0"
+	s.author = "Matt Lightner"
+	s.default_executable = %q{zip_codes}
+	s.description = %q{A library for looking up zip codes and their associated info.}
+	s.email = %q{mlightner@gmail.com}
+	s.executables = ["zipfind"]
+  s.files = ["README","bin","bin/zipfind","lib","lib/zip_code.rb","zip_codes.gemspec","data","data/zip_codes.db","Rakefile","zip_codes.rb","test.rb"]
+	s.has_rdoc = false
+	s.homepage = %q{http://github.com/mlightner/zip_codes}
+	s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "zip_codes", "--main"]
+	s.require_paths = ["lib"]
+	s.summary = %q{A library for looking up zip codes and their associated info.}
+
+  s.add_dependency(%q<sqlite3>, [">= 0.0.0"])
+end

data/zip_codes.rb

@@ -0,0 +1,24 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'sqlite3'
+
+DBFILE = File.dirname(__FILE__) + '/data/zip_codes.db'
+raise "Please populate the database first" unless File.exist?(DBFILE)
+
+begin
+  ZIP_DB = SQLite3::Database.open( DBFILE )
+  raise "No database available.  Please initialize it with Rake." unless ZIP_DB.kind_of?(SQLite3::Database)
+  ZIP_META = ZIP_DB.execute("SELECT * FROM sqlite_master;").first.last.split(/[\r\n\t]+/)
+  #raise "Can't query DB" unless (ZIP_META.kind_of?(Array) && ZIP_META.include?("zip_codes"))
+  ZIP_COLUMNS = []
+  ZIP_META.each do |l|
+    next unless l =~ /^    ([\S]+)\s+/
+    ZIP_COLUMNS.push($1.to_sym)
+  end
+rescue Exception => e
+  puts "There is a problem with your database: #{e.message}"
+end
+
+#raise ZIP_META.inspect
+
+require File.dirname(__FILE__) + '/lib/zip_code.rb'

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification 
+name: mlightner-zip_codes
+version: !ruby/object:Gem::Version 
+  version: 0.5.0
+platform: ruby
+authors: 
+- Matt Lightner
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-05-16 00:00:00 -07:00
+default_executable: zip_codes
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: sqlite3
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+    version: 
+description: A library for looking up zip codes and their associated info.
+email: mlightner@gmail.com
+executables: 
+- zipfind
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README
+- bin
+- bin/zipfind
+- lib
+- lib/zip_code.rb
+- zip_codes.gemspec
+- data
+- data/zip_codes.db
+- Rakefile
+- zip_codes.rb
+- test.rb
+has_rdoc: false
+homepage: http://github.com/mlightner/zip_codes
+licenses: 
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- zip_codes
+- --main
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 2
+summary: A library for looking up zip codes and their associated info.
+test_files: []
+