class Tracker(object):
# Class constants
CURRENT = path.expanduser("~/.tt_running")
def __init__(self):
self.task = Task()
self.fh = JSONHandler("test.json")
def track(self, name):
"""Start tracking a task.
:param name The name of the task to start.
:type name str
"""
self.task.set_name(name)
try:
with open(self.CURRENT) as f:
d = f.readline()
if d:
print "Task", d, "already running."
exit(0)
else:
f.close()
# Write task-name to ~/.tt_running
f = open(self.CURRENT, "w")
f.write(name)
f.close()
except IOError:
# Write task-name to ~/.tt_running
f = open(self.CURRENT, "w")
f.write(name)
f.close()
self.task.start_task()
data = self.task.to_json()
self.fh.write(data)
# TODO:
# Terminate task with given `task_name`
#def end(self, task_name):
def end(self):
# Read ~/.tt_running
data = None
try:
f = open(self.CURRENT)
data = f.readline()
f.close()
except IOError:
pass
if data:
self.task.set_from_dict(self.fh.load_obj(data))
self.task.end_task()
self.fh.write(self.task.to_json())
# TODO:
print self.task.show_stats()
self.reset()
else:
print "No task running."
def reset(self):
"""Reset the current running task."""
# Empty the self.CURRENT file
f = open(self.CURRENT, "w")
f.write("")
f.close()
