Source code for libqtile.widget.generic_poll_text

import json
from typing import Any, List, Tuple
from urllib.error import URLError
from urllib.request import Request, urlopen

from libqtile.log_utils import logger
from libqtile.widget import base

    import xmltodict

    def xmlparse(body):
        return xmltodict.parse(body)
except ImportError:
    # TODO: we could implement a similar parser by hand, but i'm lazy, so let's
    # punt for now
    def xmlparse(body):
        raise Exception("no xmltodict library")

[docs]class GenPollText(base.ThreadPoolText): """A generic text widget that polls using poll function to get the text""" orientations = base.ORIENTATION_HORIZONTAL defaults = [ ('func', None, 'Poll Function'), ] def __init__(self, **config): base.ThreadPoolText.__init__(self, "", **config) self.add_defaults(GenPollText.defaults) def poll(self): if not self.func: return "You need a poll function" return self.func()
[docs]class GenPollUrl(base.ThreadPoolText): """A generic text widget that polls an url and parses it using parse function""" orientations = base.ORIENTATION_HORIZONTAL defaults = [ ('url', None, 'Url'), ('data', None, 'Post Data'), ('parse', None, 'Parse Function'), ('json', True, 'Is Json?'), ('user_agent', 'Qtile', 'Set the user agent'), ('headers', {}, 'Extra Headers'), ('xml', False, 'Is XML?'), ] # type: List[Tuple[str, Any, str]] def __init__(self, **config): base.ThreadPoolText.__init__(self, "", **config) self.add_defaults(GenPollUrl.defaults) self.headers["User-agent"] = self.user_agent if self.json: self.headers['Content-Type'] = 'application/json' if and not isinstance(, str): = json.dumps( def fetch(self): req = Request(self.url,, self.headers) res = urlopen(req) charset = res.headers.get_content_charset() body = if charset: body = body.decode(charset) if self.json: body = json.loads(body) if self.xml: body = xmlparse(body) return body def poll(self): if not self.parse or not self.url: return "Invalid config" try: body = self.fetch() except URLError: return "No network" try: text = self.parse(body) except Exception: logger.exception('got exception polling widget') text = "Can't parse" return text