bizratr 0.0.5 → 0.1.0

data/Gemfile.lock

@@ -1,12 +1,13 @@
 PATH
   remote: .
   specs:
-    bizratr (0.0.5)
+    bizratr (0.1.0)
       factual-api
       foursquare2
       geocoder
       google_places
       google_places
+      koala
       levenshtein
       yelpster
 
@@ -32,6 +33,9 @@ GEM
       multi_json (~> 1.0)
       multi_xml
     json (1.7.5)
+    koala (1.5.0)
+      faraday (~> 0.7)
+      multi_json (~> 1.3)
     levenshtein (0.2.2)
     multi_json (1.3.6)
     multi_xml (0.5.1)

data/README.rdoc

@@ -1,20 +1,23 @@
 = BizRatr
-This gem pulls in business data from a variety of sources and synthesizes a singular view of the business.  Right now Yelp, Factual, Google Places, and Foursquare are used as sources.  Bizratr will collapse all of the matching sources into a singular view of each matching business.
+This gem pulls in business data from a variety of sources and synthesizes a singular view of the business.  Right now Facebook, Yelp, Factual, Google Places, and Foursquare are used as sources.  Bizratr will collapse all of the matching sources into a singular view of each matching business.
 
   >> require 'bizratr'
   >> config = { 
        :foursquare => { :client_id => 'anid', :client_secret => 'asecret' },
        :yelp => { :consumer_key => 'akey', :consumer_secret => 'asecret', :token => 'atoken', :token_secret => 'atokensecret' },
        :google_places => { :key => "akey" },
-       :factual => { :key => 'akey', :secret => 'asecret' }
+       :factual => { :key => 'akey', :secret => 'asecret' },
+       :facebook => { :key => 'akey', :secret => 'asecret' }
      }
-  >> finder = BizRatr::Finder.new(config)
+  >> finder = BizRatr::UberClient.new(config)
   >> matches = finder.search_location([40.729401, -73.996061], 'third rail coffee')
   >> puts matches.first.rating 
    => 4.55
   >> matches = finder.search_location("240 Sullivan St., New York, NY 10012", 'third rail coffee')
   >> puts matches.first.rating
    => 4.55
+  >> puts matches.first.website_likes
+   => {"share_count"=>75, "like_count"=>10, "comment_count"=>9, "click_count"=>6}
 
 
 In this example, the average of the ratings from any of the sources (normalized to a 5 point scale) is used as the rating.  Each result has address information, rating information, "like" information, and other relevant info.
@@ -25,6 +28,7 @@ You only need to specify the config options for the services you actually want t
 * Google places: https://code.google.com/apis/console/b/0/
 * Foursquare: https://foursquare.com/oauth/
 * Factual: https://www.factual.com/api-keys/request
+* Facebook: https://developers.facebook.com/apps
 
 See http://findingscience.com/bizratr for more information.
 

data/lib/bizratr/business.rb

@@ -1,8 +1,10 @@
+require 'uri'
+
 module BizRatr
   class Business
-    attr_accessor :name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city, :categories
+    attr_accessor :name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city, :categories, :website
 
-    def initialize(lat, lon, name)
+    def initialize(uberclient, lat, lon, name)
       @ids = {}
       @checkins = {}
       @users = {}
@@ -12,10 +14,11 @@ module BizRatr
       @categories = {}
       @coords = [lat, lon]
       @name = name
+      @uberclient = uberclient
     end
 
     def to_s
-      attrs = [:name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city, :categories]
+      attrs = [:name, :phone, :address, :state, :country, :zip, :twitter, :ids, :checkins, :users, :likes, :ratings, :review_counts, :coords, :city, :categories, :website]
       args = attrs.map { |k| "#{k.to_s}=#{send(k)}" }.join(", ")
       "<Business [#{args}]>"
     end
@@ -28,6 +31,23 @@ module BizRatr
       @users.values.inject { |a,b| a+b } || 0
     end
 
+    # Get all of the website like information from Facebook.  If there's no website, or an issue, return {} - otherwise
+    # you'll get something of the form {"share_count"=>75, "like_count"=>10, "comment_count"=>9, "click_count"=>6}
+    def website_likes
+      fb = @uberclient.get_connector(:facebook)
+      return {} if fb.nil? or @website.nil?
+      # normalize URL first
+      begin
+        uri = URI(@website)
+        uri.query = nil
+        uri.path = ''
+        results = fb.get_url_likes(uri.to_s)
+        (results.length > 0) ? results.first : {}
+      rescue
+        {}
+      end
+    end
+
     def total_reviews
       @review_counts.values.inject { |a,b| a+b } || 0
     end
