diff options
author | Kali Kaneko <kali@leap.se> | 2015-01-27 14:13:55 -0400 |
---|---|---|
committer | Kali Kaneko <kali@leap.se> | 2015-01-28 02:04:06 -0400 |
commit | db3a37b3c469e3ea1372b91401447501a66a5660 (patch) | |
tree | 4f1d4def836dbdfea9a9efb9787f06ac1535ea42 | |
parent | b4f0eac56da1d9b174e0cc98fed5bf51d33e825c (diff) |
implement copy interface
-rw-r--r-- | src/leap/mail/adaptors/soledad.py | 28 | ||||
-rw-r--r-- | src/leap/mail/imap/mailbox.py | 4 | ||||
-rw-r--r-- | src/leap/mail/mail.py | 18 |
3 files changed, 34 insertions, 16 deletions
diff --git a/src/leap/mail/adaptors/soledad.py b/src/leap/mail/adaptors/soledad.py index 4dc02a1..721f25e 100644 --- a/src/leap/mail/adaptors/soledad.py +++ b/src/leap/mail/adaptors/soledad.py @@ -16,6 +16,7 @@ """ Soledadad MailAdaptor module. """ +import logging import re from collections import defaultdict from email import message_from_string @@ -512,15 +513,30 @@ class MessageWrapper(object): d.append(self.fdoc.delete(store)) return defer.gatherResults(d) - def copy(self, store, newmailbox): + def copy(self, store, new_mbox_uuid): """ Return a copy of this MessageWrapper in a new mailbox. + + :param store: an instance of Soledad, or anything that behaves alike. + :param new_mbox_uuid: the uuid of the mailbox where we are copying this + message to. + :type new_mbox_uuid: str + :rtype: MessageWrapper """ - # 1. copy the fdoc, mdoc - # 2. remove the doc_id of that fdoc - # 3. create it (with new doc_id) - # 4. return new wrapper (new meta too!) - raise NotImplementedError() + new_mdoc = self.mdoc.serialize() + new_fdoc = self.fdoc.serialize() + + # the future doc_ids is properly set because we modified + # the pointers in mdoc, which has precedence. + new_wrapper = MessageWrapper(new_mdoc, new_fdoc, None, None) + new_wrapper.hdoc = self.hdoc + new_wrapper.cdocs = self.cdocs + new_wrapper.set_mbox_uuid(new_mbox_uuid) + + # XXX could flag so that it only creates mdoc/fdoc... + d = new_wrapper.create(store) + d.addCallback(lambda result: new_wrapper) + return d def set_mbox_uuid(self, mbox_uuid): """ diff --git a/src/leap/mail/imap/mailbox.py b/src/leap/mail/imap/mailbox.py index c91f127..1bc530e 100644 --- a/src/leap/mail/imap/mailbox.py +++ b/src/leap/mail/imap/mailbox.py @@ -880,8 +880,8 @@ class IMAPMailbox(object): #deferLater(self.reactor, 0, self._do_copy, message, d) #return d - # FIXME not implemented !!! --- - return self.collection.copy_msg(message, self.mbox_name) + return self.collection.copy_msg(message.message, + self.collection.mbox_uuid) # convenience fun diff --git a/src/leap/mail/mail.py b/src/leap/mail/mail.py index 4306ec3..b46d223 100644 --- a/src/leap/mail/mail.py +++ b/src/leap/mail/mail.py @@ -509,25 +509,27 @@ class MessageCollection(object): return d - def copy_msg(self, msg, newmailbox): + def copy_msg(self, msg, new_mbox_uuid): """ Copy the message to another collection. (it only makes sense for mailbox collections) """ - # TODO currently broken ------------------FIXME- if not self.is_mailbox_collection(): raise NotImplementedError() - def insert_copied_mdoc_id(wrapper): - # TODO this needs to be implemented before the copy - # interface works. - newmailbox_uuid = get_mbox_uuid_from_msg_wrapper(wrapper) + def insert_copied_mdoc_id(wrapper_new_msg): return self.mbox_indexer.insert_doc( - newmailbox_uuid, wrapper.mdoc.doc_id) + new_mbox_uuid, wrapper.mdoc.doc_id) wrapper = msg.get_wrapper() - d = wrapper.copy(self.store, newmailbox) + + def print_result(result): + print "COPY CALLBACK:>>>", result + return result + + d = wrapper.copy(self.store, new_mbox_uuid) d.addCallback(insert_copied_mdoc_id) + d.addCallback(print_result) return d def delete_msg(self, msg): |