""" This file is part of notmuch. Notmuch is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. Notmuch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with notmuch. If not, see . Copyright 2010 Sebastian Spaeth """ from ctypes import c_char_p from notmuch.globals import ( nmlib, NotmuchMessageP, NotmuchFilenamesP, Python3StringMixIn, ) from .errors import ( NullPointerError, NotInitializedError, ) class Filenames(Python3StringMixIn): """Represents a list of filenames as returned by notmuch This object contains the Filenames iterator. The main function is as_generator() which will return a generator so we can do a Filenamesth an iterator over a list of notmuch filenames. Do note that the underlying library only provides a one-time iterator (it cannot reset the iterator to the start). Thus iterating over the function will "exhaust" the list of tags, and a subsequent iteration attempt will raise a :exc:`NotInitializedError`. Also note, that any function that uses iteration (nearly all) will also exhaust the tags. So both:: for name in filenames: print name as well as:: number_of_names = len(names) and even a simple:: #str() iterates over all tags to construct a space separated list print(str(filenames)) will "exhaust" the Filenames. However, you can use :meth:`Message.get_filenames` repeatedly to get fresh Filenames objects to perform various actions on filenames. """ #notmuch_filenames_get _get = nmlib.notmuch_filenames_get _get.argtypes = [NotmuchFilenamesP] _get.restype = c_char_p def __init__(self, files_p, parent): """ :param files_p: A pointer to an underlying *notmuch_tags_t* structure. These are not publically exposed, so a user will almost never instantiate a :class:`Tags` object herself. They are usually handed back as a result, e.g. in :meth:`Database.get_all_tags`. *tags_p* must be valid, we will raise an :exc:`NullPointerError` if it is `None`. :type files_p: :class:`ctypes.c_void_p` :param parent: The parent object (ie :class:`Message` these filenames are derived from, and saves a reference to it, so we can automatically delete the db object once all derived objects are dead. """ if not files_p: raise NullPointerError() self._files_p = files_p #save reference to parent object so we keep it alive self._parent = parent def __iter__(self): """ Make Filenames an iterator """ return self _valid = nmlib.notmuch_filenames_valid _valid.argtypes = [NotmuchFilenamesP] _valid.restype = bool _move_to_next = nmlib.notmuch_filenames_move_to_next _move_to_next.argtypes = [NotmuchFilenamesP] _move_to_next.restype = None def __next__(self): if not self._files_p: raise NotInitializedError() if not self._valid(self._files_p): self._files_p = None raise StopIteration file_ = Filenames._get(self._files_p) self._move_to_next(self._files_p) return file_.decode('utf-8', 'ignore') next = __next__ # python2.x iterator protocol compatibility def as_generator(self): """Return generator of Filenames This is the main function that will usually be used by the user. .. deprecated:: 0.12 :class:`Filenames` objects implement the iterator protocol. """ return self def __unicode__(self): """Represent Filenames() as newline-separated list of full paths .. note:: As this iterates over the filenames, we will not be able to iterate over them again (as in retrieve them)! If the tags have been exhausted already, this will raise a :exc:`NotInitializedError` on subsequent attempts. However, you can use :meth:`Message.get_filenames` repeatedly to perform various actions on filenames. """ return "\n".join(self) _destroy = nmlib.notmuch_filenames_destroy _destroy.argtypes = [NotmuchMessageP] _destroy.restype = None def __del__(self): """Close and free the notmuch filenames""" if self._files_p is not None: self._destroy(self._files_p) def __len__(self): """len(:class:`Filenames`) returns the number of contained files .. note:: As this iterates over the files, we will not be able to iterate over them again! So this will fail:: #THIS FAILS files = Database().get_directory('').get_child_files() if len(files) > 0: # this 'exhausts' msgs # next line raises # NotmuchError(:attr:`STATUS`.NOT_INITIALIZED) for file in files: print file """ if not self._files_p: raise NotInitializedError() i = 0 while self._valid(self._files_p): self._move_to_next(self._files_p) i += 1 self._files_p = None return i