Unset new flag after successful write

1 files changed, 16 insertions, 0 deletions

diff --git a/src/leap/mail/imap/memorystore.py b/src/leap/mail/imap/memorystore.py
index 7cb361f..f0bdab5 100644
--- a/src/leap/mail/imap/memorystore.py
+++ b/src/leap/mail/imap/memorystore.py
@@ -271,12 +271,28 @@ class MemoryStore(object):
         return (self.get_message(*key)
                 for key in sorted(self._msg_store.keys()))
 
+    # new, dirty flags
+
     def _get_new_dirty_state(self, key):
         """
         Return `new` and `dirty` flags for a given message.
         """
         return map(lambda _set: key in _set, (self._new, self._dirty))
 
+    def set_new(self, key):
+        """
+        Add the key value to the `new` set.
+        """
+        self._new.add(key)
+
+    def unset_new(self, key):
+        """
+        Remove the key value from the `new` set.
+        """
+        print "******************"
+        print "UNSETTING NEW FOR: %s" % str(key)
+        self._new.discard(key)
+
     @property
     def is_writing(self):
         """