hamster_tracks.py

#!/usr/bin/env python
# - coding: utf-8 -
# Copyright (C) 2010 Toms Bauģis <toms.baugis at gmail.com>

"""An attempt to make an overview visualization. Consumes hamster d-bus API"""


from gi.repository import Gtk as gtk
from lib import graphics

import time, datetime as dt
from collections import defaultdict
import hamster.client


class Scene(graphics.Scene):
    def __init__(self):
        graphics.Scene.__init__(self)

        storage = hamster.client.Storage()

        self.facts = storage.get_facts(dt.date(2009,1,1), dt.date.today())

        self.day_counts = defaultdict(list)
        activities, categories = defaultdict(int), defaultdict(int)

        print len(self.facts)

        for fact in self.facts:
            self.day_counts[fact.start_time.date()].append(fact)
            activities[fact.activity] += 1
            categories[fact.category] += 1

            if fact.end_time and fact.start_time.date() != fact.end_time.date():
                self.day_counts[fact.end_time.date()].append(fact)



        self.activities = [activity[0] for activity in sorted(activities.items(), key=lambda item:item[1], reverse=True)]
        self.categories = categories.keys()

        self.connect("on-enter-frame", self.on_enter_frame)



    def on_enter_frame(self, scene, context):
        if not self.facts:
            return

        g = graphics.Graphics(context)
        g.set_line_style(width=1)

        start_date = self.facts[0].start_time.date()
        end_date = (self.facts[-1].start_time + self.facts[-1].delta).date()

        days = (end_date - start_date).days





        full_days = []
        for day in range(days):
            current_date = start_date + dt.timedelta(days=day)
            if not self.day_counts[current_date]:
                continue
            full_days.append(self.day_counts[current_date])

        day_pixel = float(self.width) / len(full_days)



        cur_x = 0
        pixel_width = max(round(day_pixel), 1)

        for day in full_days:
            cur_x += round(day_pixel)

            for j, fact in enumerate(day):

                #bar per category
                g.rectangle(cur_x, 27 + self.categories.index(fact.category) * 6, pixel_width, 6)

                #bar per activity
                g.rectangle(cur_x, 102 + self.activities.index(fact.activity) * 6, pixel_width, 6)

                #number of activities
                g.rectangle(cur_x, self.height - 3 * j, pixel_width, 3)

            g.fill("#aaa")




class BasicWindow:
    def __init__(self):
        window = gtk.Window()
        window.set_size_request(600, 300)
        window.connect("delete_event", lambda *args: gtk.main_quit())
        window.add(Scene())
        window.show_all()


example = BasicWindow()
import signal
signal.signal(signal.SIGINT, signal.SIG_DFL) # gtk3 screws up ctrl+c
gtk.main()