can track time, storing seconds format

Signed-off-by: Nico Schottelius <nico@brief.schottelius.org>
This commit is contained in:
Nico Schottelius 2012-08-02 00:28:16 +02:00
parent fc194c9f7a
commit cb1e9aec54

92
ctt.py
View file

@ -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