diff options
Diffstat (limited to 'alot/commands/globals.py')
-rw-r--r-- | alot/commands/globals.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/alot/commands/globals.py b/alot/commands/globals.py index af98694d..92918269 100644 --- a/alot/commands/globals.py +++ b/alot/commands/globals.py @@ -994,3 +994,55 @@ class ReloadCommand(Command): except ConfigError as e: ui.notify('Error when reloading config files:\n {}'.format(e), priority='error') + + +@registerCommand( + MODE, 'savequery', + arguments=[ + (['--no-flush'], {'action': 'store_false', 'dest': 'flush', + 'default': 'True', + 'help': 'postpone a writeout to the index'}), + (['alias'], {'help': 'alias to use for query string'}), + (['query'], {'help': 'query string to store', + 'nargs': argparse.REMAINDER}) + ], + help='store query string as a "named query" in the database') +class SaveQueryCommand(Command): + + """save alias for query string""" + repeatable = False + + def __init__(self, alias, query=None, flush=True, **kwargs): + """ + :param alias: name to use for query string + :type alias: str + :param query: query string to save + :type query: str or None + :param flush: immediately write out to the index + :type flush: bool + """ + self.alias = alias + if query is None: + self.query = '' + else: + self.query = ' '.join(query) + self.flush = flush + Command.__init__(self, **kwargs) + + def apply(self, ui): + searchbuffer = ui.current_buffer + query = searchbuffer.querystring + msg = 'saved alias "%s" for query string "%s"' % (self.alias, + self.query) + + try: + ui.dbman.save_named_query(self.alias, self.query) + logging.debug(msg) + ui.notify(msg) + except DatabaseROError: + ui.notify('index in read-only mode', priority='error') + return + + # flush index + if self.flush: + ui.apply_command(commands.globals.FlushCommand()) |