From a53b5d7304357c1e8c1b4eec6f153480233ea108 Mon Sep 17 00:00:00 2001
From: helmutm <helmutm@fd906abe-77d9-0310-91a1-e0d9ade77398>
Date: Sun, 9 Sep 2007 19:45:54 +0000
Subject: [PATCH] make indexes configurable

git-svn-id: svn://svn.cy55.de/Zope3/src/cybertools/trunk@2019 fd906abe-77d9-0310-91a1-e0d9ade77398
---
 tracking/btree.py      | 25 ++++++++++++++++++++-----
 tracking/interfaces.py |  1 +
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/tracking/btree.py b/tracking/btree.py
index dae4a7f..4232ae3 100644
--- a/tracking/btree.py
+++ b/tracking/btree.py
@@ -57,11 +57,14 @@ class Track(Persistent):
     implements(ITrack)
 
     metadata_attributes = ('taskId', 'runId', 'userName', 'timeStamp')
+    index_attributes = metadata_attributes
 
     @property
     def metadata(self):
         return dict((attr, getattr(self, attr)) for attr in self.metadata_attributes)
 
+    indexdata = metadata
+
     def __init__(self, taskId, runId, userName, data={}):
         self.taskId = taskId
         self.runId = runId
@@ -90,7 +93,8 @@ class TrackingStorage(BTreeContainer):
     trackNum = runId = 0
     runs = None
 
-    indexAttributes = ('taskId', 'runId', 'userName', 'timeStamp')
+    #indexAttributes = ('taskId', 'runId', 'userName', 'timeStamp')
+    indexAttributes = Track.index_attributes
 
     def __init__(self, *args, **kw):
         super(TrackingStorage, self).__init__(*args, **kw)
@@ -101,6 +105,15 @@ class TrackingStorage(BTreeContainer):
         self.currentRuns = OOBTree.OOBTree()
         self.taskUsers = OOBTree.OOBTree()
 
+    def setupIndexes(self):
+        changed = False
+        for idx in self.indexAttributes:
+            if idx not in self.indexes:
+                self.indexes[idx] = FieldIndex()
+                changed = True
+        if changed:
+            self.reindexTracks()
+
     def idFromNum(self, num):
         return '%07i' % (num)
 
@@ -163,11 +176,13 @@ class TrackingStorage(BTreeContainer):
         return trackId
 
     def indexTrack(self, trackNum, track):
-        md = track.metadata
+        ixd = track.indexdata
         for attr in self.indexAttributes:
-            self.indexes[attr].index_doc(trackNum, md[attr])
-        taskId = md['taskId']
-        userName = md['userName']
+            value = ixd[attr]
+            if value is not None:
+                self.indexes[attr].index_doc(trackNum, value)
+        taskId = ixd['taskId']
+        userName = ixd['userName']
         if taskId not in self.taskUsers:
             self.taskUsers[taskId] = OOBTree.OOTreeSet()
         self.taskUsers[taskId].update([userName])
diff --git a/tracking/interfaces.py b/tracking/interfaces.py
index 35ec29e..162924e 100644
--- a/tracking/interfaces.py
+++ b/tracking/interfaces.py
@@ -44,6 +44,7 @@ class ITrack(Interface):
 
     data = Attribute('The data for this track, typically a mapping')
     metadata = Attribute('A mapping with the track\'s metadata')
+    indexdata = Attribute('A mapping with the data to be used for indexing')
 
 
 class ITrackingStorage(Interface):