From b8341c48e45975d8fc56bcb68e582f3ae1e87ead Mon Sep 17 00:00:00 2001 From: Anders Ingemann Date: Sun, 30 Jun 2013 23:54:28 +0200 Subject: [PATCH] allow removal of multiple tasks at once --- base/tasklist.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/base/tasklist.py b/base/tasklist.py index b16dffc..9b61c35 100644 --- a/base/tasklist.py +++ b/base/tasklist.py @@ -12,8 +12,9 @@ class TaskList(object): def add(self, *args): self.tasks.update(args) - def remove(self, task): - self.tasks.discard(self.get(task)) + def remove(self, *args): + for task in args: + self.tasks.discard(self.get(task)) def replace(self, task, replacement): self.remove(task)