selectors.py
branchtls-sprint
changeset 764 9de67b681624
parent 759 e044f28372bd
child 766 33ede72b22b8
--- a/selectors.py	Wed Feb 18 10:22:08 2009 +0100
+++ b/selectors.py	Wed Feb 18 10:27:09 2009 +0100
@@ -770,7 +770,8 @@
         
     def score(self, req, rset, row, col):
         try:
-            return len(req.execute(self.rql, {'x': eid, 'u': req.user.eid}, 'x'))
+            return len(req.execute(self.rql, {'x': rset[row][col],
+                                              'u': req.user.eid}, 'x'))
         except Unauthorized:
             return 0
 
@@ -800,7 +801,8 @@
     :param scorefunc: callable expected to take an entity as argument and to
                       return a score >= 0 
     """
-    def __init__(self, scorefunc):
+    def __init__(self, scorefunc, once_is_enough=False):
+        super(EntitySelector, self).__init__(once_is_enough)
         self.score_entity = scorefunc