recommendations 0.0.1 → 0.0.2

data/lib/recommendations/item/similarity.rb

@@ -0,0 +1,22 @@
+module Recommendations
+  class Item
+    module Similarity
+      def similars(count = Recommendations.configuration.similar_count)
+        Recommendations.redis.zrange("recommendations:similars:#{id}", 0, count - 1)
+      end
+
+      def update_similars
+        Recommendations.redis.del("recommendations:similars:#{id}")
+
+        categories.each do |category, score|
+          items = Recommendations.redis.hgetall("recommendations:categories:#{category}")
+
+          items.reject {|k, v| k == id}.each do |item, item_score|
+            difference = (Float(item_score) - Float(score)).abs
+            Recommendations.redis.zincrby("recommendations:similars:#{id}", difference, item)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/recommendations/item.rb

@@ -1,9 +1,10 @@
 module Recommendations
   class Item
+    include Item::Similarity
     attr_accessor :id
     attr_writer :categories
 
-    def initialize(id, categories)
+    def initialize(id, categories = {})
       @id, @categories = id, categories
     end
 
@@ -11,23 +12,6 @@ module Recommendations
       @categories || {}
     end
 
-    def similars(count = Recommendations.configuration.similar_count)
-      Recommendations.redis.zrange("recommendations:similars:#{id}", 0, count - 1)
-    end
-
-    def update_similars
-      Recommendations.redis.del("recommendations:similars:#{id}")
-
-      categories.each do |category, score|
-        items = Recommendations.redis.hgetall("recommendations:categories:#{category}")
-
-        items.reject {|k, v| k == id}.each do |item, item_score|
-          difference = (Float(item_score) - Float(score)).abs
-          Recommendations.redis.zincrby("recommendations:similars:#{id}", difference, item)
-        end
-      end
-    end
-
     def save!
       Recommendations.redis.hmset("recommendations:item:#{id}", *categories.flatten)
 

data/lib/recommendations/user/ratings.rb

@@ -0,0 +1,13 @@
+module Recommendations
+  class User
+    module Ratings
+      def rate(item_id, rating)
+        Recommendations.redis.hset("recommendations:ratings:#{id}", item_id, rating)
+      end
+
+      def ratings
+        Recommendations.redis.hgetall("recommendations:ratings:#{id}")
+      end
+    end
+  end
+end

data/lib/recommendations/user.rb

@@ -0,0 +1,36 @@
+module Recommendations
+  MAX_VAL = 2
+
+  class User
+    include User::Ratings
+
+    attr_accessor :id
+
+    def initialize(id)
+      @id = id
+    end
+
+    def recommendations
+      Recommendations.redis.hgetall("recommendations:user:#{id}")
+    end
+
+    def update_recommendations
+      ratings = self.ratings
+
+      recommendations = ratings.inject({}) do |buffer, (item_id, score)|
+        similars = Item.find(item_id).similars
+        similars.reject! {|i| ratings.key?(i)}
+
+        similars.each do |similar|
+          buffer[similar] ||= 0
+          buffer[similar] += (Float(score) - 3) # 0..5 score => -2..2
+        end
+
+        buffer
+      end
+
+      Recommendations.redis.hmset("recommendations:user:#{id}", *recommendations.flatten)
+    end
+  end
+end
+

data/lib/recommendations/version.rb

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

data/lib/recommendations.rb

@@ -2,6 +2,9 @@ require "redis"
 
 require "recommendations/version"
 require "recommendations/configuration"
+require "recommendations/user/ratings"
+require "recommendations/user"
+require "recommendations/item/similarity"
 require "recommendations/item"
 
 module Recommendations

data/spec/recommendations/item_spec.rb

@@ -0,0 +1,32 @@
+require "spec_helper"
+
+describe Recommendations::Item do
+  subject {Recommendations::Item.new("1", violent: 100, romantic: 25)}
+
+  describe "#initialize" do
+    its(:id)         {should == "1"}
+    its(:categories) {should == {violent: 100, romantic: 25}}
+  end
+
+  describe "#similars" do
+    context "when count was not specified" do
+      let(:count) {Recommendations.configuration.similar_count - 1}
+
+      it "gets sorted set range from similar count configuration" do
+        Recommendations.redis.should_receive(:zrange)
+                             .with("recommendations:similars:1", 0, count)
+                             .and_return ["2", "3", "4"]
+        subject.similars.should == ["2", "3", "4"]
+      end
+    end
+
+    context "when count was specified" do
+      it "gets sorted set range until count" do
+        Recommendations.redis.should_receive(:zrange)
+                             .with("recommendations:similars:1", 0, 9)
+                             .and_return ["2", "3", "4"]
+        subject.similars(10).should == ["2", "3", "4"]
+      end
+    end
+  end
+end

