From fbd4bd4bdaae915c469ade5dad3123f4f852a3de Mon Sep 17 00:00:00 2001 From: Bill Peck Date: Thu, 7 May 2015 11:00:11 -0400 Subject: small help changes in commands Update_task will now remove it from the DB if its gone on disk. --- git_taskrepo/taskrepo.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'git_taskrepo/taskrepo.py') diff --git a/git_taskrepo/taskrepo.py b/git_taskrepo/taskrepo.py index b53d782..026c1c4 100644 --- a/git_taskrepo/taskrepo.py +++ b/git_taskrepo/taskrepo.py @@ -24,6 +24,18 @@ class TRX_TestInfo(TRX): def only_ascii(s): return filter(lambda x: x in string.printable, s) +def _delete_taskrepo(taskrepo, taskname): + with taskrepo: + cur = taskrepo.cursor() + cur.execute("SELECT id FROM tasks WHERE name=?", (taskname,)) + result = cur.fetchone() + if result: + taskid = result[0] + for key in keys: + # Clear old values + cur.execute("DELETE FROM %s WHERE task_id=?" % key, (taskid,)) + cur.execute("DELETE FROM tasks WHERE id=?", (taskid,)) + def _update_taskrepo(taskrepo, taskname, testinfo): with taskrepo: cur = taskrepo.cursor() @@ -60,6 +72,7 @@ def update_taskrepo(repo, taskrepo, taskpath): raise _update_taskrepo(taskrepo, taskname, testinfo) else: + _delete_taskrepo(taskrepo, taskname) raise TRX_TestInfo('No testinfo.desc') def parse_testinfo(filename): -- cgit