From b495c61ee47bf16ab1055541b3e69c4c1e55b129 Mon Sep 17 00:00:00 2001 From: Rob Golding Date: Wed, 10 Apr 2013 14:13:28 +0100 Subject: [PATCH] Improve get_tasks to allow arbitrary filtering --- tasklib/task.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/tasklib/task.py b/tasklib/task.py index f74e0d2..5e2b71b 100644 --- a/tasklib/task.py +++ b/tasklib/task.py @@ -47,6 +47,9 @@ class Task(object): class TaskWarrior(object): + DEFAULT_FILTERS = { + 'status': 'pending', + } def __init__(self, data_location='~/.task', create=True): if not os.path.exists(data_location): @@ -70,10 +73,19 @@ class TaskWarrior(object): raise TaskWarriorException(stderr.strip()) return stdout.strip().split('\n') - def get_tasks(self, project=None, status=PENDING): - command = 'export status:{0}'.format(status) - if project is not None: - command += ' project:{0}'.format(project) + def _format_filter_kwarg(self, kwarg): + key, val = kwarg[0], kwarg[1] + if key in ['tag', 'tags']: + key = 'tags.equal' + key = key.replace('__', '.') + return '{0}:{1}'.format(key, val) + + def get_tasks(self, **filter_kwargs): + filters = self.DEFAULT_FILTERS + filters.update(filter_kwargs) + filter_commands = ' '.join(map(self._format_filter_kwarg, + filters.items())) + command = '{0} export'.format(filter_commands) tasks = [] for line in self._execute(command): if line: -- 2.39.5