57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
# pylint: disable=C0111,R0903
|
|
|
|
"""Displays the status of watson (time-tracking tool)
|
|
|
|
Requires the following executable:
|
|
* watson
|
|
"""
|
|
|
|
import bumblebee.input
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
import bumblebee.util
|
|
import bumblebee.popup_v2
|
|
|
|
import logging
|
|
import re
|
|
import functools
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
super(Module, self).__init__(engine, config,
|
|
bumblebee.output.Widget(full_text=self.text))
|
|
self._tracking = False
|
|
self._project = ''
|
|
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
|
|
cmd=self.toggle)
|
|
|
|
def toggle(self, widget):
|
|
self._project = 'hit'
|
|
if self._tracking:
|
|
bumblebee.util.execute('watson stop')
|
|
else:
|
|
bumblebee.util.execute('watson restart')
|
|
self._tracking = not self._tracking
|
|
|
|
def text(self, widget):
|
|
if self._tracking:
|
|
return self._project
|
|
else:
|
|
return 'Paused'
|
|
|
|
def update(self, widgets):
|
|
output = bumblebee.util.execute('watson status')
|
|
if re.match('No project started', output):
|
|
self._tracking = False
|
|
return
|
|
|
|
self._tracking = True
|
|
m = re.search(r'Project (.+) started', output)
|
|
self._project = m.group(1)
|
|
|
|
#
|
|
def state(self, widget):
|
|
return 'on' if self._tracking else 'off'
|
|
# return [widget.get('status', None), widget.get('period', None)]
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|