From cb1e9aec548ec9f218907431d14837e73ab8102f Mon Sep 17 00:00:00 2001
From: Nico Schottelius <nico@brief.schottelius.org>
Date: Thu, 2 Aug 2012 00:28:16 +0200
Subject: [PATCH] can track time, storing seconds format

Signed-off-by: Nico Schottelius <nico@brief.schottelius.org>
---
 ctt.py | 92 +++++++++++++++++++++++++++++++++++++++++++---------------
 1 file changed, 69 insertions(+), 23 deletions(-)

diff --git a/ctt.py b/ctt.py
index 81af5ec..e2504b2 100755
--- a/ctt.py
+++ b/ctt.py
@@ -4,56 +4,102 @@ import argparse
 import calendar
 import datetime
 
-import signal
+#import signal
 
 import locale
+import logging
 import time
 
 import os
 import os.path
 import sys
 
-#def user_datetime(when):
-#    """Print time for the user"""
-#    return when.ctime()
+log = logging.getLogger(__name__)
+
+# Our output format
+def user_datetime(when):
+    """Print time for the user"""
+    return when.ctime()
+
+class CTT:
+    def __init__(self, project):
+        self.project = project
+        self.tracked_time = False
+
+        self._init_home()
+
+
+    def _init_home(self):
+        # Create output directory
+        home = os.environ['HOME']
+        self.ctt_dir = os.path.join(home, ".ctt")
+        self.project_dir = os.path.join(self.ctt_dir, self.project)
+
+
+    # Track time and return information from tracking
+    def track_time(self):
+        self.start = datetime.datetime.now()
+
+        try:
+            # Dummy read data to wait for keyboard irq
+            input()
+        except KeyboardInterrupt:
+            pass
+
+        self.stop = datetime.datetime.now()
+
+        self.tracked_time = True
+
+    def write_time(self):
+        if not self.tracked_time:
+            return
+
+        start_seconds =  self.start.strftime("%s")
+        stop_seconds =  self.stop.strftime("%s")
+        time_dir = os.path.join(self.project_dir, start_seconds)
+        os.makedirs(time_dir, mode=0o700, exist_ok=True)
+
+        filename = os.path.join(time_dir, "end")
+
+        with open(filename, "w") as fd:
+            fd.write("%s\n" % stop_seconds)
+
+    def duration(self):
+        if self.tracked_time:
+            delta = self.stop - self.start
+        else:
+            delta = 0
+
+        return delta
 
 
 # Setup locale for calendar printing
 # Setup locale to get Timezone information?
-print(locale.getlocale())
+#print(locale.getlocale())
 
 # Record project
 # Record tags
 
 parser = argparse.ArgumentParser()
-parser.add_argument("project_name", nargs=1)
+parser.add_argument("project", help="Project to track time for", nargs=1)
+parser.add_argument("-t", "--track", help="Track time (until Ctrl-C is pressed)",
+    action="store_true")
 
 args = parser.parse_args(sys.argv[1:])
+print(args)
 
+ctt = CTT(args.project[0])
+
+ctt.track_time()
+ctt.write_time()
+print(ctt.duration())
 
 sys.exit(0)
 
 # Setup signal handler
 
 # Start tracking
-start = datetime.datetime.now()
-print("Starting at %s" % (start.ctime()))
-
-time.sleep(2)
-
-# Stop tracking
-stop = datetime.datetime.now()
-print("Stopped at %s" % (stop.ctime()))
-
-delta = stop - start
-print("%s, %s, seconds=%s, %s" % (type(delta), delta, delta.total_seconds(), start.strftime("%s")))
-
 # Save stuff to our home directory
 
-# Prepare home directory
-home = os.environ['HOME']
-ctt_home = home + os.path.join(home, ".ctt")
-os.makedirs(ctt_home, mode=0o700, exist_ok=True)
-
 # Create datetime from userinput
 # Wed Aug  1 23:35:53 2012