2010-03-25 08:39:21 +01:00
|
|
|
import os
|
2010-03-25 11:39:31 +01:00
|
|
|
from ctypes import c_int, c_char_p, c_void_p, c_uint, c_long, byref
|
2010-03-17 11:44:36 +01:00
|
|
|
from cnotmuch.globals import nmlib, STATUS, NotmuchError, Enum
|
2010-03-24 11:32:30 +01:00
|
|
|
from cnotmuch.thread import Threads
|
|
|
|
from cnotmuch.message import Messages
|
|
|
|
from cnotmuch.tag import Tags
|
2010-03-15 15:47:15 +01:00
|
|
|
|
|
|
|
class Database(object):
|
2010-03-17 11:45:13 +01:00
|
|
|
"""Represents a notmuch database (wraps notmuch_database_t)
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-17 12:32:22 +01:00
|
|
|
.. note:: Do remember that as soon as we tear down this object,
|
|
|
|
all underlying derived objects such as queries, threads,
|
|
|
|
messages, tags etc will be freed by the underlying library
|
|
|
|
as well. Accessing these objects will lead to segfaults and
|
|
|
|
other unexpected behavior. See above for more details.
|
2010-03-15 15:47:15 +01:00
|
|
|
"""
|
|
|
|
_std_db_path = None
|
2010-03-17 11:45:13 +01:00
|
|
|
"""Class attribute to cache user's default database"""
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-18 18:48:36 +01:00
|
|
|
MODE = Enum(['READ_ONLY','READ_WRITE'])
|
|
|
|
"""Constants: Mode in which to open the database"""
|
|
|
|
|
2010-03-25 11:39:31 +01:00
|
|
|
"""notmuch_database_get_directory"""
|
|
|
|
_get_directory = nmlib.notmuch_database_get_directory
|
|
|
|
_get_directory.restype = c_void_p
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
"""notmuch_database_get_path"""
|
2010-03-15 15:47:15 +01:00
|
|
|
_get_path = nmlib.notmuch_database_get_path
|
|
|
|
_get_path.restype = c_char_p
|
|
|
|
|
2010-03-18 08:50:48 +01:00
|
|
|
"""notmuch_database_get_version"""
|
|
|
|
_get_version = nmlib.notmuch_database_get_version
|
|
|
|
_get_version.restype = c_uint
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
"""notmuch_database_open"""
|
2010-03-15 15:47:15 +01:00
|
|
|
_open = nmlib.notmuch_database_open
|
|
|
|
_open.restype = c_void_p
|
|
|
|
|
2010-03-25 13:13:44 +01:00
|
|
|
"""notmuch_database_upgrade"""
|
|
|
|
_upgrade = nmlib.notmuch_database_upgrade
|
|
|
|
_upgrade.argtypes = [c_void_p, c_void_p, c_void_p]
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
""" notmuch_database_find_message"""
|
2010-03-16 11:27:31 +01:00
|
|
|
_find_message = nmlib.notmuch_database_find_message
|
|
|
|
_find_message.restype = c_void_p
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
"""notmuch_database_get_all_tags"""
|
2010-03-15 15:47:15 +01:00
|
|
|
_get_all_tags = nmlib.notmuch_database_get_all_tags
|
|
|
|
_get_all_tags.restype = c_void_p
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
"""notmuch_database_create"""
|
2010-03-16 11:27:31 +01:00
|
|
|
_create = nmlib.notmuch_database_create
|
|
|
|
_create.restype = c_void_p
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-18 18:48:36 +01:00
|
|
|
def __init__(self, path=None, create=False, mode= 0):
|
2010-03-25 14:28:56 +01:00
|
|
|
"""If *path* is `None`, we will try to read a users notmuch
|
2010-03-18 18:48:36 +01:00
|
|
|
configuration and use his configured database. The location of the
|
|
|
|
configuration file can be specified through the environment variable
|
|
|
|
*NOTMUCH_CONFIG*, falling back to the default `~/.notmuch-config`.
|
|
|
|
|
|
|
|
If *create* is `True`, the database will always be created in
|
|
|
|
:attr:`MODE`.READ_WRITE mode. Default mode for opening is READ_ONLY.
|
2010-03-17 11:45:13 +01:00
|
|
|
|
|
|
|
:param path: Directory to open/create the database in (see
|
|
|
|
above for behavior if `None`)
|
|
|
|
:type path: `str` or `None`
|
2010-03-17 15:14:44 +01:00
|
|
|
:param create: Pass `False` to open an existing, `True` to create a new
|
2010-03-17 11:45:13 +01:00
|
|
|
database.
|
|
|
|
:type create: bool
|
2010-03-17 15:14:44 +01:00
|
|
|
:param mode: Mode to open a database in. Is always
|
2010-03-17 11:45:13 +01:00
|
|
|
:attr:`MODE`.READ_WRITE when creating a new one.
|
|
|
|
:type mode: :attr:`MODE`
|
|
|
|
:returns: Nothing
|
|
|
|
:exception: :exc:`NotmuchError` in case of failure.
|
2010-03-16 11:27:31 +01:00
|
|
|
"""
|
2010-03-15 15:47:15 +01:00
|
|
|
self._db = None
|
2010-03-16 11:27:31 +01:00
|
|
|
if path is None:
|
|
|
|
# no path specified. use a user's default database
|
|
|
|
if Database._std_db_path is None:
|
|
|
|
#the following line throws a NotmuchError if it fails
|
|
|
|
Database._std_db_path = self._get_user_default_db()
|
|
|
|
path = Database._std_db_path
|
|
|
|
|
2010-03-15 15:47:15 +01:00
|
|
|
if create == False:
|
2010-03-17 12:32:22 +01:00
|
|
|
self.open(path, mode)
|
2010-03-15 15:47:15 +01:00
|
|
|
else:
|
2010-03-17 11:45:13 +01:00
|
|
|
self.create(path)
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-19 10:51:35 +01:00
|
|
|
def _verify_initialized_db(self):
|
|
|
|
"""Raises a NotmuchError in case self._db is still None"""
|
|
|
|
if self._db is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
2010-03-17 11:45:13 +01:00
|
|
|
def create(self, path):
|
|
|
|
"""Creates a new notmuch database
|
2010-03-16 11:27:31 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
This function is used by __init__() and usually does not need
|
2010-03-17 12:32:22 +01:00
|
|
|
to be called directly. It wraps the underlying
|
|
|
|
*notmuch_database_create* function and creates a new notmuch
|
2010-03-25 14:28:56 +01:00
|
|
|
database at *path*. It will always return a database in :attr:`MODE`
|
|
|
|
.READ_WRITE mode as creating an empty database for
|
2010-03-17 12:32:22 +01:00
|
|
|
reading only does not make a great deal of sense.
|
2010-03-17 11:45:13 +01:00
|
|
|
|
|
|
|
:param path: A directory in which we should create the database.
|
|
|
|
:type path: str
|
|
|
|
:returns: Nothing
|
|
|
|
:exception: :exc:`NotmuchError` in case of any failure
|
|
|
|
(after printing an error message on stderr).
|
2010-03-16 11:27:31 +01:00
|
|
|
"""
|
2010-03-17 11:45:13 +01:00
|
|
|
if self._db is not None:
|
|
|
|
raise NotmuchError(
|
|
|
|
message="Cannot create db, this Database() already has an open one.")
|
|
|
|
|
2010-03-18 18:48:36 +01:00
|
|
|
res = Database._create(path, Database.MODE.READ_WRITE)
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-16 11:27:31 +01:00
|
|
|
if res is None:
|
|
|
|
raise NotmuchError(
|
|
|
|
message="Could not create the specified database")
|
|
|
|
self._db = res
|
|
|
|
|
2010-03-18 18:48:36 +01:00
|
|
|
def open(self, path, mode= 0):
|
2010-03-17 12:32:22 +01:00
|
|
|
"""Opens an existing database
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
This function is used by __init__() and usually does not need
|
2010-03-17 12:32:22 +01:00
|
|
|
to be called directly. It wraps the underlying
|
|
|
|
*notmuch_database_open* function.
|
|
|
|
|
|
|
|
:param status: Open the database in read-only or read-write mode
|
|
|
|
:type status: :attr:`MODE`
|
|
|
|
:returns: Nothing
|
2010-03-17 15:14:44 +01:00
|
|
|
:exception: Raises :exc:`NotmuchError` in case
|
2010-03-17 12:32:22 +01:00
|
|
|
of any failure (after printing an error message on stderr).
|
2010-03-15 15:47:15 +01:00
|
|
|
"""
|
2010-03-17 11:45:13 +01:00
|
|
|
|
2010-03-17 12:32:22 +01:00
|
|
|
res = Database._open(path, mode)
|
2010-03-15 15:47:15 +01:00
|
|
|
|
|
|
|
if res is None:
|
|
|
|
raise NotmuchError(
|
|
|
|
message="Could not open the specified database")
|
|
|
|
self._db = res
|
|
|
|
|
|
|
|
def get_path(self):
|
2010-03-17 12:32:22 +01:00
|
|
|
"""Returns the file path of an open database
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
Wraps *notmuch_database_get_path*."""
|
2010-03-19 10:51:35 +01:00
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
2010-03-15 15:47:15 +01:00
|
|
|
return Database._get_path(self._db)
|
|
|
|
|
2010-03-18 08:50:48 +01:00
|
|
|
def get_version(self):
|
|
|
|
"""Returns the database format version
|
|
|
|
|
|
|
|
:returns: The database version as positive integer
|
|
|
|
:exception: :exc:`NotmuchError` with STATUS.NOT_INITIALIZED if
|
|
|
|
the database was not intitialized.
|
|
|
|
"""
|
2010-03-19 10:51:35 +01:00
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
2010-03-18 08:50:48 +01:00
|
|
|
|
|
|
|
return Database._get_version (self._db)
|
|
|
|
|
|
|
|
def needs_upgrade(self):
|
|
|
|
"""Does this database need to be upgraded before writing to it?
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
If this function returns `True` then no functions that modify the
|
|
|
|
database (:meth:`add_message`,
|
|
|
|
:meth:`Message.add_tag`, :meth:`Directory.set_mtime`,
|
|
|
|
etc.) will work unless :meth:`upgrade` is called successfully first.
|
2010-03-18 08:50:48 +01:00
|
|
|
|
|
|
|
:returns: `True` or `False`
|
|
|
|
:exception: :exc:`NotmuchError` with STATUS.NOT_INITIALIZED if
|
|
|
|
the database was not intitialized.
|
|
|
|
"""
|
2010-03-19 10:51:35 +01:00
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
2010-03-18 08:50:48 +01:00
|
|
|
|
2010-03-25 11:39:31 +01:00
|
|
|
return notmuch_database_needs_upgrade(self._db)
|
|
|
|
|
2010-03-25 13:13:44 +01:00
|
|
|
def upgrade(self):
|
|
|
|
"""Upgrades the current database
|
|
|
|
|
|
|
|
After opening a database in read-write mode, the client should
|
|
|
|
check if an upgrade is needed (notmuch_database_needs_upgrade) and
|
|
|
|
if so, upgrade with this function before making any modifications.
|
|
|
|
|
|
|
|
NOT IMPLEMENTED: The optional progress_notify callback can be
|
|
|
|
used by the caller to provide progress indication to the
|
|
|
|
user. If non-NULL it will be called periodically with
|
2010-03-25 14:28:56 +01:00
|
|
|
'progress' as a floating-point value in the range of [0.0..1.0]
|
|
|
|
indicating the progress made so far in the upgrade process.
|
|
|
|
|
|
|
|
:TODO: catch exceptions, document return values and etc...
|
2010-03-25 13:13:44 +01:00
|
|
|
"""
|
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
|
|
|
status = Database._upgrade (self._db, None, None)
|
|
|
|
#TODO: catch exceptions, document return values and etc
|
|
|
|
return status
|
|
|
|
|
2010-03-25 11:39:31 +01:00
|
|
|
def get_directory(self, path):
|
|
|
|
"""Returns a :class:`Directory` of path,
|
|
|
|
(creating it if it does not exist(?))
|
|
|
|
|
|
|
|
.. warning:: This call needs a writeable database in
|
|
|
|
Database.MODE.READ_WRITE mode. The underlying library will exit the
|
|
|
|
program if this method is used on a read-only database!
|
|
|
|
|
|
|
|
:param path: A str containing the path relative to the path of database
|
2010-03-25 14:28:56 +01:00
|
|
|
(see :meth:`get_path`), or else should be an absolute path
|
|
|
|
with initial components that match the path of 'database'.
|
2010-03-25 11:39:31 +01:00
|
|
|
:returns: :class:`Directory` or raises an exception.
|
|
|
|
:exception: :exc:`NotmuchError`
|
|
|
|
|
|
|
|
STATUS.NOT_INITIALIZED
|
|
|
|
If the database was not intitialized.
|
|
|
|
|
|
|
|
STATUS.FILE_ERROR
|
|
|
|
If path is not relative database or absolute with initial
|
|
|
|
components same as database.
|
|
|
|
|
|
|
|
"""
|
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
|
|
|
# sanity checking if path is valid, and make path absolute
|
|
|
|
if path[0] == os.sep:
|
|
|
|
# we got an absolute path
|
|
|
|
if not path.startswith(self.get_path()):
|
|
|
|
# but its initial components are not equal to the db path
|
|
|
|
raise NotmuchError(STATUS.FILE_ERROR,
|
|
|
|
message="Database().get_directory() called with a wrong absolute path.")
|
|
|
|
abs_dirpath = path
|
|
|
|
else:
|
|
|
|
#we got a relative path, make it absolute
|
|
|
|
abs_dirpath = os.path.abspath(os.path.join(self.get_path(),path))
|
|
|
|
|
|
|
|
dir_p = Database._get_directory(self._db, path);
|
|
|
|
|
|
|
|
# return the Directory, init it with the absolute path
|
|
|
|
return Directory(abs_dirpath, dir_p, self)
|
2010-03-18 08:50:48 +01:00
|
|
|
|
2010-03-19 10:53:14 +01:00
|
|
|
def add_message(self, filename):
|
|
|
|
"""Adds a new message to the database
|
|
|
|
|
|
|
|
`filename` should be a path relative to the path of the open
|
|
|
|
database (see :meth:`get_path`), or else should be an absolute
|
|
|
|
filename with initial components that match the path of the
|
|
|
|
database.
|
|
|
|
|
|
|
|
The file should be a single mail message (not a multi-message mbox)
|
|
|
|
that is expected to remain at its current location, since the
|
|
|
|
notmuch database will reference the filename, and will not copy the
|
|
|
|
entire contents of the file.
|
|
|
|
|
|
|
|
:returns: On success, we return
|
|
|
|
|
|
|
|
1) a :class:`Message` object that can be used for things
|
|
|
|
such as adding tags to the just-added message.
|
|
|
|
2) one of the following STATUS values:
|
|
|
|
|
|
|
|
STATUS.SUCCESS
|
|
|
|
Message successfully added to database.
|
|
|
|
STATUS.DUPLICATE_MESSAGE_ID
|
|
|
|
Message has the same message ID as another message already
|
|
|
|
in the database. The new filename was successfully added
|
|
|
|
to the message in the database.
|
|
|
|
|
|
|
|
:rtype: 2-tuple(:class:`Message`, STATUS)
|
|
|
|
|
|
|
|
:exception: Raises a :exc:`NotmuchError` with the following meaning.
|
|
|
|
If such an exception occurs, nothing was added to the database.
|
|
|
|
|
|
|
|
STATUS.FILE_ERROR
|
|
|
|
An error occurred trying to open the file, (such as
|
|
|
|
permission denied, or file not found, etc.).
|
|
|
|
STATUS.FILE_NOT_EMAIL
|
|
|
|
The contents of filename don't look like an email message.
|
|
|
|
STATUS.READ_ONLY_DATABASE
|
|
|
|
Database was opened in read-only mode so no message can
|
|
|
|
be added.
|
|
|
|
STATUS.NOT_INITIALIZED
|
|
|
|
The database has not been initialized.
|
|
|
|
"""
|
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
|
|
|
msg_p = c_void_p()
|
|
|
|
status = nmlib.notmuch_database_add_message(self._db,
|
|
|
|
filename,
|
|
|
|
byref(msg_p))
|
|
|
|
|
|
|
|
if not status in [STATUS.SUCCESS,STATUS.DUPLICATE_MESSAGE_ID]:
|
|
|
|
raise NotmuchError(status)
|
|
|
|
|
|
|
|
#construct Message() and return
|
|
|
|
msg = Message(msg_p, self)
|
|
|
|
return (msg, status)
|
|
|
|
|
|
|
|
def remove_message(self, filename):
|
|
|
|
"""Removes a message from the given notmuch database
|
|
|
|
|
|
|
|
Note that only this particular filename association is removed from
|
|
|
|
the database. If the same message (as determined by the message ID)
|
|
|
|
is still available via other filenames, then the message will
|
|
|
|
persist in the database for those filenames. When the last filename
|
|
|
|
is removed for a particular message, the database content for that
|
|
|
|
message will be entirely removed.
|
|
|
|
|
2010-03-25 14:28:56 +01:00
|
|
|
:returns: A STATUS value with the following meaning:
|
2010-03-19 10:53:14 +01:00
|
|
|
|
|
|
|
STATUS.SUCCESS
|
|
|
|
The last filename was removed and the message was removed
|
|
|
|
from the database.
|
|
|
|
STATUS.DUPLICATE_MESSAGE_ID
|
|
|
|
This filename was removed but the message persists in the
|
|
|
|
database with at least one other filename.
|
|
|
|
|
|
|
|
:exception: Raises a :exc:`NotmuchError` with the following meaning.
|
|
|
|
If such an exception occurs, nothing was removed from the database.
|
|
|
|
|
|
|
|
STATUS.READ_ONLY_DATABASE
|
|
|
|
Database was opened in read-only mode so no message can be
|
|
|
|
removed.
|
|
|
|
STATUS.NOT_INITIALIZED
|
|
|
|
The database has not been initialized.
|
|
|
|
"""
|
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
|
|
|
status = nmlib.notmuch_database_remove_message(self._db,
|
|
|
|
filename)
|
|
|
|
|
2010-03-16 11:27:31 +01:00
|
|
|
def find_message(self, msgid):
|
2010-03-17 12:32:22 +01:00
|
|
|
"""Returns a :class:`Message` as identified by its message ID
|
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
Wraps the underlying *notmuch_database_find_message* function.
|
2010-03-16 11:27:31 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
:param msgid: The message ID
|
2010-03-17 11:45:13 +01:00
|
|
|
:type msgid: string
|
|
|
|
:returns: :class:`Message` or `None` if no message is found or if an
|
2010-03-16 11:27:31 +01:00
|
|
|
out-of-memory situation occurs.
|
2010-03-17 11:45:13 +01:00
|
|
|
:exception: :exc:`NotmuchError` with STATUS.NOT_INITIALIZED if
|
|
|
|
the database was not intitialized.
|
2010-03-16 11:27:31 +01:00
|
|
|
"""
|
2010-03-19 10:51:35 +01:00
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
2010-03-16 11:27:31 +01:00
|
|
|
msg_p = Database._find_message(self._db, msgid)
|
|
|
|
if msg_p is None:
|
|
|
|
return None
|
|
|
|
return Message(msg_p, self)
|
2010-03-15 15:47:15 +01:00
|
|
|
|
|
|
|
def get_all_tags(self):
|
2010-03-17 11:45:13 +01:00
|
|
|
"""Returns :class:`Tags` with a list of all tags found in the database
|
2010-03-15 15:47:15 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
:returns: :class:`Tags`
|
|
|
|
:execption: :exc:`NotmuchError` with STATUS.NULL_POINTER on error
|
2010-03-15 15:47:15 +01:00
|
|
|
"""
|
2010-03-19 10:51:35 +01:00
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
2010-03-16 11:27:31 +01:00
|
|
|
|
2010-03-15 15:47:15 +01:00
|
|
|
tags_p = Database._get_all_tags (self._db)
|
|
|
|
if tags_p == None:
|
|
|
|
raise NotmuchError(STATUS.NULL_POINTER)
|
|
|
|
return Tags(tags_p, self)
|
|
|
|
|
2010-03-19 10:55:06 +01:00
|
|
|
def create_query(self, querystring):
|
|
|
|
"""Returns a :class:`Query` derived from this database
|
|
|
|
|
|
|
|
This is a shorthand method for doing::
|
2010-03-19 11:01:19 +01:00
|
|
|
|
2010-03-19 10:55:06 +01:00
|
|
|
# short version
|
|
|
|
# Automatically frees the Database() when 'q' is deleted
|
|
|
|
|
|
|
|
q = Database(dbpath).create_query('from:"Biene Maja"')
|
|
|
|
|
|
|
|
# long version, which is functionally equivalent but will keep the
|
|
|
|
# Database in the 'db' variable around after we delete 'q':
|
|
|
|
|
|
|
|
db = Database(dbpath)
|
|
|
|
q = Query(db,'from:"Biene Maja"')
|
|
|
|
|
|
|
|
This function is a python extension and not in the underlying C API.
|
|
|
|
"""
|
|
|
|
# Raise a NotmuchError if not initialized
|
|
|
|
self._verify_initialized_db()
|
|
|
|
|
2010-03-22 11:19:51 +01:00
|
|
|
return Query(self, querystring)
|
2010-03-19 10:55:06 +01:00
|
|
|
|
2010-03-15 15:47:15 +01:00
|
|
|
def __repr__(self):
|
|
|
|
return "'Notmuch DB " + self.get_path() + "'"
|
|
|
|
|
|
|
|
def __del__(self):
|
|
|
|
"""Close and free the notmuch database if needed"""
|
|
|
|
if self._db is not None:
|
|
|
|
nmlib.notmuch_database_close(self._db)
|
|
|
|
|
2010-03-16 11:27:31 +01:00
|
|
|
def _get_user_default_db(self):
|
|
|
|
""" Reads a user's notmuch config and returns his db location
|
|
|
|
|
|
|
|
Throws a NotmuchError if it cannot find it"""
|
|
|
|
from ConfigParser import SafeConfigParser
|
|
|
|
config = SafeConfigParser()
|
2010-03-18 18:48:36 +01:00
|
|
|
conf_f = os.getenv('NOTMUCH_CONFIG',
|
|
|
|
os.path.expanduser('~/.notmuch-config'))
|
|
|
|
config.read(conf_f)
|
2010-03-16 11:27:31 +01:00
|
|
|
if not config.has_option('database','path'):
|
|
|
|
raise NotmuchError(message=
|
|
|
|
"No DB path specified and no user default found")
|
|
|
|
return config.get('database','path')
|
|
|
|
|
2010-03-15 15:47:15 +01:00
|
|
|
@property
|
|
|
|
def db_p(self):
|
2010-03-19 10:52:15 +01:00
|
|
|
"""Property returning a pointer to `notmuch_database_t` or `None`
|
2010-03-17 11:45:13 +01:00
|
|
|
|
2010-03-19 10:52:15 +01:00
|
|
|
This should normally not be needed by a user (and is not yet
|
|
|
|
guaranteed to remain stable in future versions).
|
|
|
|
"""
|
2010-03-15 15:47:15 +01:00
|
|
|
return self._db
|
|
|
|
|
2010-03-16 12:39:47 +01:00
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class Query(object):
|
2010-03-19 10:52:15 +01:00
|
|
|
"""Represents a search query on an opened :class:`Database`.
|
2010-03-16 12:39:47 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
A query selects and filters a subset of messages from the notmuch
|
|
|
|
database we derive from.
|
|
|
|
|
2010-03-24 11:07:22 +01:00
|
|
|
Query() provides an instance attribute :attr:`sort`, which
|
|
|
|
contains the sort order (if specified via :meth:`set_sort`) or
|
|
|
|
`None`.
|
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
Technically, it wraps the underlying *notmuch_query_t* struct.
|
|
|
|
|
|
|
|
.. note:: Do remember that as soon as we tear down this object,
|
2010-03-17 17:09:58 +01:00
|
|
|
all underlying derived objects such as threads,
|
2010-03-17 15:14:44 +01:00
|
|
|
messages, tags etc will be freed by the underlying library
|
|
|
|
as well. Accessing these objects will lead to segfaults and
|
|
|
|
other unexpected behavior. See above for more details.
|
2010-03-16 12:39:47 +01:00
|
|
|
"""
|
2010-03-16 14:28:05 +01:00
|
|
|
# constants
|
2010-03-17 15:14:44 +01:00
|
|
|
SORT = Enum(['OLDEST_FIRST','NEWEST_FIRST','MESSAGE_ID'])
|
|
|
|
"""Constants: Sort order in which to return results"""
|
2010-03-16 14:28:05 +01:00
|
|
|
|
|
|
|
"""notmuch_query_create"""
|
|
|
|
_create = nmlib.notmuch_query_create
|
|
|
|
_create.restype = c_void_p
|
|
|
|
|
2010-03-24 11:07:22 +01:00
|
|
|
"""notmuch_query_search_threads"""
|
|
|
|
_search_threads = nmlib.notmuch_query_search_threads
|
|
|
|
_search_threads.restype = c_void_p
|
|
|
|
|
2010-03-16 14:28:05 +01:00
|
|
|
"""notmuch_query_search_messages"""
|
|
|
|
_search_messages = nmlib.notmuch_query_search_messages
|
|
|
|
_search_messages.restype = c_void_p
|
|
|
|
|
2010-03-19 08:32:29 +01:00
|
|
|
|
|
|
|
"""notmuch_query_count_messages"""
|
2010-03-19 08:44:34 +01:00
|
|
|
_count_messages = nmlib.notmuch_query_count_messages
|
2010-03-19 08:32:29 +01:00
|
|
|
_count_messages.restype = c_uint
|
|
|
|
|
2010-03-16 12:39:47 +01:00
|
|
|
def __init__(self, db, querystr):
|
2010-03-17 15:14:44 +01:00
|
|
|
"""
|
|
|
|
:param db: An open database which we derive the Query from.
|
|
|
|
:type db: :class:`Database`
|
|
|
|
:param querystr: The query string for the message.
|
|
|
|
:type querystr: str
|
|
|
|
"""
|
2010-03-16 14:28:05 +01:00
|
|
|
self._db = None
|
|
|
|
self._query = None
|
2010-03-24 11:07:22 +01:00
|
|
|
self.sort = None
|
2010-03-16 14:28:05 +01:00
|
|
|
self.create(db, querystr)
|
|
|
|
|
|
|
|
def create(self, db, querystr):
|
2010-03-19 10:52:15 +01:00
|
|
|
"""Creates a new query derived from a Database
|
2010-03-17 15:14:44 +01:00
|
|
|
|
|
|
|
This function is utilized by __init__() and usually does not need to
|
|
|
|
be called directly.
|
2010-03-16 14:28:05 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
:param db: Database to create the query from.
|
|
|
|
:type db: :class:`Database`
|
|
|
|
:param querystr: The query string
|
|
|
|
:type querystr: str
|
|
|
|
:returns: Nothing
|
|
|
|
:exception: :exc:`NotmuchError`
|
|
|
|
|
|
|
|
* STATUS.NOT_INITIALIZED if db is not inited
|
|
|
|
* STATUS.NULL_POINTER if the query creation failed
|
|
|
|
(too little memory)
|
2010-03-16 14:28:05 +01:00
|
|
|
"""
|
|
|
|
if db.db_p is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
# create reference to parent db to keep it alive
|
|
|
|
self._db = db
|
|
|
|
|
|
|
|
# create query, return None if too little mem available
|
|
|
|
query_p = Query._create(db.db_p, querystr)
|
|
|
|
if query_p is None:
|
|
|
|
NotmuchError(STATUS.NULL_POINTER)
|
|
|
|
self._query = query_p
|
|
|
|
|
2010-03-16 16:20:47 +01:00
|
|
|
def set_sort(self, sort):
|
2010-03-17 15:14:44 +01:00
|
|
|
"""Set the sort order future results will be delivered in
|
|
|
|
|
|
|
|
Wraps the underlying *notmuch_query_set_sort* function.
|
2010-03-16 16:20:47 +01:00
|
|
|
|
2010-03-17 15:14:44 +01:00
|
|
|
:param sort: Sort order (see :attr:`Query.SORT`)
|
|
|
|
:returns: Nothing
|
|
|
|
:exception: :exc:`NotmuchError` STATUS.NOT_INITIALIZED if query has not
|
|
|
|
been initialized.
|
2010-03-16 16:20:47 +01:00
|
|
|
"""
|
|
|
|
if self._query is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
2010-03-24 11:07:22 +01:00
|
|
|
self.sort = sort
|
2010-03-16 16:20:47 +01:00
|
|
|
nmlib.notmuch_query_set_sort(self._query, sort)
|
2010-03-16 14:28:05 +01:00
|
|
|
|
2010-03-24 11:07:22 +01:00
|
|
|
def search_threads(self):
|
|
|
|
"""Execute a query for threads
|
|
|
|
|
|
|
|
Execute a query for threads, returning a :class:`Threads` iterator.
|
|
|
|
The returned threads are owned by the query and as such, will only be
|
|
|
|
valid until the Query is deleted.
|
|
|
|
|
2010-03-25 08:39:21 +01:00
|
|
|
The method sets :attr:`Message.FLAG`\.MATCH for those messages that
|
|
|
|
match the query. The method :meth:`Message.get_flag` allows us
|
|
|
|
to get the value of this flag.
|
|
|
|
|
2010-03-24 11:07:22 +01:00
|
|
|
Technically, it wraps the underlying
|
|
|
|
*notmuch_query_search_threads* function.
|
|
|
|
|
|
|
|
:returns: :class:`Threads`
|
|
|
|
:exception: :exc:`NotmuchError`
|
|
|
|
|
|
|
|
* STATUS.NOT_INITIALIZED if query is not inited
|
2010-03-24 11:32:30 +01:00
|
|
|
* STATUS.NULL_POINTER if search_threads failed
|
2010-03-24 11:07:22 +01:00
|
|
|
"""
|
|
|
|
if self._query is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
|
|
|
threads_p = Query._search_threads(self._query)
|
|
|
|
|
|
|
|
if threads_p is None:
|
|
|
|
NotmuchError(STATUS.NULL_POINTER)
|
|
|
|
|
|
|
|
return Threads(threads_p,self)
|
|
|
|
|
2010-03-16 14:28:05 +01:00
|
|
|
def search_messages(self):
|
2010-03-17 17:09:58 +01:00
|
|
|
"""Filter messages according to the query and return
|
|
|
|
:class:`Messages` in the defined sort order
|
2010-03-17 15:14:44 +01:00
|
|
|
|
|
|
|
Technically, it wraps the underlying
|
|
|
|
*notmuch_query_search_messages* function.
|
|
|
|
|
|
|
|
:returns: :class:`Messages`
|
|
|
|
:exception: :exc:`NotmuchError`
|
|
|
|
|
|
|
|
* STATUS.NOT_INITIALIZED if query is not inited
|
|
|
|
* STATUS.NULL_POINTER if search_messages failed
|
2010-03-16 14:28:05 +01:00
|
|
|
"""
|
|
|
|
if self._query is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
|
|
|
msgs_p = Query._search_messages(self._query)
|
|
|
|
|
|
|
|
if msgs_p is None:
|
|
|
|
NotmuchError(STATUS.NULL_POINTER)
|
|
|
|
|
|
|
|
return Messages(msgs_p,self)
|
|
|
|
|
2010-03-19 08:32:29 +01:00
|
|
|
def count_messages(self):
|
|
|
|
"""Estimate the number of messages matching the query
|
|
|
|
|
|
|
|
This function performs a search and returns Xapian's best
|
2010-03-19 08:44:34 +01:00
|
|
|
guess as to the number of matching messages. It is much faster
|
|
|
|
than performing :meth:`search_messages` and counting the
|
|
|
|
result with `len()` (although it always returned the same
|
|
|
|
result in my tests). Technically, it wraps the underlying
|
2010-03-19 08:32:29 +01:00
|
|
|
*notmuch_query_count_messages* function.
|
|
|
|
|
|
|
|
:returns: :class:`Messages`
|
|
|
|
:exception: :exc:`NotmuchError`
|
|
|
|
|
|
|
|
* STATUS.NOT_INITIALIZED if query is not inited
|
|
|
|
"""
|
|
|
|
if self._query is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
|
|
|
return Query._count_messages(self._query)
|
2010-03-16 14:28:05 +01:00
|
|
|
|
|
|
|
def __del__(self):
|
|
|
|
"""Close and free the Query"""
|
|
|
|
if self._query is not None:
|
|
|
|
nmlib.notmuch_query_destroy (self._query)
|
2010-03-25 11:39:31 +01:00
|
|
|
|
|
|
|
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class Directory(object):
|
|
|
|
"""Represents a directory entry in the notmuch directory
|
|
|
|
|
|
|
|
Modifying attributes of this object will modify the
|
|
|
|
database, not the real directory attributes.
|
|
|
|
|
|
|
|
The Directory object is usually derived from another object
|
|
|
|
e.g. via :meth:`Database.get_directory`, and will automatically be
|
|
|
|
become invalid whenever that parent is deleted. You should
|
|
|
|
therefore initialized this object handing it a reference to the
|
|
|
|
parent, preventing the parent from automatically being garbage
|
|
|
|
collected.
|
|
|
|
"""
|
|
|
|
|
|
|
|
"""notmuch_directory_get_mtime"""
|
|
|
|
_get_mtime = nmlib.notmuch_directory_get_mtime
|
|
|
|
_get_mtime.restype = c_long
|
|
|
|
|
|
|
|
"""notmuch_directory_set_mtime"""
|
|
|
|
_set_mtime = nmlib.notmuch_directory_set_mtime
|
|
|
|
_set_mtime.argtypes = [c_char_p, c_long]
|
|
|
|
|
2010-03-25 12:01:20 +01:00
|
|
|
"""notmuch_directory_get_child_files"""
|
|
|
|
_get_child_files = nmlib.notmuch_directory_get_child_files
|
|
|
|
_get_child_files.restype = c_void_p
|
|
|
|
|
2010-03-25 11:39:31 +01:00
|
|
|
"""notmuch_directory_get_child_directories"""
|
|
|
|
_get_child_directories = nmlib.notmuch_directory_get_child_directories
|
|
|
|
_get_child_directories.restype = c_void_p
|
|
|
|
|
|
|
|
def _verify_dir_initialized(self):
|
|
|
|
"""Raises a NotmuchError(STATUS.NOT_INITIALIZED) if the dir_p is None"""
|
|
|
|
if self._dir_p is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
|
|
|
def __init__(self, path, dir_p, parent):
|
|
|
|
"""
|
|
|
|
:param path: The absolute path of the directory object.
|
|
|
|
:param dir_p: The pointer to an internal notmuch_directory_t object.
|
|
|
|
:param parent: The object this Directory is derived from
|
2010-03-25 14:28:56 +01:00
|
|
|
(usually a :class:`Database`). We do not directly use
|
2010-03-25 11:39:31 +01:00
|
|
|
this, but store a reference to it as long as
|
|
|
|
this Directory object lives. This keeps the
|
|
|
|
parent object alive.
|
|
|
|
"""
|
|
|
|
self._path = path
|
|
|
|
self._dir_p = dir_p
|
|
|
|
self._parent = parent
|
|
|
|
|
|
|
|
|
|
|
|
def set_mtime (self, mtime):
|
|
|
|
"""Sets the mtime value of this directory in the database
|
|
|
|
|
|
|
|
The intention is for the caller to use the mtime to allow efficient
|
|
|
|
identification of new messages to be added to the database. The
|
|
|
|
recommended usage is as follows:
|
|
|
|
|
|
|
|
* Read the mtime of a directory from the filesystem
|
|
|
|
|
|
|
|
* Call :meth:`Database.add_message` for all mail files in
|
|
|
|
the directory
|
|
|
|
|
|
|
|
* Call notmuch_directory_set_mtime with the mtime read from the
|
|
|
|
filesystem. Then, when wanting to check for updates to the
|
|
|
|
directory in the future, the client can call :meth:`get_mtime`
|
|
|
|
and know that it only needs to add files if the mtime of the
|
|
|
|
directory and files are newer than the stored timestamp.
|
|
|
|
|
|
|
|
.. note:: :meth:`get_mtime` function does not allow the caller
|
|
|
|
to distinguish a timestamp of 0 from a non-existent
|
|
|
|
timestamp. So don't store a timestamp of 0 unless you are
|
|
|
|
comfortable with that.
|
|
|
|
|
|
|
|
:param mtime: A (time_t) timestamp
|
|
|
|
:returns: Nothing on success, raising an exception on failure.
|
|
|
|
:exception: :exc:`NotmuchError`:
|
|
|
|
|
|
|
|
STATUS.XAPIAN_EXCEPTION
|
|
|
|
A Xapian exception occurred, mtime not stored.
|
|
|
|
STATUS.READ_ONLY_DATABASE
|
|
|
|
Database was opened in read-only mode so directory
|
|
|
|
mtime cannot be modified.
|
|
|
|
STATUS.NOT_INITIALIZED
|
|
|
|
The directory has not been initialized
|
|
|
|
"""
|
|
|
|
#Raise a NotmuchError(STATUS.NOT_INITIALIZED) if the dir_p is None
|
|
|
|
self._verify_dir_initialized()
|
|
|
|
|
|
|
|
#TODO: make sure, we convert the mtime parameter to a 'c_long'
|
|
|
|
status = Directory._set_mtime(self._dir_p, mtime)
|
|
|
|
|
|
|
|
#return on success
|
|
|
|
if status == STATUS.SUCCESS:
|
|
|
|
return
|
|
|
|
#fail with Exception otherwise
|
|
|
|
raise NotmuchError(status)
|
|
|
|
|
|
|
|
def get_mtime (self):
|
|
|
|
"""Gets the mtime value of this directory in the database
|
|
|
|
|
|
|
|
Retrieves a previously stored mtime for this directory.
|
|
|
|
|
|
|
|
:param mtime: A (time_t) timestamp
|
|
|
|
:returns: Nothing on success, raising an exception on failure.
|
|
|
|
:exception: :exc:`NotmuchError`:
|
|
|
|
|
|
|
|
STATUS.NOT_INITIALIZED
|
|
|
|
The directory has not been initialized
|
|
|
|
"""
|
|
|
|
#Raise a NotmuchError(STATUS.NOT_INITIALIZED) if self.dir_p is None
|
|
|
|
self._verify_dir_initialized()
|
|
|
|
|
|
|
|
return Directory._get_mtime (self._dir_p)
|
|
|
|
|
|
|
|
# Make mtime attribute a property of Directory()
|
|
|
|
mtime = property(get_mtime, set_mtime, doc="""Property that allows getting
|
2010-03-25 14:28:56 +01:00
|
|
|
and setting of the Directory *mtime* (read-write)
|
|
|
|
|
|
|
|
See :meth:`get_mtime` and :meth:`set_mtime` for usage and
|
|
|
|
possible exceptions.""")
|
2010-03-25 11:39:31 +01:00
|
|
|
|
|
|
|
def get_child_files(self):
|
2010-03-25 12:01:20 +01:00
|
|
|
"""Gets a Filenames iterator listing all the filenames of
|
|
|
|
messages in the database within the given directory.
|
2010-03-25 11:39:31 +01:00
|
|
|
|
2010-03-25 12:01:20 +01:00
|
|
|
The returned filenames will be the basename-entries only (not
|
|
|
|
complete paths.
|
|
|
|
"""
|
|
|
|
#Raise a NotmuchError(STATUS.NOT_INITIALIZED) if self._dir_p is None
|
|
|
|
self._verify_dir_initialized()
|
|
|
|
|
|
|
|
files_p = Directory._get_child_files(self._dir_p)
|
|
|
|
return Filenames(files_p, self)
|
2010-03-25 11:39:31 +01:00
|
|
|
|
|
|
|
def get_child_directories(self):
|
2010-03-25 14:28:56 +01:00
|
|
|
"""Gets a :class:`Filenames` iterator listing all the filenames of
|
2010-03-25 11:39:31 +01:00
|
|
|
sub-directories in the database within the given directory
|
|
|
|
|
|
|
|
The returned filenames will be the basename-entries only (not
|
|
|
|
complete paths.
|
|
|
|
"""
|
2010-03-25 12:01:20 +01:00
|
|
|
#Raise a NotmuchError(STATUS.NOT_INITIALIZED) if self._dir_p is None
|
|
|
|
self._verify_dir_initialized()
|
|
|
|
|
|
|
|
files_p = Directory._get_child_directories(self._dir_p)
|
|
|
|
return Filenames(files_p, self)
|
2010-03-25 11:39:31 +01:00
|
|
|
|
|
|
|
@property
|
|
|
|
def path(self):
|
2010-03-25 14:28:56 +01:00
|
|
|
"""Returns the absolute path of this Directory (read-only)"""
|
2010-03-25 11:39:31 +01:00
|
|
|
return self._path
|
|
|
|
|
|
|
|
def __repr__(self):
|
|
|
|
"""Object representation"""
|
|
|
|
return "<cnotmuch Directory object '%s'>" % self._path
|
|
|
|
|
|
|
|
def __del__(self):
|
|
|
|
"""Close and free the Directory"""
|
|
|
|
if self._dir_p is not None:
|
|
|
|
nmlib.notmuch_directory_destroy(self._dir_p)
|
2010-03-25 12:01:20 +01:00
|
|
|
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class Filenames(object):
|
2010-03-25 14:28:56 +01:00
|
|
|
"""An iterator over File- or Directory names that are stored in the database
|
2010-03-25 12:01:20 +01:00
|
|
|
"""
|
|
|
|
|
|
|
|
#notmuch_filenames_get
|
|
|
|
_get = nmlib.notmuch_filenames_get
|
|
|
|
_get.restype = c_char_p
|
|
|
|
|
|
|
|
def __init__(self, files_p, parent):
|
|
|
|
"""
|
|
|
|
:param files_p: The pointer to an internal notmuch_filenames_t object.
|
|
|
|
:param parent: The object this Directory is derived from
|
|
|
|
(usually a Directory()). We do not directly use
|
|
|
|
this, but store a reference to it as long as
|
|
|
|
this Directory object lives. This keeps the
|
|
|
|
parent object alive.
|
|
|
|
"""
|
|
|
|
self._files_p = files_p
|
|
|
|
self._parent = parent
|
|
|
|
|
|
|
|
def __iter__(self):
|
|
|
|
""" Make Filenames an iterator """
|
|
|
|
return self
|
|
|
|
|
|
|
|
def next(self):
|
|
|
|
if self._files_p is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
|
|
|
if not nmlib.notmuch_filenames_valid(self._files_p):
|
|
|
|
self._files_p = None
|
|
|
|
raise StopIteration
|
|
|
|
|
|
|
|
file = Filenames._get (self._files_p)
|
|
|
|
nmlib.notmuch_filenames_move_to_next(self._files_p)
|
|
|
|
return file
|
|
|
|
|
|
|
|
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(STATUS.NOT_INITIALIZED)!!!
|
|
|
|
for file in files: print file
|
|
|
|
"""
|
|
|
|
if self._files_p is None:
|
|
|
|
raise NotmuchError(STATUS.NOT_INITIALIZED)
|
|
|
|
|
|
|
|
i=0
|
|
|
|
while nmlib.notmuch_filenames_valid(self._files_p):
|
|
|
|
nmlib.notmuch_filenames_move_to_next(self._files_p)
|
|
|
|
i += 1
|
|
|
|
self._files_p = None
|
|
|
|
return i
|
|
|
|
|
|
|
|
def __del__(self):
|
|
|
|
"""Close and free Filenames"""
|
|
|
|
if self._files_p is not None:
|
|
|
|
nmlib.notmuch_filenames_destroy(self._files_p)
|