bumblebee-status/modules/contrib/todo.py
2020-04-26 16:39:24 +02:00

44 lines
1.1 KiB
Python

# pylint: disable=C0111,R0903
"""Displays the number of todo items from a text file
Parameters:
* todo.file: File to read TODOs from (defaults to ~/Documents/todo.txt)
"""
import os.path
import core.module
import core.widget
import core.input
class Module(core.module.Module):
def __init__(self, config, theme):
super().__init__(config, theme, core.widget.Widget(self.output))
self.__doc = os.path.expanduser(self.parameter('file', '~/Documents/todo.txt'))
self.__todos = self.count_items()
core.input.register(self, button=core.input.LEFT_MOUSE, cmd='xdg-open {}'.format(self.__doc))
def output(self, widget):
return str(self.__todos)
def update(self):
self.__todos = self.count_items()
def state(self, widgets):
if self.__todos == 0:
return 'empty'
return 'items'
def count_items(self):
try:
i = -1
with open(self.__doc) as f:
for i, l in enumerate(f):
pass
return i+1
except Exception:
return 0
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4