60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
|
import time
|
||
|
|
from zope import interface
|
||
|
|
from zope.component import getUtility
|
||
|
|
|
||
|
|
from BTrees.OOBTree import OOBTree, OOSet, intersection
|
||
|
|
|
||
|
|
from interfaces import ICommentingTool
|
||
|
|
# The commenting tool, which is a local utility
|
||
|
|
|
||
|
|
class CommentingTool(object):
|
||
|
|
interface.implements(ICommentingTool)
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self._id2uid = OOBTree() # The comment ID to object UID
|
||
|
|
self._id2text = OOBTree() # The text for a comment
|
||
|
|
self._wfstate2id = OOBTree() # To search on wf states
|
||
|
|
self._creator2id = OOBTree() # To search/order on creator ids
|
||
|
|
|
||
|
|
def index(self, comment):
|
||
|
|
# Store the object in the store:
|
||
|
|
id = comment.comment_id
|
||
|
|
self._id2uid[id] = comment.__parent__._parent_uid
|
||
|
|
self._id2text[id] = comment.text
|
||
|
|
|
||
|
|
# TODO
|
||
|
|
## Index on workflow state
|
||
|
|
#wfstate = comment.getWorkflowState()
|
||
|
|
#if not wfstate in self._wfstate2id:
|
||
|
|
#self._wfstate2id[wfstate] = OOSet()
|
||
|
|
#self._wfstate2id[wfstate].insert(id)
|
||
|
|
|
||
|
|
# Index on creator
|
||
|
|
creator = comment.creator
|
||
|
|
if not creator in self._creator2id:
|
||
|
|
self._creator2id[creator] = OOSet()
|
||
|
|
self._creator2id[creator].insert(id)
|
||
|
|
|
||
|
|
def search(self, creator=None):
|
||
|
|
if creator is not None:
|
||
|
|
# Get all replies for a certain object
|
||
|
|
ids = self._creator2ids.get(creator, None)
|
||
|
|
if ids is None:
|
||
|
|
raise StopIteration
|
||
|
|
else:
|
||
|
|
ids = self._id2uid.keys()
|
||
|
|
|
||
|
|
for id in ids:
|
||
|
|
yield {'id': id,
|
||
|
|
'text': self._id2text[id]
|
||
|
|
# TODO: More data + maybe brains or something?
|
||
|
|
}
|
||
|
|
|
||
|
|
def object_added_handler(obj, event):
|
||
|
|
tool = getUtility(ICommentingTool)
|
||
|
|
tool.index(obj)
|
||
|
|
|
||
|
|
def object_removed_handler(obj, event):
|
||
|
|
tool = getUtility(ICommentingTool)
|
||
|
|
tool.unindex(obj)
|