bizratr 0.0.1

data/lib/bizratr/business.rb

@@ -0,0 +1,102 @@
+module BizRatr
+  class Business
+    attr_accessor :name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city
+
+    def initialize(lat, lon, name)
+      @ids = {}
+      @checkins = {}
+      @users = {}
+      @likes = {}
+      @ratings = {}
+      @review_counts = {}
+      @coords = [lat, lon]
+      @name = name
+    end
+
+    def to_s
+      attrs = [:name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city]
+      args = attrs.map { |k| "#{k.to_s}=#{send(k)}" }.join(", ")
+      "<Business [#{args}]>"
+    end
+
+    def rating
+      @ratings.values.inject(:+) / @ratings.length
+    end
+
+    def merge(other)
+      @ids = @ids.merge(other.ids)
+      @phone ||= other.phone
+      @address ||= other.address
+      @state ||= other.state
+      @country ||= other.country
+      @zip ||= other.zip
+      @twitter ||= other.twitter
+      @city ||= other.city
+      @checkins = @checkins.merge(other.checkins)
+      @users = @users.merge(other.users)
+      @likes = @likes.merge(other.likes)
+      @ratings = @ratings.merge(other.ratings)
+      @review_counts = @review_counts.merge(other.review_counts)
+      @coords[0] = (@coords[0].to_f + other.coords[0].to_f) / 2.0
+      @coords[1] = (@coords[1].to_f + other.coords[1].to_f) / 2.0
+      return self
+    end
+
+    def add_id(connector, id)
+      @ids[connector] = id
+    end
+
+    def add_checkins(connector, checkins)
+      @checkins[connector] = checkins
+    end
+
+    def add_users(connector, users)
+      @users[connector] = users
+    end
+
+    def add_likes(connector, likes)
+      @users[connector] = likes
+    end
+
+    def add_rating(connector, rating)
+      @ratings[connector] = rating
+    end
+
+    def add_review_counts(connector, counts)
+      @review_counts[connector] = counts
+    end
+
+    def any_equal_ids?(other)
+      @ids.each { |k,v|
+        return true if other.ids[k] == v
+      }
+      false
+    end
+
+    def distance_to(other)
+      rpd = 0.017453293  #  PI/180
+      dlat = other.coords[0] - @coords[0]
+      dlon = other.coords[1] - @coords[1]
+      dlon_rad = dlon * rpd 
+      dlat_rad = dlat * rpd
+      lat1_rad = @coords[0] * rpd
+      lon1_rad = @coords[1] * rpd
+      lat2_rad = other.coords[0] * rpd
+      lon2_rad = other.coords[1] * rpd
+      a = (Math.sin(dlat_rad/2))**2 + Math.cos(lat1_rad) * Math.cos(lat2_rad) * (Math.sin(dlon_rad/2))**2
+      2 * Math.atan2( Math.sqrt(a), Math.sqrt(1-a)) * 3956 # 3956 is the radius of the great circle in miles
+    end
+
+    def name_distance_to(other)
+      Levenshtein::normalized_distance(@name, other.name)
+    end
+
+    def ==(other)
+      return false if other.nil?
+      return true if any_equal_ids?(other)
+      return true if @phone == other.phone and not @phone.nil?
+      return true if distance_to(other) < 0.3 and name_distance_to(other) < 0.4
+      false
+    end
+  end
+end

data/lib/bizratr/connector.rb

