]> git.notmuchmail.org Git - notmuch/blobdiff - bindings/python/notmuch/message.py
python: more error handling fixes
[notmuch] / bindings / python / notmuch / message.py
index fa6c58c36afab977bbee58b6762a2a676f550885..28723c10c246c871dcf7fb03d5f534c4d63d4ed8 100644 (file)
@@ -117,7 +117,7 @@ class Messages(object):
         :TODO: Make the iterator work more than once and cache the tags in
                the Python object.(?)
         """
-        if msgs_p is None:
+        if not msgs_p:
             raise NotmuchError(STATUS.NULL_POINTER)
 
         self._msgs = msgs_p
@@ -135,7 +135,7 @@ class Messages(object):
             :meth:`collect_tags` will iterate over the messages and therefore
             will not allow further iterations.
         """
-        if self._msgs is None:
+        if not self._msgs:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         # collect all tags (returns NULL on error)
@@ -160,7 +160,7 @@ class Messages(object):
     _move_to_next.restype = None
 
     def __next__(self):
-        if self._msgs is None:
+        if not self._msgs:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         if not self._valid(self._msgs):
@@ -234,10 +234,10 @@ class Messages(object):
                 next_indent = indent + 1
 
             # get replies and print them also out (if there are any)
-            replies = msg.get_replies()
-            if not replies is None:
+            replies = msg.get_replies().format_messages(format, next_indent, entire_thread)
+            if replies:
                 result.append(set_sep)
-                result.extend(replies.format_messages(format, next_indent, entire_thread))
+                result.extend(replies)
 
             result.append(set_end)
         result.append(set_end)
@@ -255,6 +255,17 @@ class Messages(object):
         """
         handle.write(''.join(self.format_messages(format, indent, entire_thread)))
 
+
+class EmptyMessagesResult(Messages):
+    def __init__(self, parent):
+        self._msgs = None
+        self._parent = parent
+
+    def __next__(self):
+        raise StopIteration()
+    next = __next__
+
+
 class Message(Python3StringMixIn):
     """Represents a single Email message
 
@@ -338,7 +349,7 @@ class Message(Python3StringMixIn):
               automatically delete the parent object once all derived
               objects are dead.
         """
-        if msg_p is None:
+        if not msg_p:
             raise NotmuchError(STATUS.NULL_POINTER)
         self._msg = msg_p
         #keep reference to parent, so we keep it alive
@@ -351,9 +362,9 @@ class Message(Python3StringMixIn):
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
                     is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
-        return Message._get_message_id(self._msg).decode('utf-8', errors='ignore')
+        return Message._get_message_id(self._msg).decode('utf-8', 'ignore')
 
     def get_thread_id(self):
         """Returns the thread ID
@@ -368,10 +379,10 @@ class Message(Python3StringMixIn):
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
                     is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
-        return Message._get_thread_id(self._msg).decode('utf-8', errors='ignore')
+        return Message._get_thread_id(self._msg).decode('utf-8', 'ignore')
 
     def get_replies(self):
         """Gets all direct replies to this message as :class:`Messages`
@@ -385,20 +396,19 @@ class Message(Python3StringMixIn):
             number of subsequent calls to :meth:`get_replies`). If this message
             was obtained through some non-thread means, (such as by a call to
             :meth:`Query.search_messages`), then this function will return
-            `None`.
+            an empty Messages iterator.
 
-        :returns: :class:`Messages` or `None` if there are no replies to
-            this message.
+        :returns: :class:`Messages`.
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
                     is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         msgs_p = Message._get_replies(self._msg)
 
-        if msgs_p is None:
-            return None
+        if not msgs_p:
+            return EmptyMessagesResult(self)
 
         return Messages(msgs_p, self)
 
@@ -414,7 +424,7 @@ class Message(Python3StringMixIn):
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
                     is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
         return Message._get_date(self._msg)
 
@@ -437,14 +447,14 @@ class Message(Python3StringMixIn):
                       is not initialized.
                     * STATUS.NULL_POINTER if any error occured.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         #Returns NULL if any error occurs.
         header = Message._get_header(self._msg, _str(header))
         if header == None:
             raise NotmuchError(STATUS.NULL_POINTER)
-        return header.decode('UTF-8', errors='ignore')
+        return header.decode('UTF-8', 'ignore')
 
     def get_filename(self):
         """Returns the file path of the message file
@@ -453,9 +463,9 @@ class Message(Python3StringMixIn):
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
               is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
-        return Message._get_filename(self._msg).decode('utf-8', errors='ignore')
+        return Message._get_filename(self._msg).decode('utf-8', 'ignore')
 
     def get_filenames(self):
         """Get all filenames for the email corresponding to 'message'
@@ -463,7 +473,7 @@ class Message(Python3StringMixIn):
         Returns a Filenames() generator with all absolute filepaths for
         messages recorded to have the same Message-ID. These files must
         not necessarily have identical content."""
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         files_p = Message._get_filenames(self._msg)
@@ -483,7 +493,7 @@ class Message(Python3StringMixIn):
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
               is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
         return Message._get_flag(self._msg, flag)
 
@@ -498,7 +508,7 @@ class Message(Python3StringMixIn):
         :exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if the message
               is not initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
         self._set_flag(self._msg, flag, value)
 
@@ -512,7 +522,7 @@ class Message(Python3StringMixIn):
                         is not initialized.
                       * STATUS.NULL_POINTER, on error
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         tags_p = Message._get_tags(self._msg)
@@ -555,7 +565,7 @@ class Message(Python3StringMixIn):
                   STATUS.NOT_INITIALIZED
                      The message has not been initialized.
        """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         status = self._add_tag(self._msg, _str(tag))
@@ -603,7 +613,7 @@ class Message(Python3StringMixIn):
                    STATUS.NOT_INITIALIZED
                      The message has not been initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         status = self._remove_tag(self._msg, _str(tag))
@@ -644,7 +654,7 @@ class Message(Python3StringMixIn):
                    STATUS.NOT_INITIALIZED
                      The message has not been initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         status = self._remove_all_tags(self._msg)
@@ -702,7 +712,7 @@ class Message(Python3StringMixIn):
                    STATUS.NOT_INITIALIZED
                      The message has not been initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         status = self._freeze(self._msg)
@@ -741,7 +751,7 @@ class Message(Python3StringMixIn):
                    STATUS.NOT_INITIALIZED
                      The message has not been initialized.
         """
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
 
         status = self._thaw(self._msg)
@@ -777,7 +787,7 @@ class Message(Python3StringMixIn):
 
         :returns: a :class:`STATUS` value. In short, you want to see
             notmuch.STATUS.SUCCESS here. See there for details."""
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
         return Message._tags_to_maildir_flags(self._msg)
 
@@ -804,7 +814,7 @@ class Message(Python3StringMixIn):
 
         :returns: a :class:`STATUS`. In short, you want to see
             notmuch.STATUS.SUCCESS here. See there for details."""
-        if self._msg is None:
+        if not self._msg:
             raise NotmuchError(STATUS.NOT_INITIALIZED)
         return Message._tags_to_maildir_flags(self._msg)
 
@@ -947,7 +957,7 @@ class Message(Python3StringMixIn):
     def __hash__(self):
         """Implement hash(), so we can use Message() sets"""
         file = self.get_filename()
-        if file is None:
+        if not file:
             return None
         return hash(file)