Permalink
Find file
Fetching contributors…
Cannot retrieve contributors at this time
114 lines (76 sloc) 3.55 KB
import pygtk
import gtk
from harvest import Harvest
# temporary import till we figure out config loading
import scratch
class HarvestNotifier:
timer_running = False
current_project = None
current_task = None
def __init__(self):
self.harvest = Harvest(scratch.url, scratch.username, scratch.password)
self.statusIcon = gtk.StatusIcon()
self.statusIcon.set_from_stock(gtk.STOCK_ABOUT)
self.statusIcon.set_visible(True)
self.statusIcon.set_tooltip("Hello World")
self.menu = gtk.Menu()
sep = gtk.SeparatorMenuItem()
self.menu.append(sep)
project_menu = gtk.Menu()
for client, projects in self.harvest.get_project_in_categories().iteritems():
client_item = gtk.MenuItem(client)
project_menu.append(client_item)
for project in projects:
task_menu = gtk.Menu()
p = gtk.MenuItem(project['name'])
for task in project['tasks']:
t = gtk.MenuItem(task['name'])
t.connect('activate', self.task_cb, self.menu, {"project": project, "task": task})
task_menu.append(t)
p.set_submenu(task_menu)
project_menu.append(p)
sep = gtk.SeparatorMenuItem()
project_menu.append(sep)
self.menuItem = gtk.MenuItem('Projects')
self.menuItem.set_submenu(project_menu)
self.menu.append(self.menuItem)
self.menuItem = gtk.ImageMenuItem(gtk.STOCK_QUIT)
self.menuItem.connect('activate', self.quit_cb, self.statusIcon)
self.menu.append(self.menuItem)
self.statusIcon.connect('popup-menu', self.popup_menu_cb, self.menu)
self.statusIcon.set_visible(1)
gtk.main()
def task_cb(self, widget, event, data = None):
if self.timer_running:
self.menu.remove(self.menu.get_children()[0])
self.harvest.timer_toggle(data['project'], data['task'])
# button showing what timer we're running
timer = gtk.MenuItem(data['project']['name'] + " - " + data['task']['name'])
timer.connect('activate', self.timer_click_cb, self.menu, data)
self.menu.prepend(timer)
self.timer_running = True
self.current_project = data['project']
self.current_task = data['task']
def timer_click_cb(self, widget, event, data = None):
""" Event triggered if a timer is running and someone clicks on the timer """
if not self.timer_running:
# we shouldn't get here, but just in case
return
self.harvest.timer_toggle(data['project'], data['task'])
# remove the timer button
self.menu.remove(widget)
self.timer_running = False
self.current_project = None
self.current_task = None
def quit_cb(self, widget, data = None):
if self.timer_running:
self.harvest.timer_toggle(self.current_project, self.current_task)
gtk.main_quit()
def popup_menu_cb(self, widget, button, time, data = None):
if button == 3:
if data:
data.show_all()
data.popup(None, None, gtk.status_icon_position_menu,
3, time, self.statusIcon)
if __name__ == "__main__":
notifier = HarvestNotifier()