@@ -0,0 +1,144 @@
+require 'foursquare2'
+require 'yelpster'
+require 'levenshtein'
+require 'google_places'
+require 'geocoder'
+
+module BizRatr
+  class Connector
+    def search_location(location, query)
+      raise "Not implemented"
+    end
+
+    def geocode(address)
+      Geocoder.coordinates(address)
+    end
+  end
+
+  class GooglePlacesConnector < Connector
+    def initialize(config)
+      @client = GooglePlaces::Client.new(config[:key])
+    end
+
+    def search_location(location, query)
+      location = geocode(location) if location.is_a? String
+      results = @client.spots(location[0], location[1], :name => query)
+      results.map { |item| make_business(item) }
+    end
+
+    def make_business(item)
+      b = Business.new(item.lat, item.lng, item.name)
+      b.add_id(:google_places, item.id)
+      b.phone = item.formatted_phone_number
+      b.city = item.city || item.vicinity.split(',').last
+      b.country = item.country
+      b.zip = item.postal_code
+      b.address = item.vicinity.split(',').first
+      b.add_rating(:google_places, item.rating)
+      b
+    end
+  end
+
+  class FourSquareConnector < Connector
+    def initialize(config)
+      @client = Foursquare2::Client.new(config)
+    end
+
+    def search_location(location, query)
+      if location.is_a? Array
+        results = @client.search_venues(:ll => location.join(","), :query => query)
+      else
+        results = @client.search_venues(:near => location, :query => query)
+      end
+      results['groups'].first['items'].map { |item| make_business(item) }
+    end
+
+    def make_business(item)
+      b = Business.new(item['location']['lat'], item['location']['lng'], item['name'])
+      b.add_id(:foursquare, item['id'])
+      b.phone = item['contact'].fetch('phone', nil)
+      b.twitter = item['contact'].fetch('twitter', nil)
+      b.state = item['location']['state']
+      b.city = item['location']['city']
+      b.country = item['location']['cc']
+      b.address = item['location']['address']
+      b.add_checkins(:foursquare, item['stats']['checkinsCount'])
+      b.add_users(:foursquare, item['stats']['usersCount'])
+      b
+    end
+  end
+
+  class YelpConnector < Connector
+    def initialize(config)
+      @config = config
+      @client = Yelp::Client.new
+    end
+
+    def search_location(location, query)
+      # while yelp does support searching by address, it does so much more shittily w/o lat/lon
+      location = geocode(location) if location.is_a? String
+      config = { :term => query, :latitude => location.first, :longitude => location.last }
+      result = @client.search Yelp::V2::Search::Request::GeoPoint.new(config.merge(@config)) 
+      result['businesses'].map { |item| 
+        make_business(item) 
+      }
+    end
+
+    def make_business(item)
+      b = Business.new(item['location']['coordinate']['latitude'], item['location']['coordinate']['longitude'], item['name'])
+      b.add_id(:yelp, item['id'])
+      b.state = item['location']['state_code']
+      b.zip = item['location']['postal_code']
+      b.country = item['location']['country_code']
+      b.city = item['location']['city']
+      b.address = item['location']['address'].first
+      b.phone = item['phone']
+      b.add_rating(:yelp, item['rating'])
+      b.add_review_counts(:yelp, item['review_count'])
+      b
+    end
+  end
+
+  class Finder
+    def initialize(config)
+      @connectors = config.map { |key, value| 
+        case key
+        when :foursquare then FourSquareConnector.new(value)
+        when :yelp then YelpConnector.new(value)
+        when :google_places then GooglePlacesConnector.new(value)
+        else raise "No such connector found: #{key}"
+        end
+      }
+    end
+
+    # Search for a business (or business category) near lat/lon coordinates.  The 
+    # location parameter should be either an address string or an array consisting of
+    # [ lat, lon ].
+    def search_location(location, query)
+      merge @connectors.map { |c|
+        c.search_location(location, query)
+      }
+    end
+
+    private
+    def merge(lists)
+      lists.inject([]) { |o,t| merge_lists(o, t) }
+    end
+    
+    def merge_lists(one, two)
+      one.map { |first|
+        result = first
+        two.map! { |second|
+          if first == second
+            result = first.merge(second)
+            nil
+          else
+            second
+          end
+        }
+        result
+      } + two.select { |s| not s.nil? }
+    end
+  end
+
+end

data/lib/bizratr/version.rb

@@ -0,0 +1,3 @@
+module BizRatr
+  VERSION = "0.0.1"
+end

data/lib/bizratr.rb

@@ -0,0 +1,4 @@
+require 'bizratr/business'
+require 'bizratr/connector'
+require 'bizratr/version'
+

metadata

@@ -0,0 +1,134 @@
+--- !ruby/object:Gem::Specification
+name: bizratr
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Brian Muller
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-17 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: foursquare2
+  requirement: &70329591509040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70329591509040
+- !ruby/object:Gem::Dependency
+  name: yelpster
+  requirement: &70329591508340 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70329591508340
+- !ruby/object:Gem::Dependency
+  name: google_places
+  requirement: &70329591507700 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70329591507700
+- !ruby/object:Gem::Dependency
+  name: levenshtein
+  requirement: &70329591507040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70329591507040
+- !ruby/object:Gem::Dependency
+  name: google_places
+  requirement: &70329591506420 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70329591506420
+- !ruby/object:Gem::Dependency
+  name: geocoder
+  requirement: &70329591505820 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70329591505820
+description: Get business ratings.
+email: bamuller@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/bizratr/business.rb
+- lib/bizratr/connector.rb
+- lib/bizratr/version.rb
+- lib/bizratr.rb
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- Rakefile
+- README.rdoc
+- docs/BizRatr/Business.html
+- docs/BizRatr/Connector.html
+- docs/BizRatr/Finder.html
+- docs/BizRatr/FourSquareConnector.html
+- docs/BizRatr/GooglePlacesConnector.html
+- docs/BizRatr/YelpConnector.html
+- docs/BizRatr.html
+- docs/created.rid
+- docs/index.html
+- docs/lib/bizratr/business_rb.html
+- docs/lib/bizratr/connector_rb.html
+- docs/lib/bizratr/version_rb.html
+- docs/lib/bizratr_rb.html
+- docs/rdoc.css
+- docs/README_rdoc.html
+homepage: https://github.com/bmuller/bizratr
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: bizratr
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: Get business ratings.
+test_files: []