#!/usr/bin/env python
-"""This is a notmuch implementation in python. It's goal is to allow running the test suite on the cnotmuch python bindings."""
+"""This is a notmuch implementation in python. It's goal is to allow running the test suite on the cnotmuch python bindings.
+
+This "binary" honors the NOTMUCH_CONFIG environmen variable for reading a user's
+notmuch configuration (e.g. the database path)
+
+This code is licensed under the GNU GPL v3+."""
+from __future__ import with_statement # This isn't required in Python 2.6
import sys, os, re, logging
+from subprocess import call
from cnotmuch.notmuch import Database, Query
PREFIX=re.compile('(\w+):(.*$)')
#TODO Handle variable: NOTMUCH-CONFIG
if __name__ == '__main__':
# Handle command line options
- # No option
+ #-------------------------------------
+ # No option given, print USAGE and exit
if len(sys.argv) == 1:
print USAGE
-
+ #-------------------------------------
elif sys.argv[1] == 'setup':
- """ Interactively setup notmuch for first use. """
+ """Interactively setup notmuch for first use."""
print "Not implemented."
-
+ #-------------------------------------
+ elif sys.argv[1] == 'new':
+ """ Interactively setup notmuch for first use. """
+ #print "Not implemented. We cheat by calling the proper notmuch"
+ call(['notmuch new'],shell=True)
+ #-------------------------------------
elif sys.argv[1] == 'help':
if len(sys.argv) == 2: print HELPTEXT
else: print "Not implemented"
-
- elif sys.argv[1] == 'new':
- #TODO: handle --verbose
- print "Not implemented."
-
- elif sys.argv[1] == 'count':
+ #-------------------------------------
+ elif sys.argv[1] == 'search':
db = Database()
if len(sys.argv) == 2:
#no further search term
else:
#mangle arguments wrapping terms with spaces in quotes
querystr = quote_query_line(sys.argv[2:])
- logging.debug("count "+querystr)
- print(len(Query(db,querystr).search_messages()))
-
+ logging.debug("search "+querystr)
+ t = Query(db,querystr).search_threads()
+ for thread in t:
+ print(str(thread))
+ #-------------------------------------
+ elif sys.argv[1] == 'show':
+ db = Database()
+ if len(sys.argv) == 2:
+ #no further search term
+ querystr=''
+ else:
+ #mangle arguments wrapping terms with spaces in quotes
+ querystr = quote_query_line(sys.argv[2:])
+ logging.debug("show "+querystr)
+ m = Query(db,querystr).search_messages()
+ for msg in m:
+ print(msg.format_as_text())
+ #-------------------------------------
+ elif sys.argv[1] == 'count':
+ if len(sys.argv) == 2:
+ #no further search term, count all
+ querystr=''
+ else:
+ #mangle arguments wrapping terms with spaces in quotes
+ querystr = quote_query_line(sys.argv[2:])
+ print(Database().create_query(querystr).count_messages())
+
+ #-------------------------------------
+ elif sys.argv[1] == 'tag':
+ #build lists of tags to be added and removed
+ add, remove = [], []
+ while not sys.argv[2]=='--' and \
+ (sys.argv[2].startswith('+') or sys.argv[2].startswith('-')):
+ if sys.argv[2].startswith('+'):
+ #append to add list without initial +
+ add.append(sys.argv.pop(2)[1:])
+ else:
+ #append to remove list without initial -
+ remove.append(sys.argv.pop(2)[1:])
+ #skip eventual '--'
+ if sys.argv[2]=='--': sys.argv.pop(2)
+ #the rest is search terms
+ querystr = quote_query_line(sys.argv[2:])
+ logging.debug("tag search-term "+querystr)
+ db = Database(mode=Database.MODE.READ_WRITE)
+ m = Query(db,querystr).search_messages()
+ for msg in m:
+ #actually add and remove all tags
+ map(msg.add_tag, add)
+ map(msg.remove_tag, remove)
+ #-------------------------------------
elif sys.argv[1] == 'search-tags':
if len(sys.argv) == 2:
#no further search term
db = Database()
m = Query(db,querystr).search_messages()
print("\n".join([t for t in m.collect_tags()]))
-
+ #-------------------------------------
elif sys.argv[1] == 'dump':
#TODO: implement "dump <filename>"
+ if len(sys.argv) == 2:
+ f = sys.stdout
+ else:
+ f = open(sys.argv[2],"w")
db = Database()
q = Query(db,'')
- q.set_sort(Query.SORT_MESSAGE_ID)
+ q.set_sort(Query.SORT.MESSAGE_ID)
m = q.search_messages()
for msg in m:
- print("%s (%s)" % (msg.get_message_id(), msg.get_tags()))
-
+ f.write("%s (%s)\n" % (msg.get_message_id(), msg.get_tags()))
+ #-------------------------------------
+ elif sys.argv[1] == 'restore':
+ import re
+ if len(sys.argv) == 2:
+ print("No filename given. Reading dump from stdin.")
+ f = sys.stdin
+ else:
+ f = open(sys.argv[2],"r")
+ #split the msg id and the tags
+ MSGID_TAGS = re.compile("(\S+)\s\((.*)\)$")
+ db = Database(mode=Database.MODE.READ_WRITE)
+
+ #read each line of the dump file
+ for line in f:
+ m = MSGID_TAGS.match(line)
+ if not m:
+ sys.stderr.write("Warning: Ignoring invalid input line: %s" %
+ line)
+ continue
+ # split line in components and fetch message
+ msg_id = m.group(1)
+ new_tags= set(m.group(2).split())
+ msg = db.find_message(msg_id)
+
+ if msg == None:
+ sys.stderr.write(
+ "Warning: Cannot apply tags to missing message: %s\n" % id)
+ continue
+
+ #do nothing if the old set of tags is the same as the new one
+ old_tags = set(msg.get_tags())
+ if old_tags == new_tags: continue
+
+ #set the new tags
+ msg.freeze()
+ #only remove tags if the new ones are not a superset anyway
+ if not (new_tags > old_tags): msg.remove_all_tags()
+ for tag in new_tags: msg.add_tag(tag)
+ msg.thaw()
+
+ #-------------------------------------
else:
# unknown command
print "Error: Unknown command '%s' (see \"notmuch help\")" % sys.argv[1]
#TODO: implement
"""
+setup
+new
search [options...] <search-terms> [...]
-
- Search for messages matching the given search terms.
-
show <search-terms> [...]
-
- Show all messages matching the search terms.
-
reply [options...] <search-terms> [...]
-
- Construct a reply template for a set of messages.
-
-tag +<tag>|-<tag> [...] [--] <search-terms> [...]
-
- Add/remove tags for all messages matching the search terms.
-
-dump [<filename>]
-
- Create a plain-text dump of the tags for each message.
-
restore <filename>
- search-tags [<search-terms> [...] ]
-
- List all tags found in the database or matching messages.
"""