def bootstrap_member_deletion_queue(portal, out):
annot = IAnnotations(portal)
if annot.has_key(REMOVAL_QUEUE_KEY):
print >> out, 'Already have member deletion queue, skipping'
return
annot[REMOVAL_QUEUE_KEY] = zc.queue.Queue()
print >> out, "Added member deletion queue"
return
def __init__(self, site_root):
self.site_root = site_root
self.annot = IAnnotations(site_root)
class TransientMessage(object):
adapts(IOpenSiteRoot)
implements(ITransientMessage)
key = 'transient-message'
def __init__(self, site_root):
self.site_root = site_root
self.annot = IAnnotations(site_root)
def _category_annot(self, mem_id, category):
tm_annot = self.annot.setdefault(self.key, OOBTree())
mem_annot = tm_annot.setdefault(mem_id, OOBTree())
category_annot = mem_annot.setdefault(category, IOBTree())
return category_annot
def store(self, mem_id, category, msg):
"""
This takes an OBJECT and stores it on a per-user basis.
If we receive a string, we're going to see if it's a message id
and attempt to translate it. If not, WE WILL LEAVE IT ALONE!!!
"""
cat = self._category_annot(mem_id, category)
try:
new_id = cat.maxKey() + 1
except ValueError:
new_id = 0
# Try to turn a str into a Message.
if isinstance(msg, basestring) and not isinstance(msg, Message):
# The reason for that comparison is: if it's already a
# Message, then -- very confusingly -- Message is a
# subclass of basestring ... but calling _() on a Message
# returns an instance that compares equal to the original
# but translates differently! No kidding. WTF.
msg = _(msg)
if isinstance(msg, Message):
msg = translate(msg, context=self.site_root)
if isinstance(msg, Message) or isinstance(msg, basestring):
cleaner = Cleaner()
msg = cleaner.clean_html(msg)
# clean_html wraps plain text messages in a paragraph tag. If
# that has happened, we'll remove it to restore the original message.
if msg.startswith('<p>'):
msg = msg[3:-4]
cat[new_id] = msg
def get_msgs(self, mem_id, category):
cat = self._category_annot(mem_id, category)
return cat.items()
def get_all_msgs(self, mem_id):
tm_annot = self.annot.setdefault(self.key, OOBTree())
mem_annot = tm_annot.setdefault(mem_id, OOBTree())
items = dict(mem_annot)
for category in items.iterkeys():
items[category] = dict(items[category])
return items
def pop(self, mem_id, category, idx):
cat = self._category_annot(mem_id, category)
return cat.pop(idx)