bumblebee-status/bumblebee/modules/prime.py

70 lines
2.1 KiB
Python
Raw Normal View History

# pylint: disable=C0111,R0903
"""Displays and changes the current selected prime video card
Left click will call 'sudo prime-select nvidia'
Right click will call 'sudo prime-select nvidia'
Running these commands without a password requires editing your sudoers file
(always use visudo, it's very easy to make a mistake and get locked out of your computer!)
sudo visudo -f /etc/sudoers.d/prime
Then put a line like this in there:
user ALL=(ALL) NOPASSWD: /usr/bin/prime-select
If you can't figure out the sudoers thing, then don't worry, it's still really useful.
Parameters:
* prime.nvidiastring: String to use when nvidia is selected (defaults to "intel")
* prime.intelstring: String to use when intel is selected (defaults to "intel")
Requires the following executable:
* prime-select
"""
import bumblebee.util
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.query)
)
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
cmd=self._chooseNvidia)
engine.input.register_callback(self, button=bumblebee.input.RIGHT_MOUSE,
cmd=self._chooseIntel)
self.nvidiastring = self.parameter("nvidiastring", "nv")
self.intelstring = self.parameter("intelstring", "it")
def _chooseNvidia(self, event):
bumblebee.util.execute("sudo prime-select nvidia")
def _chooseIntel(self, event):
bumblebee.util.execute("sudo prime-select intel")
def _prev_keymap(self, event):
self._set_keymap(-1)
def query(self, widget):
try:
res = bumblebee.util.execute("prime-select query")
except RuntimeError:
return ["n/a"]
for line in res.split("\n"):
if not line: continue
if "nvidia" in line:
return self.nvidiastring
if "intel" in line:
return self.intelstring
return ["n/a"]
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4