data/spec/recommendations/user/ratings_spec.rb

@@ -0,0 +1,32 @@
+require "spec_helper"
+
+describe Recommendations::User::Ratings do
+  subject {Recommendations::User.new("1")}
+
+  describe "#rate" do
+    it "stores a user rating for this item" do
+      subject.rate("890", 4.5)
+      Recommendations.redis.hget("recommendations:ratings:1", "890").should == "4.5"
+    end
+  end
+
+  describe "#ratings" do
+    let :ratings do
+      {
+        "1" => "5",
+        "2" => "4",
+        "3" => "3",
+        "4" => "2",
+        "5" => "1"
+      }
+    end
+
+    before do
+      ratings.each {|k, v| subject.rate(k, v)}
+    end
+
+    it "returns all ratings" do
+      subject.ratings.should == ratings
+    end
+  end
+end

data/spec/recommendations/user_spec.rb

@@ -0,0 +1,25 @@
+require "spec_helper"
+
+describe Recommendations::User do
+  describe "#update_recommendations" do
+    subject {Recommendations::User.new("1")}
+
+    before do
+      i1 = Recommendations::Item.new("1", violento: 100, romantico: 0,   acao: 50)
+      i2 = Recommendations::Item.new("2", violento: 50,  romantico: 0,   acao: 50) # 1.50
+      i3 = Recommendations::Item.new("3", violento: 100, romantico: 100, acao: 0)  # 1.150, 2.200
+      i4 = Recommendations::Item.new("4", violento: 70,  romantico: 30,  acao: 20) # 1.90,  2.80
+
+      items = [i1, i2, i3, i4]
+      items.each(&:save!)
+      items.each(&:update_similars)
+
+      subject.rate("1", 5)
+    end
+
+    it "updates recommendations" do
+      subject.update_recommendations
+      subject.recommendations.should == {"2" => "2.0", "4" => "2.0", "3" => "2.0"}
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1 +1,8 @@
 require "rspec"
+require "recommendations"
+
+RSpec.configure do |config|
+  config.before do
+    Recommendations.redis.flushdb
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: recommendations
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-05-16 00:00:00.000000000Z
+date: 2012-05-18 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70243608433720 !ruby/object:Gem::Requirement
+  requirement: &70235365846640 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,10 +21,10 @@ dependencies:
         version: '2.10'
   type: :development
   prerelease: false
-  version_requirements: *70243608433720
+  version_requirements: *70235365846640
 - !ruby/object:Gem::Dependency
   name: redis
-  requirement: &70243608433280 !ruby/object:Gem::Requirement
+  requirement: &70235365823420 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70243608433280
+  version_requirements: *70235365823420
 description: recommendations between similar items
 email:
 - victorc.rodrigues@gmail.com
@@ -50,8 +50,14 @@ files:
 - lib/recommendations.rb
 - lib/recommendations/configuration.rb
 - lib/recommendations/item.rb
+- lib/recommendations/item/similarity.rb
+- lib/recommendations/user.rb
+- lib/recommendations/user/ratings.rb
 - lib/recommendations/version.rb
 - recommendations.gemspec
+- spec/recommendations/item_spec.rb
+- spec/recommendations/user/ratings_spec.rb
+- spec/recommendations/user_spec.rb
 - spec/recommendations_spec.rb
 - spec/spec_helper.rb
 homepage: ''
@@ -79,5 +85,8 @@ signing_key:
 specification_version: 3
 summary: ''
 test_files:
+- spec/recommendations/item_spec.rb
+- spec/recommendations/user/ratings_spec.rb
+- spec/recommendations/user_spec.rb
 - spec/recommendations_spec.rb
 - spec/spec_helper.rb