70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
|
# pylint: disable=C0111,R0903
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""Shows a widget for each connected screen and allows the user to loop through different orientations.
|
||
|
|
|
||
|
|
Requires the following executable:
|
||
|
|
* swaymsg
|
||
|
|
"""
|
||
|
|
|
||
|
|
import bumblebee.util
|
||
|
|
import bumblebee.input
|
||
|
|
import bumblebee.output
|
||
|
|
import bumblebee.engine
|
||
|
|
import json
|
||
|
|
from os import environ, path
|
||
|
|
|
||
|
|
possible_orientations = ["normal", "90", "180", "270"]
|
||
|
|
|
||
|
|
class Module(bumblebee.engine.Module):
|
||
|
|
def __init__(self, engine, config):
|
||
|
|
self.service = "sway-auto-rotate@%s" % path.basename(environ['SWAYSOCK']).replace('-', '\\\\x2d')
|
||
|
|
widgets = []
|
||
|
|
self._engine = engine
|
||
|
|
super(Module, self).__init__(engine, config, widgets)
|
||
|
|
self.update_widgets(widgets)
|
||
|
|
|
||
|
|
def update_widgets(self, widgets):
|
||
|
|
for display in json.loads(bumblebee.util.execute("/usr/bin/swaymsg -t get_outputs -r")):
|
||
|
|
name = display['name']
|
||
|
|
widget = self.widget(name)
|
||
|
|
if not widget:
|
||
|
|
widget = bumblebee.output.Widget(name=name, full_text="若 "+name)
|
||
|
|
widget.set("auto", bumblebee.util.execute("systemctl --user show %s -p ActiveState" % self.service).split("\n")[0].split("=")[1])
|
||
|
|
self._engine.input.register_callback(widget, button=bumblebee.input.LEFT_MOUSE, cmd=self._toggle)
|
||
|
|
self._engine.input.register_callback(widget, button=bumblebee.input.RIGHT_MOUSE, cmd=self._toggle_auto)
|
||
|
|
widget.set("orientation", display['transform'])
|
||
|
|
widgets.append(widget)
|
||
|
|
|
||
|
|
def update(self, widgets):
|
||
|
|
if len(widgets) <= 0:
|
||
|
|
self.update_widgets(widgets)
|
||
|
|
|
||
|
|
def state(self, widget):
|
||
|
|
curr = widget.get("auto", "inactive")
|
||
|
|
if curr == "inactive":
|
||
|
|
return ["warning"]
|
||
|
|
return [curr]
|
||
|
|
|
||
|
|
def _toggle_auto(self, event):
|
||
|
|
widget = self.widget_by_id(event["instance"])
|
||
|
|
if widget.get("auto") == "inactive":
|
||
|
|
bumblebee.util.execute("systemctl --user start %s" % self.service)
|
||
|
|
else:
|
||
|
|
bumblebee.util.execute("systemctl --user stop %s" % self.service)
|
||
|
|
widget.set("auto", bumblebee.util.execute("systemctl --user show %s -p ActiveState" % self.service).split("\n")[0].split("=")[1])
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
bumblebee.util.execute("swaymsg 'output {} transform {}'".format(widget.name, new_orientation))
|
||
|
|
|
||
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|