mirror of
https://git.notmuchmail.org/git/notmuch
synced 2024-11-30 22:54:11 +01:00
2d895a0119
This reverses the logic of StandaloneMessage to instead create a OwnedMessage. Only the Thread class allows retrieving messages more then once so it can explicitly create such messages. The added test fails with SIGABRT without the fix for the message re-use in threads being present.
710 lines
25 KiB
Python
710 lines
25 KiB
Python
import collections
|
|
import contextlib
|
|
import os
|
|
import pathlib
|
|
import weakref
|
|
|
|
import notmuch2._base as base
|
|
import notmuch2._capi as capi
|
|
import notmuch2._errors as errors
|
|
import notmuch2._tags as tags
|
|
|
|
|
|
__all__ = ['Message']
|
|
|
|
|
|
class Message(base.NotmuchObject):
|
|
"""An email message stored in the notmuch database retrieved via a query.
|
|
|
|
This should not be directly created, instead it will be returned
|
|
by calling methods on :class:`Database`. A message keeps a
|
|
reference to the database object since the database object can not
|
|
be released while the message is in use.
|
|
|
|
Note that this represents a message in the notmuch database. For
|
|
full email functionality you may want to use the :mod:`email`
|
|
package from Python's standard library. You could e.g. create
|
|
this as such::
|
|
|
|
notmuch_msg = db.get_message(msgid) # or from a query
|
|
parser = email.parser.BytesParser(policy=email.policy.default)
|
|
with notmuch_msg.path.open('rb) as fp:
|
|
email_msg = parser.parse(fp)
|
|
|
|
Most commonly the functionality provided by notmuch is sufficient
|
|
to read email however.
|
|
|
|
Messages are considered equal when they have the same message ID.
|
|
This is how libnotmuch treats messages as well, the
|
|
:meth:`pathnames` function returns multiple results for
|
|
duplicates.
|
|
|
|
:param parent: The parent object. This is probably one off a
|
|
:class:`Database`, :class:`Thread` or :class:`Query`.
|
|
:type parent: NotmuchObject
|
|
:param db: The database instance this message is associated with.
|
|
This could be the same as the parent.
|
|
:type db: Database
|
|
:param msg_p: The C pointer to the ``notmuch_message_t``.
|
|
:type msg_p: <cdata>
|
|
:param dup: Whether the message was a duplicate on insertion.
|
|
:type dup: None or bool
|
|
"""
|
|
_msg_p = base.MemoryPointer()
|
|
|
|
def __init__(self, parent, msg_p, *, db):
|
|
self._parent = parent
|
|
self._msg_p = msg_p
|
|
self._db = db
|
|
|
|
@property
|
|
def alive(self):
|
|
if not self._parent.alive:
|
|
return False
|
|
try:
|
|
self._msg_p
|
|
except errors.ObjectDestroyedError:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def __del__(self):
|
|
self._destroy()
|
|
|
|
def _destroy(self):
|
|
if self.alive:
|
|
capi.lib.notmuch_message_destroy(self._msg_p)
|
|
self._msg_p = None
|
|
|
|
@property
|
|
def messageid(self):
|
|
"""The message ID as a string.
|
|
|
|
The message ID is decoded with the ignore error handler. This
|
|
is fine as long as the message ID is well formed. If it is
|
|
not valid ASCII then this will be lossy. So if you need to be
|
|
able to write the exact same message ID back you should use
|
|
:attr:`messageidb`.
|
|
|
|
Note that notmuch will decode the message ID value and thus
|
|
strip off the surrounding ``<`` and ``>`` characters. This is
|
|
different from Python's :mod:`email` package behaviour which
|
|
leaves these characters in place.
|
|
|
|
:returns: The message ID.
|
|
:rtype: :class:`BinString`, this is a normal str but calling
|
|
bytes() on it will return the original bytes used to create
|
|
it.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_get_message_id(self._msg_p)
|
|
return base.BinString(capi.ffi.string(ret))
|
|
|
|
@property
|
|
def threadid(self):
|
|
"""The thread ID.
|
|
|
|
The thread ID is decoded with the surrogateescape error
|
|
handler so that it is possible to reconstruct the original
|
|
thread ID if it is not valid UTF-8.
|
|
|
|
:returns: The thread ID.
|
|
:rtype: :class:`BinString`, this is a normal str but calling
|
|
bytes() on it will return the original bytes used to create
|
|
it.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_get_thread_id(self._msg_p)
|
|
return base.BinString(capi.ffi.string(ret))
|
|
|
|
@property
|
|
def path(self):
|
|
"""A pathname of the message as a pathlib.Path instance.
|
|
|
|
If multiple files in the database contain the same message ID
|
|
this will be just one of the files, chosen at random.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_get_filename(self._msg_p)
|
|
return pathlib.Path(os.fsdecode(capi.ffi.string(ret)))
|
|
|
|
@property
|
|
def pathb(self):
|
|
"""A pathname of the message as a bytes object.
|
|
|
|
See :attr:`path` for details, this is the same but does return
|
|
the path as a bytes object which is faster but less convenient.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_get_filename(self._msg_p)
|
|
return capi.ffi.string(ret)
|
|
|
|
def filenames(self):
|
|
"""Return an iterator of all files for this message.
|
|
|
|
If multiple files contained the same message ID they will all
|
|
be returned here. The files are returned as intances of
|
|
:class:`pathlib.Path`.
|
|
|
|
:returns: Iterator yielding :class:`pathlib.Path` instances.
|
|
:rtype: iter
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
fnames_p = capi.lib.notmuch_message_get_filenames(self._msg_p)
|
|
return PathIter(self, fnames_p)
|
|
|
|
def filenamesb(self):
|
|
"""Return an iterator of all files for this message.
|
|
|
|
This is like :meth:`pathnames` but the files are returned as
|
|
byte objects instead.
|
|
|
|
:returns: Iterator yielding :class:`bytes` instances.
|
|
:rtype: iter
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
fnames_p = capi.lib.notmuch_message_get_filenames(self._msg_p)
|
|
return FilenamesIter(self, fnames_p)
|
|
|
|
@property
|
|
def ghost(self):
|
|
"""Indicates whether this message is a ghost message.
|
|
|
|
A ghost message if a message which we know exists, but it has
|
|
no files or content associated with it. This can happen if
|
|
it was referenced by some other message. Only the
|
|
:attr:`messageid` and :attr:`threadid` attributes are valid
|
|
for it.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_get_flag(
|
|
self._msg_p, capi.lib.NOTMUCH_MESSAGE_FLAG_GHOST)
|
|
return bool(ret)
|
|
|
|
@property
|
|
def excluded(self):
|
|
"""Indicates whether this message was excluded from the query.
|
|
|
|
When a message is created from a search, sometimes messages
|
|
that where excluded by the search query could still be
|
|
returned by it, e.g. because they are part of a thread
|
|
matching the query. the :meth:`Database.query` method allows
|
|
these messages to be flagged, which results in this property
|
|
being set to *True*.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_get_flag(
|
|
self._msg_p, capi.lib.NOTMUCH_MESSAGE_FLAG_EXCLUDED)
|
|
return bool(ret)
|
|
|
|
@property
|
|
def date(self):
|
|
"""The message date as an integer.
|
|
|
|
The time the message was sent as an integer number of seconds
|
|
since the *epoch*, 1 Jan 1970. This is derived from the
|
|
message's header, you can get the original header value with
|
|
:meth:`header`.
|
|
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
return capi.lib.notmuch_message_get_date(self._msg_p)
|
|
|
|
def header(self, name):
|
|
"""Return the value of the named header.
|
|
|
|
Returns the header from notmuch, some common headers are
|
|
stored in the database, others are read from the file.
|
|
Headers are returned with their newlines stripped and
|
|
collapsed concatenated together if they occur multiple times.
|
|
You may be better off using the standard library email
|
|
package's ``email.message_from_file(msg.path.open())`` if that
|
|
is not sufficient for you.
|
|
|
|
:param header: Case-insensitive header name to retrieve.
|
|
:type header: str or bytes
|
|
|
|
:returns: The header value, an empty string if the header is
|
|
not present.
|
|
:rtype: str
|
|
|
|
:raises LookupError: if the header is not present.
|
|
:raises NullPointerError: For unexpected notmuch errors.
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
# The returned is supposedly guaranteed to be UTF-8. Header
|
|
# names must be ASCII as per RFC x822.
|
|
if isinstance(name, str):
|
|
name = name.encode('ascii')
|
|
ret = capi.lib.notmuch_message_get_header(self._msg_p, name)
|
|
if ret == capi.ffi.NULL:
|
|
raise errors.NullPointerError()
|
|
hdr = capi.ffi.string(ret)
|
|
if not hdr:
|
|
raise LookupError
|
|
return hdr.decode(encoding='utf-8')
|
|
|
|
@property
|
|
def tags(self):
|
|
"""The tags associated with the message.
|
|
|
|
This behaves as a set. But removing and adding items to the
|
|
set removes and adds them to the message in the database.
|
|
|
|
:raises ReadOnlyDatabaseError: When manipulating tags on a
|
|
database opened in read-only mode.
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
try:
|
|
ref = self._cached_tagset
|
|
except AttributeError:
|
|
tagset = None
|
|
else:
|
|
tagset = ref()
|
|
if tagset is None:
|
|
tagset = tags.MutableTagSet(
|
|
self, '_msg_p', capi.lib.notmuch_message_get_tags)
|
|
self._cached_tagset = weakref.ref(tagset)
|
|
return tagset
|
|
|
|
@contextlib.contextmanager
|
|
def frozen(self):
|
|
"""Context manager to freeze the message state.
|
|
|
|
This allows you to perform atomic tag updates::
|
|
|
|
with msg.frozen():
|
|
msg.tags.clear()
|
|
msg.tags.add('foo')
|
|
|
|
Using This would ensure the message never ends up with no tags
|
|
applied at all.
|
|
|
|
It is safe to nest calls to this context manager.
|
|
|
|
:raises ReadOnlyDatabaseError: if the database is opened in
|
|
read-only mode.
|
|
:raises UnbalancedFreezeThawError: if you somehow managed to
|
|
call __exit__ of this context manager more than once. Why
|
|
did you do that?
|
|
:raises ObjectDestroyedError: if used after destroyed.
|
|
"""
|
|
ret = capi.lib.notmuch_message_freeze(self._msg_p)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
self._frozen = True
|
|
try:
|
|
yield
|
|
except Exception:
|
|
# Only way to "rollback" these changes is to destroy
|
|
# ourselves and re-create. Behold.
|
|
msgid = self.messageid
|
|
self._destroy()
|
|
with contextlib.suppress(Exception):
|
|
new = self._db.find(msgid)
|
|
self._msg_p = new._msg_p
|
|
new._msg_p = None
|
|
del new
|
|
raise
|
|
else:
|
|
ret = capi.lib.notmuch_message_thaw(self._msg_p)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
self._frozen = False
|
|
|
|
@property
|
|
def properties(self):
|
|
"""A map of arbitrary key-value pairs associated with the message.
|
|
|
|
Be aware that properties may be used by other extensions to
|
|
store state in. So delete or modify with care.
|
|
|
|
The properties map is somewhat special. It is essentially a
|
|
multimap-like structure where each key can have multiple
|
|
values. Therefore accessing a single item using
|
|
:meth:`PropertiesMap.get` or :meth:`PropertiesMap.__getitem__`
|
|
will only return you the *first* item if there are multiple
|
|
and thus are only recommended if you know there to be only one
|
|
value.
|
|
|
|
Instead the map has an additional :meth:`PropertiesMap.all`
|
|
method which can be used to retrieve all properties of a given
|
|
key. This method also allows iterating of a a subset of the
|
|
keys starting with a given prefix.
|
|
"""
|
|
try:
|
|
ref = self._cached_props
|
|
except AttributeError:
|
|
props = None
|
|
else:
|
|
props = ref()
|
|
if props is None:
|
|
props = PropertiesMap(self, '_msg_p')
|
|
self._cached_props = weakref.ref(props)
|
|
return props
|
|
|
|
def replies(self):
|
|
"""Return an iterator of all replies to this message.
|
|
|
|
This method will only work if the message was created from a
|
|
thread. Otherwise it will yield no results.
|
|
|
|
:returns: An iterator yielding :class:`Message` instances.
|
|
:rtype: MessageIter
|
|
"""
|
|
# The notmuch_messages_valid call accepts NULL and this will
|
|
# become an empty iterator, raising StopIteration immediately.
|
|
# Hence no return value checking here.
|
|
msgs_p = capi.lib.notmuch_message_get_replies(self._msg_p)
|
|
return MessageIter(self, msgs_p, db=self._db)
|
|
|
|
def __hash__(self):
|
|
return hash(self.messageid)
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(other, self.__class__):
|
|
return self.messageid == other.messageid
|
|
|
|
|
|
class OwnedMessage(Message):
|
|
"""An email message owned by parent thread object.
|
|
|
|
This subclass of Message is used for messages that are retrieved
|
|
from the notmuch database via a parent :class:`notmuch2.Thread`
|
|
object, which "owns" this message. This means that when this
|
|
message object is destroyed, by calling :func:`del` or
|
|
:meth:`_destroy` directly or indirectly, the message is not freed
|
|
in the notmuch API and the parent :class:`notmuch2.Thread` object
|
|
can return the same object again when needed.
|
|
"""
|
|
|
|
@property
|
|
def alive(self):
|
|
return self._parent.alive
|
|
|
|
def _destroy(self):
|
|
pass
|
|
|
|
|
|
class FilenamesIter(base.NotmuchIter):
|
|
"""Iterator for binary filenames objects."""
|
|
|
|
def __init__(self, parent, iter_p):
|
|
super().__init__(parent, iter_p,
|
|
fn_destroy=capi.lib.notmuch_filenames_destroy,
|
|
fn_valid=capi.lib.notmuch_filenames_valid,
|
|
fn_get=capi.lib.notmuch_filenames_get,
|
|
fn_next=capi.lib.notmuch_filenames_move_to_next)
|
|
|
|
def __next__(self):
|
|
fname = super().__next__()
|
|
return capi.ffi.string(fname)
|
|
|
|
|
|
class PathIter(FilenamesIter):
|
|
"""Iterator for pathlib.Path objects."""
|
|
|
|
def __next__(self):
|
|
fname = super().__next__()
|
|
return pathlib.Path(os.fsdecode(fname))
|
|
|
|
|
|
class PropertiesMap(base.NotmuchObject, collections.abc.MutableMapping):
|
|
"""A mutable mapping to manage properties.
|
|
|
|
Both keys and values of properties are supposed to be UTF-8
|
|
strings in libnotmuch. However since the uderlying API uses
|
|
bytestrings you can use either str or bytes to represent keys and
|
|
all returned keys and values use :class:`BinString`.
|
|
|
|
Also be aware that ``iter(this_map)`` will return duplicate keys,
|
|
while the :class:`collections.abc.KeysView` returned by
|
|
:meth:`keys` is a :class:`collections.abc.Set` subclass. This
|
|
means the former will yield duplicate keys while the latter won't.
|
|
It also means ``len(list(iter(this_map)))`` could be different
|
|
than ``len(this_map.keys())``. ``len(this_map)`` will correspond
|
|
with the lenght of the default iterator.
|
|
|
|
Be aware that libnotmuch exposes all of this as iterators, so
|
|
quite a few operations have O(n) performance instead of the usual
|
|
O(1).
|
|
"""
|
|
Property = collections.namedtuple('Property', ['key', 'value'])
|
|
_marker = object()
|
|
|
|
def __init__(self, msg, ptr_name):
|
|
self._msg = msg
|
|
self._ptr = lambda: getattr(msg, ptr_name)
|
|
|
|
@property
|
|
def alive(self):
|
|
if not self._msg.alive:
|
|
return False
|
|
try:
|
|
self._ptr
|
|
except errors.ObjectDestroyedError:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def _destroy(self):
|
|
pass
|
|
|
|
def __iter__(self):
|
|
"""Return an iterator which iterates over the keys.
|
|
|
|
Be aware that a single key may have multiple values associated
|
|
with it, if so it will appear multiple times here.
|
|
"""
|
|
iter_p = capi.lib.notmuch_message_get_properties(self._ptr(), b'', 0)
|
|
return PropertiesKeyIter(self, iter_p)
|
|
|
|
def __len__(self):
|
|
iter_p = capi.lib.notmuch_message_get_properties(self._ptr(), b'', 0)
|
|
it = base.NotmuchIter(
|
|
self, iter_p,
|
|
fn_destroy=capi.lib.notmuch_message_properties_destroy,
|
|
fn_valid=capi.lib.notmuch_message_properties_valid,
|
|
fn_get=capi.lib.notmuch_message_properties_key,
|
|
fn_next=capi.lib.notmuch_message_properties_move_to_next,
|
|
)
|
|
return len(list(it))
|
|
|
|
def __getitem__(self, key):
|
|
"""Return **the first** peroperty associated with a key."""
|
|
if isinstance(key, str):
|
|
key = key.encode('utf-8')
|
|
value_pp = capi.ffi.new('char**')
|
|
ret = capi.lib.notmuch_message_get_property(self._ptr(), key, value_pp)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
if value_pp[0] == capi.ffi.NULL:
|
|
raise KeyError
|
|
return base.BinString.from_cffi(value_pp[0])
|
|
|
|
def keys(self):
|
|
"""Return a :class:`collections.abc.KeysView` for this map.
|
|
|
|
Even when keys occur multiple times this is a subset of set()
|
|
so will only contain them once.
|
|
"""
|
|
return collections.abc.KeysView({k: None for k in self})
|
|
|
|
def items(self):
|
|
"""Return a :class:`collections.abc.ItemsView` for this map.
|
|
|
|
The ItemsView treats a ``(key, value)`` pair as unique, so
|
|
dupcliate ``(key, value)`` pairs will be merged together.
|
|
However duplicate keys with different values will be returned.
|
|
"""
|
|
items = set()
|
|
props_p = capi.lib.notmuch_message_get_properties(self._ptr(), b'', 0)
|
|
while capi.lib.notmuch_message_properties_valid(props_p):
|
|
key = capi.lib.notmuch_message_properties_key(props_p)
|
|
value = capi.lib.notmuch_message_properties_value(props_p)
|
|
items.add((base.BinString.from_cffi(key),
|
|
base.BinString.from_cffi(value)))
|
|
capi.lib.notmuch_message_properties_move_to_next(props_p)
|
|
capi.lib.notmuch_message_properties_destroy(props_p)
|
|
return PropertiesItemsView(items)
|
|
|
|
def values(self):
|
|
"""Return a :class:`collecions.abc.ValuesView` for this map.
|
|
|
|
All unique property values are included in the view.
|
|
"""
|
|
values = set()
|
|
props_p = capi.lib.notmuch_message_get_properties(self._ptr(), b'', 0)
|
|
while capi.lib.notmuch_message_properties_valid(props_p):
|
|
value = capi.lib.notmuch_message_properties_value(props_p)
|
|
values.add(base.BinString.from_cffi(value))
|
|
capi.lib.notmuch_message_properties_move_to_next(props_p)
|
|
capi.lib.notmuch_message_properties_destroy(props_p)
|
|
return PropertiesValuesView(values)
|
|
|
|
def __setitem__(self, key, value):
|
|
"""Add a key-value pair to the properties.
|
|
|
|
You may prefer to use :meth:`add` for clarity since this
|
|
method usually implies implicit overwriting of an existing key
|
|
if it exists, while for properties this is not the case.
|
|
"""
|
|
self.add(key, value)
|
|
|
|
def add(self, key, value):
|
|
"""Add a key-value pair to the properties."""
|
|
if isinstance(key, str):
|
|
key = key.encode('utf-8')
|
|
if isinstance(value, str):
|
|
value = value.encode('utf-8')
|
|
ret = capi.lib.notmuch_message_add_property(self._ptr(), key, value)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
|
|
def __delitem__(self, key):
|
|
"""Remove all properties with this key."""
|
|
if isinstance(key, str):
|
|
key = key.encode('utf-8')
|
|
ret = capi.lib.notmuch_message_remove_all_properties(self._ptr(), key)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
|
|
def remove(self, key, value):
|
|
"""Remove a key-value pair from the properties."""
|
|
if isinstance(key, str):
|
|
key = key.encode('utf-8')
|
|
if isinstance(value, str):
|
|
value = value.encode('utf-8')
|
|
ret = capi.lib.notmuch_message_remove_property(self._ptr(), key, value)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
|
|
def pop(self, key, default=_marker):
|
|
try:
|
|
value = self[key]
|
|
except KeyError:
|
|
if default is self._marker:
|
|
raise
|
|
else:
|
|
return default
|
|
else:
|
|
self.remove(key, value)
|
|
return value
|
|
|
|
def popitem(self):
|
|
try:
|
|
key = next(iter(self))
|
|
except StopIteration:
|
|
raise KeyError
|
|
value = self.pop(key)
|
|
return (key, value)
|
|
|
|
def clear(self):
|
|
ret = capi.lib.notmuch_message_remove_all_properties(self._ptr(),
|
|
capi.ffi.NULL)
|
|
if ret != capi.lib.NOTMUCH_STATUS_SUCCESS:
|
|
raise errors.NotmuchError(ret)
|
|
|
|
def getall(self, prefix='', *, exact=False):
|
|
"""Return an iterator yielding all properties for a given key prefix.
|
|
|
|
The returned iterator yields all peroperties which start with
|
|
a given key prefix as ``(key, value)`` namedtuples. If called
|
|
with ``exact=True`` then only properties which exactly match
|
|
the prefix are returned, those a key longer than the prefix
|
|
will not be included.
|
|
|
|
:param prefix: The prefix of the key.
|
|
"""
|
|
if isinstance(prefix, str):
|
|
prefix = prefix.encode('utf-8')
|
|
props_p = capi.lib.notmuch_message_get_properties(self._ptr(),
|
|
prefix, exact)
|
|
return PropertiesIter(self, props_p)
|
|
|
|
|
|
class PropertiesKeyIter(base.NotmuchIter):
|
|
|
|
def __init__(self, parent, iter_p):
|
|
super().__init__(
|
|
parent,
|
|
iter_p,
|
|
fn_destroy=capi.lib.notmuch_message_properties_destroy,
|
|
fn_valid=capi.lib.notmuch_message_properties_valid,
|
|
fn_get=capi.lib.notmuch_message_properties_key,
|
|
fn_next=capi.lib.notmuch_message_properties_move_to_next)
|
|
|
|
def __next__(self):
|
|
item = super().__next__()
|
|
return base.BinString.from_cffi(item)
|
|
|
|
|
|
class PropertiesIter(base.NotmuchIter):
|
|
|
|
def __init__(self, parent, iter_p):
|
|
super().__init__(
|
|
parent,
|
|
iter_p,
|
|
fn_destroy=capi.lib.notmuch_message_properties_destroy,
|
|
fn_valid=capi.lib.notmuch_message_properties_valid,
|
|
fn_get=capi.lib.notmuch_message_properties_key,
|
|
fn_next=capi.lib.notmuch_message_properties_move_to_next,
|
|
)
|
|
|
|
def __next__(self):
|
|
if not self._fn_valid(self._iter_p):
|
|
self._destroy()
|
|
raise StopIteration
|
|
key = capi.lib.notmuch_message_properties_key(self._iter_p)
|
|
value = capi.lib.notmuch_message_properties_value(self._iter_p)
|
|
capi.lib.notmuch_message_properties_move_to_next(self._iter_p)
|
|
return PropertiesMap.Property(base.BinString.from_cffi(key),
|
|
base.BinString.from_cffi(value))
|
|
|
|
|
|
class PropertiesItemsView(collections.abc.Set):
|
|
|
|
__slots__ = ('_items',)
|
|
|
|
def __init__(self, items):
|
|
self._items = items
|
|
|
|
@classmethod
|
|
def _from_iterable(self, it):
|
|
return set(it)
|
|
|
|
def __len__(self):
|
|
return len(self._items)
|
|
|
|
def __contains__(self, item):
|
|
return item in self._items
|
|
|
|
def __iter__(self):
|
|
yield from self._items
|
|
|
|
|
|
collections.abc.ItemsView.register(PropertiesItemsView)
|
|
|
|
|
|
class PropertiesValuesView(collections.abc.Set):
|
|
|
|
__slots__ = ('_values',)
|
|
|
|
def __init__(self, values):
|
|
self._values = values
|
|
|
|
def __len__(self):
|
|
return len(self._values)
|
|
|
|
def __contains__(self, value):
|
|
return value in self._values
|
|
|
|
def __iter__(self):
|
|
yield from self._values
|
|
|
|
|
|
collections.abc.ValuesView.register(PropertiesValuesView)
|
|
|
|
|
|
class MessageIter(base.NotmuchIter):
|
|
|
|
def __init__(self, parent, msgs_p, *, db, msg_cls=Message):
|
|
self._db = db
|
|
self._msg_cls = msg_cls
|
|
super().__init__(parent, msgs_p,
|
|
fn_destroy=capi.lib.notmuch_messages_destroy,
|
|
fn_valid=capi.lib.notmuch_messages_valid,
|
|
fn_get=capi.lib.notmuch_messages_get,
|
|
fn_next=capi.lib.notmuch_messages_move_to_next)
|
|
|
|
def __next__(self):
|
|
msg_p = super().__next__()
|
|
return self._msg_cls(self, msg_p, db=self._db)
|