@@ -49,6 +69,7 @@ module BizRatr
       @zip ||= other.zip
       @twitter ||= other.twitter
       @city ||= other.city
+      @website ||= other.website
       @checkins = @checkins.merge(other.checkins)
       @users = @users.merge(other.users)
       @likes = @likes.merge(other.likes)
@@ -82,7 +103,7 @@ module BizRatr
     end
 
     def add_likes(connector, likes)
-      @users[connector] = likes
+      @likes[connector] = likes
     end
 
     def add_rating(connector, rating)

data/lib/bizratr/connector.rb

@@ -4,34 +4,74 @@ require 'levenshtein'
 require 'google_places'
 require 'geocoder'
 require 'factual'
+require 'koala'
 
 module BizRatr
   class Connector
+    def initialize(uberclient, config)
+      @uberclient = uberclient
+      @config = config
+      @client = make_client(@config)
+    end
+
+    def make_client(config)
+      raise "Not implemented"
+    end
+
     def search_location(location, query)
       raise "Not implemented"
     end
+  end
 
-    def geocode(address)
-      Geocoder.coordinates(address)
+  class FacebookConnector < Connector
+    def make_client(config)
+      oauth = Koala::Facebook::OAuth.new(config[:key], config[:secret])
+      Koala::Facebook::API.new(oauth.get_app_access_token)
+    end
+
+    def search_location(location, query)
+      results = @client.search(query, { :type => 'place', :center => location.join(','), :distance => 1000 })
+      results.map { |item|
+        make_business(@client.get_object(item['id']))
+      }
+    end
+
+    def make_business(item)
+      b = Business.new(@uberclient, item['location']['latitude'], item['location']['longitude'], item['name'])
+      b.add_id(:facebook, item['id'])
+      b.city = item['location']['city']
+      b.country = item['location']['country']
+      b.phone = item['phone'] unless item['phone'] == "nope"
+      b.zip = item['location']['zip']
+      b.website = item['website'].split(' ').first
+      b.address = item['location']['street']
+      b.add_categories(:facebook, [item['category']])
+      b.add_checkins(:facebook, item['checkins'] + item['were_here_count'])
+      b.add_likes(:facebook, item['likes'])
+      b
+    end
+
+    def get_url_likes(url)
+      @client.fql_query("SELECT share_count, like_count, comment_count, click_count FROM link_stat WHERE url='#{url}'")
     end
   end
 
   class FactualConnector < Connector
-    def initialize(config)
-      @client = Factual.new(config[:key], config[:secret])
+    def make_client(config)
+      Factual.new(config[:key], config[:secret])
     end
 
     def search_location(location, query)
-      location = geocode(location) if location.is_a? String
       results = @client.table("places").filters("name" => query).geo("$circle" => { "$center" => location, "$meters" => 1000 })
       results.map { |item| make_business(item) }
     end
 
     def make_business(item)
-      b = Business.new(item['latitude'], item['longitude'], item['name'])
+      b = Business.new(@uberclient, item['latitude'], item['longitude'], item['name'])
       b.add_id(:factual, item['factual_id'])
       b.add_categories(:factual, item['category'].split(",").map(&:strip))
       b.phone = item['tel'].gsub(/[\ ()-]*/, '')
+      b.website = item['website']
       b.city = item['locality']
       b.country = item['country']
       b.zip = item['postcode']
@@ -41,46 +81,43 @@ module BizRatr
   end
 
   class GooglePlacesConnector < Connector
-    def initialize(config)
-      @client = GooglePlaces::Client.new(config[:key])
+    def make_client(config)
+      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 = Business.new(@uberclient, 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 || "").gsub(/[\ ()-]*/, '')
       b.city = item.city || item.vicinity.split(',').last
       b.country = item.country
+      b.website = item.website
       b.zip = item.postal_code
       b.address = item.vicinity.split(',').first
       b.add_rating(:google_places, item.rating)
+      b.add_review_counts(:google_places, item.reviews.length)
       b
     end
   end
 
   class FourSquareConnector < Connector
