l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[modules/nvidiagpu] Automatically hide if inactive

Browse source 
If no nvidia card is found, hide the widget

see #603
This commit is contained in:
tobi-wan-kenobi 2020-04-26 16:09:45 +02:00
parent c3d2ef5f0b
commit 6f9abf616d
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
bumblebee/modules/nvidiagpu.py
View file

@ -13,15 +13,23 @@ import subprocess
import bumblebee.input import bumblebee.input
import bumblebee.output import bumblebee.output
import bumblebee.engine import bumblebee.engine
import bumblebee.util
class Module(bumblebee.engine.Module): class Module(bumblebee.engine.Module):
def __init__(self, engine, config): def __init__(self, engine, config):
super(Module, self).__init__(engine, config, bumblebee.output.Widget(full_text=self.utilization)) super(Module, self).__init__(engine, config, bumblebee.output.Widget(full_text=self.utilization))
self._utilization = "Not found: 0 0/0" self._utilization = "not found: 0°C 0/0 MiB"
def utilization(self, widget): def utilization(self, widget):
return self._utilization return self._utilization
def hidden(self):
hide = bumblebee.util.asbool(self.parameter("hide", False))
if hide and self._utilization == "not found: 0°C 0/0 MiB":
return True
return False
def update(self, widgets): def update(self, widgets):
sp = subprocess.Popen(['nvidia-smi', '-q'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) sp = subprocess.Popen(['nvidia-smi', '-q'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
out_str = sp.communicate() out_str = sp.communicate()