bumblebee-status/modules/contrib/rotation.py

# pylint: disable=C0111,R0903

"""Shows a widget for each connected screen and allows the user to loop through different orientations.

Requires the following executable:
    * xrandr
"""

import core.module
import core.widget
import core.input

import util.cli

possible_orientations = ['normal', 'left', 'inverted', 'right']

class Module(core.module.Module):
    def __init__(self, config):
        super().__init__(config, [])

    def update(self):
        widgets = self.widgets()
        for line in util.cli.execute('xrandr -q').split('\n'):
            if not ' connected' in line:
                continue
            display = line.split(' ', 2)[0]

            orientation = 'normal'
            for curr_orient in possible_orientations:
                if((line.split(' ')).count(curr_orient) > 1):
                    orientation = curr_orient
                    break

            widget = self.widget(display)
            if not widget:
                widget = core.widget.Widget(full_text=display, name=display)
                core.input.register(widget, button=core.input.LEFT_MOUSE, cmd=self.__toggle)
            widget.set('orientation', orientation)
            widgets.append(widget)

    def state(self, widget):
        return widget.get('orientation', 'normal')

    def __toggle(self, event):
        widget = self.widget_by_id(event['instance'])

        # compute new orientation based on current orientation
        idx = possible_orientations.index(widget.get('orientation'))
        idx = (idx + 1) % len(possible_orientations)
        new_orientation = possible_orientations[idx]

        widget.set('orientation', new_orientation)

        util.cli.execute('xrandr --output {} --rotation {}'.format(widget.name, new_orientation))

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4