bizratr 0.0.1 → 0.0.2

data/lib/bizratr/business.rb

@@ -1,6 +1,6 @@
 module BizRatr
   class Business
-    attr_accessor :name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city
+    attr_accessor :name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city, :categories
 
     def initialize(lat, lon, name)
       @ids = {}
@@ -9,12 +9,13 @@ module BizRatr
       @likes = {}
       @ratings = {}
       @review_counts = {}
+      @categories = {}
       @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]
+      attrs = [:name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city, :categories]
       args = attrs.map { |k| "#{k.to_s}=#{send(k)}" }.join(", ")
       "<Business [#{args}]>"
     end
@@ -54,6 +55,15 @@ module BizRatr
       @users[connector] = users
     end
 
+    def add_categories(connector, categories)
+      @categories[connector] = categories
+    end
+
+    # Get all categories from all connectors.
+    def flattened_categories
+      @categories.values.flatten.map { |c| c.downcase }.uniq
+    end
+
     def add_likes(connector, likes)
       @users[connector] = likes
     end
@@ -88,7 +98,8 @@ module BizRatr
     end
 
     def name_distance_to(other)
-      Levenshtein::normalized_distance(@name, other.name)
+      name = other.is_a?(Business) ? other.name : other
+      Levenshtein::normalized_distance(@name, name)
     end
 
     def ==(other)

data/lib/bizratr/connector.rb

@@ -29,6 +29,7 @@ module BizRatr
     def make_business(item)
       b = Business.new(item.lat, item.lng, item.name)
       b.add_id(:google_places, item.id)
+      b.add_categories(:google_places, item.types)
       b.phone = item.formatted_phone_number
       b.city = item.city || item.vicinity.split(',').last
       b.country = item.country
@@ -56,6 +57,8 @@ module BizRatr
     def make_business(item)
       b = Business.new(item['location']['lat'], item['location']['lng'], item['name'])
       b.add_id(:foursquare, item['id'])
+      categories = item.categories.map { |c| [ c.name ] + c.parents }.flatten
+      b.add_categories(:foursquare, categories)
       b.phone = item['contact'].fetch('phone', nil)
       b.twitter = item['contact'].fetch('twitter', nil)
       b.state = item['location']['state']
@@ -87,6 +90,7 @@ module BizRatr
     def make_business(item)
       b = Business.new(item['location']['coordinate']['latitude'], item['location']['coordinate']['longitude'], item['name'])
       b.add_id(:yelp, item['id'])
+      b.add_categories(:yelp, item['categories'].map(&:first))
       b.state = item['location']['state_code']
       b.zip = item['location']['postal_code']
       b.country = item['location']['country_code']
@@ -120,6 +124,14 @@ module BizRatr
       }
     end
 
+    # Search a location (just like search_location) but only return the best
+    # matching business (based on name).
+    def search_location_strictly(location, name)
+      search_location(location, name).inject(nil) { |a,b|
+        (a.nil? or b.name_distance_to(name) < a.name_distance_to(name)) ? b : a
+      }
+    end
+
     private
     def merge(lists)
       lists.inject([]) { |o,t| merge_lists(o, t) }

data/lib/bizratr/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bizratr
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-09-17 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: foursquare2
-  requirement: &70329591509040 !ruby/object:Gem::Requirement
+  requirement: &70324969386460 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70329591509040
+  version_requirements: *70324969386460
 - !ruby/object:Gem::Dependency
   name: yelpster
-  requirement: &70329591508340 !ruby/object:Gem::Requirement
+  requirement: &70324969385780 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70329591508340
+  version_requirements: *70324969385780
 - !ruby/object:Gem::Dependency
   name: google_places
-  requirement: &70329591507700 !ruby/object:Gem::Requirement
+  requirement: &70324969385140 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70329591507700
+  version_requirements: *70324969385140
 - !ruby/object:Gem::Dependency
   name: levenshtein
-  requirement: &70329591507040 !ruby/object:Gem::Requirement
+  requirement: &70324969384480 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70329591507040
+  version_requirements: *70324969384480
 - !ruby/object:Gem::Dependency
   name: google_places
-  requirement: &70329591506420 !ruby/object:Gem::Requirement
+  requirement: &70324969383840 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,10 +65,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70329591506420
+  version_requirements: *70324969383840
 - !ruby/object:Gem::Dependency
   name: geocoder
-  requirement: &70329591505820 !ruby/object:Gem::Requirement
+  requirement: &70324969383200 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -76,7 +76,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70329591505820
+  version_requirements: *70324969383200
 description: Get business ratings.
 email: bamuller@gmail.com
 executables: []