From 9b32a26c4818248911b1fdbec1baf0efdcf30bc9 Mon Sep 17 00:00:00 2001 From: Emma Tinten Date: Mon, 30 Dec 2019 15:02:27 +0100 Subject: [PATCH] Added simple yubikey state module --- bumblebee/modules/yubikey.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 bumblebee/modules/yubikey.py diff --git a/bumblebee/modules/yubikey.py b/bumblebee/modules/yubikey.py new file mode 100644 index 0000000..db32843 --- /dev/null +++ b/bumblebee/modules/yubikey.py @@ -0,0 +1,33 @@ +# pylint: disable=C0111,R0903 + +"""Shows yubikey information + +Requires: https://github.com/Yubico/python-yubico + +The output indicates that a YubiKey is not connected or it displays +the corresponding serial number. + +""" + +import yubico + +import bumblebee.input +import bumblebee.output +import bumblebee.engine + +class Module(bumblebee.engine.Module): + def __init__(self, engine, config): + super(Module, self).__init__(engine, config, + bumblebee.output.Widget(full_text=self.keystate)) + self._keystate = "No YubiKey" + + def keystate(self, widget): + return self._keystate + + def update(self, widget): + try: + self._keystate = "YubiKey: " + str(yubico.find_yubikey(debug=False).serial()) + except yubico.yubico_exception.YubicoError: + self._keystate = "No YubiKey" + +# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4