stat.py

#!/usr/bin/python3
from datetime import date, datetime, timedelta

import store
import worklog


def pretty_duration(minutes):
    hours, minutes = divmod(int(minutes), 60)
    return f"{hours}:{minutes:02}"


def load_journal():
    return worklog.parse("\n".join(store.read_all()))


def period_start(period):
    today = date.today()
    if period == "day":
        return today
    if period == "week":
        return today - timedelta(days=today.weekday())
    if period == "month":
        return today.replace(day=1)
    if period == "year":
        return today.replace(month=1, day=1)
    return datetime.strptime(period, "%Y-%m-%d").date()


def main(start=None, project=None):
    journal = load_journal()
    segments = [s for s in journal.segments if start is None or s.start.date() >= start]
    if project:
        segments = [s for s in segments if worklog.project(s.task) == project]

    by_task = {}
    for s in segments:
        by_task.setdefault(s.task, []).append(s)

    total = 0
    for task in sorted(by_task, key=lambda t: (worklog.project(t), min(s.start for s in by_task[t]))):
        task_segments = by_task[task]
        task_total = journal.total_minutes(task_segments)
        print(task, pretty_duration(task_total))
        by_day = {}
        for s in task_segments:
            by_day.setdefault(s.start.date(), []).append(s)
        for day in sorted(by_day):
            day_segments = by_day[day]
            print("\t", day, pretty_duration(journal.total_minutes(day_segments)))
            for s in sorted(day_segments, key=lambda s: s.start):
                duration = (s.end - s.start).total_seconds() // 60
                print("\t\t", s.start.strftime("%H:%M"), "+", pretty_duration(duration), s.comment)
        total += task_total
    print("Total time shown:", pretty_duration(total))


if __name__ == "__main__":
    from sys import argv
    if len(argv) == 1:
        main()
    elif argv[1] in ("-h", "--help"):
        print("USAGE: stat.py [period [project]]")
        print("  period may be: 'day', 'week', 'month', 'year' or a starting date like '2018-01-01'")
    else:
        main(period_start(argv[1]), argv[2] if len(argv) > 2 else None)