bumblebee-status/modules/contrib/taskwarrior.py
2020-04-13 14:22:31 +02:00

36 lines
1 KiB
Python

"""Displays the number of pending tasks in TaskWarrior.
Requires the following library:
* taskw
Parameters:
* taskwarrior.taskrc : path to the taskrc file (defaults to ~/.taskrc)
"""
from taskw import TaskWarrior
import core.module
import core.widget
import core.decorators
class Module(core.module.Module):
def __init__(self, config):
super().__init__(config, core.widget.Widget(self.output))
self.__pending_tasks = '0'
def update(self):
"""Return a string with the number of pending tasks from TaskWarrior."""
try:
taskrc = self.parameter('taskrc', '~/.taskrc')
w = TaskWarrior(config_filename=taskrc)
pending_tasks = w.filter_tasks({'status': 'pending'})
self.__pending_tasks = str(len(pending_tasks))
except:
self.__pending_tasks = 'n/a'
def output(self, _):
"""Format the task counter to output in bumblebee."""
return '{}'.format(self.__pending_tasks)
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4