lhark/rc
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

[work] small fixes

Browse source
This commit is contained in:
Lara 2020-02-28 17:37:00 -05:00
parent d9b0a0fbab
commit 6ced297495
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
bin/work
View file

@ -52,7 +52,11 @@ def td_format(td):
def get_today_fields(): def get_today_fields():
try: try:
with open(TODAY_FILE, "r") as f: with open(TODAY_FILE, "r") as f:
line_count = 0
for line in f: for line in f:
line_count += 1
if line_count > 1:
log("Warning: incomplete day leftover in {}".format(TODAY_FILE))
if line.startswith(start_time.strftime("%Y-%m-%d")): if line.startswith(start_time.strftime("%Y-%m-%d")):
return line.strip().split(",") return line.strip().split(",")
except FileNotFoundError: except FileNotFoundError:
@ -131,7 +135,7 @@ def work_start(args):
hour = start_time.strftime("%H:%M") hour = start_time.strftime("%H:%M")
if fields: if fields:
die("You already started working") die("You already started working")
with open(TODAY_FILE, "w") as f: with open(TODAY_FILE, "a") as f:
f.write("{},{}".format( f.write("{},{}".format(
start_time.strftime("%Y-%m-%d"), hour)) start_time.strftime("%Y-%m-%d"), hour))
log("Started working at {}".format(hour)) log("Started working at {}".format(hour))
@ -205,7 +209,8 @@ def work_parse(args):
for line in f: for line in f:
fields = line.split(",") fields = line.split(",")
if len(fields) < 6: if len(fields) < 6:
break log("Record: '{}' hasn't got enough fields ({})".format(line, len(fields)))
continue
morning = datetime.strptime(fields[1], "%H:%M") morning = datetime.strptime(fields[1], "%H:%M")
break_start = datetime.strptime(fields[2], "%H:%M") break_start = datetime.strptime(fields[2], "%H:%M")
break_end = datetime.strptime(fields[3], "%H:%M") break_end = datetime.strptime(fields[3], "%H:%M")