tblancher's timetracker timekeeping system
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
timetracker/timetracker.py

105 lines
2.9 KiB

#!/usr/bin/python
import fileinput
import math
import re
import sys
import time
# Variables
start = {} # Data structure for storing the beginning of a time frame
finish = {} # Data structure for storing the end of a time frame
ind = {} # Data structure for storing individual timecards
misc = {} # Data structure for storing miscellaneous timecards
empty = re.compile("^\s*$") # regex for finding empty lines (and skipping them)
comment = re.compile("^\s*#") # regex for finding empty lines (and skipping them)
delim = re.compile(":\s{2}") # regex for splitting timestamp entry
begin = re.compile("^Begin\s+(?P<action>.*)\s*$")
end = re.compile("^End\s+(?P<action>.*)\s*$")
incident = re.compile("^(PDROP|CHG)-?\d+")
for line in fileinput.input():
if empty.match(line) or comment.match(line): # if the current line is empty
continue # skip to the next line
(datetime,entry) = delim.split(line.strip())
timestamp = time.mktime(time.strptime(datetime,"%Y-%m-%d %H:%M:%S"))
if begin.match(entry):
action = begin.sub("\g<action>",entry)
if action not in start:
start[action] = []
start[action].append(timestamp)
if end.match(entry):
action = end.sub("\g<action>",entry)
if action not in finish:
finish[action] = []
finish[action].append(timestamp)
# We're here, we've captured all of the data
gtot = 0
for act in sorted(start.keys()):
if act in start:
bc = len(start[act])
else: bc = 0
if act in finish:
ec = len(finish[act])
else: ec = 0
if bc - ec > 1:
print >> sys.stderr, "ERROR: Missing more than one End"
sys.exit(2)
elif bc > ec: # bc should be exactly one greater
start[act].pop()
elif ec > bc:
print >> sys.stderr, "ERROR: Missing a Begin"
sys.exit(1)
sigma = 0
while (len(start[act]) > 0 and len(start[act]) > 0):
beg = start[act].pop(0)
en = finish[act].pop(0)
sigma += en - beg
ind[act] = sigma
gtot += sigma
ttot = 0
utot = 0
table = "{a:<50} {s:>10} {h:>10} {m:>10} {f:>10}"
for act in sorted(ind.keys()):
minutes = "{:d}min".format(int(ind[act]%3600/60))
hrs = "{:d}hrs".format(int(ind[act]/3600))
fhrs = "{:.2f}hrs".format(ind[act]/3600)
sec = "{:d}s".format(int(ind[act]))
print table.format(a=act, s=sec, h=hrs, m=minutes, f=fhrs)
if incident.match(act):
ttot += ind[act]
else:
utot += ind[act]
total = "{t:<20} {s:>10} {h:>10}"
ttots = "{:d}s".format(int(ttot))
utots = "{:d}s".format(int(utot))
ttoth = "{:3.2f}hrs".format(ttot / 3600)
utoth = "{:3.2f}hrs".format(utot / 3600)
gtots = "{:d}s".format(int(gtot))
gtoth = "{:3.2f}hrs".format(gtot / 3600)
print
print total.format(t="Incident total:", s=ttots, h=ttoth)
print total.format(t="Unnumbered total:", s=utots, h=utoth)
print total.format(t="Section total:", s=gtots, h=gtoth)