From 6a5f6c379c31442baf2ccafff5ca0299db16d7dc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Ren=C3=A9=20Hanke?= <florian.hanke@gmail.com>
Date: Thu, 7 Feb 2013 18:09:34 +1100
Subject: [PATCH] ! Weight strategy

---
 server/lib/picky/backends/backend.rb | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/server/lib/picky/backends/backend.rb b/server/lib/picky/backends/backend.rb
index 490b2900..42513a1c 100644
--- a/server/lib/picky/backends/backend.rb
+++ b/server/lib/picky/backends/backend.rb
@@ -291,10 +291,11 @@ def add id, str_or_sym, weight_strategy, similarity_strategy, where
           ids = @inverted[str_or_sym] ||= []
           ids.send where, id
         end
-
+        
         # Weights.
         #
-        @weights[str_or_sym] = weight_strategy.weight_for ids.size
+        @weights[str_or_sym] = weight_strategy.respond_to?(:[]) && weight_strategy[str_or_sym] ||
+                               weight_strategy.weight_for(ids.size)
 
         # Similarity.
         #