localwaka/goalctl

#!/usr/bin/env python3
# kinds of goals:
# - overall
#     "i want to code at least {time} per {interval}"
# - in a project
# - in a language
# - in an editor

from datetime import timedelta, date, datetime, time
import sys
import json
import sqlite3
from config import timeout
import utils

conn = sqlite3.connect("data.db")

if len(sys.argv) < 2:
    print("provide a file", file=sys.stderr)
    sys.exit(1)

filename = sys.argv[1]
with open(filename) as f:
    data = json.load(f)

goal_type = data["type"]
if goal_type == "overall":
    # in seconds
    goal_min = int(data["goal_min"])

    # figure out what timestamps to pull from
    if data["interval"] == "day":
        # except!
        def convert_weekday(s):
            s = s.lower()
            return dict(
                sunday=0,
                monday=1,
                tuesday=2,
                wednesday=3,
                thursday=4,
                friday=5,
                saturday=6,
            ).get(s)
        excepts = list(map(convert_weekday, data["except"]))

        if len(sys.argv) > 2:
            # if a day is given, print that day
            d = datetime.strptime(sys.argv[2], "%Y-%m-%d").date()
            start_time = datetime.combine(d, time(hour=0, minute=0, second=0))
            end_time = datetime.combine(d, time(hour=23, minute=59, second=59))
        else:
            # find the current day!
            today = date.today()
            start_time = datetime.combine(today, time(hour=0, minute=0, second=0))
            end_time = datetime.now()

        coded_secs = utils.get_seconds_coded(conn, start_time, end_time, timeout)
        print("minutes coded:", (coded_secs/60))

        # get all the entries
        c = conn.cursor()
        curs = c.execute("""
            select time from heartbeats where time > ? and time < ? order by time;
        """, [start_stamp, end_stamp])

        coded_secs = 0
        for (first, second) in window(curs, 2):
            diff = second[0] - first[0]
            if diff < timeout:
                coded_secs += diff
        print("coded: {}:{:02d} min".format(int(coded_secs)//60, int(coded_secs)%60))

    elif data["interval"] == "week":
        # start of the week
        today = date.today()
        last_monday = today - timedelta(days=today.weekday())
        print(last_monday)

# vim: set ft=python: