bumblebee-status/modules/contrib/watson.py

# 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