From 4edabbdf3bd4c0f3d9580e047127545ea724daa3 Mon Sep 17 00:00:00 2001 From: Josh Holtrop Date: Sun, 2 Jan 2011 22:10:58 -0500 Subject: [PATCH] processing start, out, and task commands --- dwtt | 25 ++++++++++++++++++++----- 1 file changed, 20 insertions(+), 5 deletions(-) diff --git a/dwtt b/dwtt index 503f9c6..9ff0614 100755 --- a/dwtt +++ b/dwtt @@ -6,7 +6,7 @@ import getopt from CmdWindow import CmdWindow from Command import Command -from DataStore import DataStore +from DataStore import DataStore, TaskRef PROGRAM_NAME = 'dwtt' @@ -48,10 +48,26 @@ def doCmdWindow(): return c.main() def processStart(cmd, store): - pass + processOut(cmd, store) + task = store.getTaskByName(cmd.argstr) + if task is None: + parent = store.getParentTaskByName(cmd.argstr) + if parent is None: + return 'Could not find task "%s"' % \ + (':'.join(cmd.argstr.split(':')[:-1])) + taskid = store.createTask(cmd.argstr.split(':')[-1], '', parent.taskid) + else: + taskid = task.taskid + store.updateCurrentTask(TaskRef(taskid, cmd.time)) def processOut(cmd, store): - pass + ct = store.getCurrentTask() + if ct is not None: + seconds = (cmd.time - ct.time).seconds + store.addTime(cmd.time.strftime('%Y-%m-%d'), ct.taskid, seconds) + store.clearCurrentTask() + return False + return 'No current task defined' def processTask(cmd, store): parts = cmd.argstr.split(',', 1) @@ -63,8 +79,7 @@ def processTask(cmd, store): # the task already exists, update it store.updateTask(task.taskid, nameparts[-1], longname) return False - parenttaskname = ':'.join(nameparts[:-1]) - parenttask = store.getTaskByName(parenttaskname) + parenttask = store.getParentTaskByName(fullname) if parenttask is None: return 'Parent task not found' store.createTask(nameparts[-1].strip(), longname, parenttask.taskid)