-    def initialize(config)
-      @client = Foursquare2::Client.new(config)
+    def make_client(config)
+      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 = @client.search_venues(:ll => location.join(","), :query => query)
       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 = Business.new(@uberclient, 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)
@@ -91,15 +128,15 @@ module BizRatr
       b.country = item['location']['cc']
       b.address = item['location']['address']
       b.add_checkins(:foursquare, item['stats']['checkinsCount'])
+      b.website = item['url']
       b.add_users(:foursquare, item['stats']['usersCount'])
       b
     end
   end
 
   class YelpConnector < Connector
-    def initialize(config)
-      @config = config
-      @client = Yelp::Client.new
+    def make_client(config)
+      Yelp::Client.new
     end
 
     def search_location(location, query)
@@ -113,7 +150,7 @@ module BizRatr
     end
 
     def make_business(item)
-      b = Business.new(item['location']['coordinate']['latitude'], item['location']['coordinate']['longitude'], item['name'])
+      b = Business.new(@uberclient, 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']
@@ -128,16 +165,18 @@ module BizRatr
     end
   end
 
-  class Finder
+  class UberClient
     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)
-        when :factual then FactualConnector.new(value)
-        else raise "No such connector found: #{key}"
-        end
+      @connectors = {}
+      config.each { |key, value| 
+        @connectors[key] = case key
+                           when :foursquare then FourSquareConnector.new(self, value)
+                           when :yelp then YelpConnector.new(self, value)
+                           when :google_places then GooglePlacesConnector.new(self, value)
+                           when :factual then FactualConnector.new(self, value)
+                           when :facebook then FacebookConnector.new(self, value)
+                           else raise "No such connector found: #{key}"
+                           end
       }
     end
 
@@ -145,11 +184,20 @@ module BizRatr
     # location parameter should be either an address string or an array consisting of
     # [ lat, lon ].
     def search_location(location, query)
-      merge @connectors.map { |c|
+      location = geocode(location) if location.is_a? String
+      merge @connectors.values.map { |c|
         c.search_location(location, query)
       }
     end
 
+    def geocode(address)
+      Geocoder.coordinates(address)
+    end
+
+    def get_connector(key)
+      @connectors.fetch(key, nil)
+    end
+
     # Search a location (just like search_location) but only return the best
     # matching business (based on name).
     def search_location_strictly(location, name)

data/lib/bizratr/version.rb

@@ -1,3 +1,3 @@
 module BizRatr
-  VERSION = "0.0.5"
+  VERSION = "0.1.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bizratr
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-09-17 00:00:00.000000000Z
+date: 2012-09-19 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: foursquare2
-  requirement: &70115383331060 !ruby/object:Gem::Requirement
+  requirement: &70168127272320 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383331060
+  version_requirements: *70168127272320
 - !ruby/object:Gem::Dependency
   name: yelpster
-  requirement: &70115383329940 !ruby/object:Gem::Requirement
+  requirement: &70168127271660 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383329940
+  version_requirements: *70168127271660
 - !ruby/object:Gem::Dependency
   name: google_places
-  requirement: &70115383328980 !ruby/object:Gem::Requirement
+  requirement: &70168127271020 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383328980
+  version_requirements: *70168127271020
 - !ruby/object:Gem::Dependency
   name: levenshtein
-  requirement: &70115383327940 !ruby/object:Gem::Requirement
+  requirement: &70168127270360 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383327940
+  version_requirements: *70168127270360
 - !ruby/object:Gem::Dependency
   name: google_places
-  requirement: &70115383327080 !ruby/object:Gem::Requirement
+  requirement: &70168127269740 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,10 +65,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383327080
+  version_requirements: *70168127269740
 - !ruby/object:Gem::Dependency
   name: geocoder
-  requirement: &70115383326020 !ruby/object:Gem::Requirement
+  requirement: &70168127269100 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -76,10 +76,21 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383326020
+  version_requirements: *70168127269100
+- !ruby/object:Gem::Dependency
+  name: koala
+  requirement: &70168127268520 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70168127268520
 - !ruby/object:Gem::Dependency
   name: factual-api
-  requirement: &70115383325040 !ruby/object:Gem::Requirement
+  requirement: &70168127246840 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -87,7 +98,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70115383325040
+  version_requirements: *70168127246840
 description: Get business ratings.
 email: bamuller@gmail.com
 executables: []