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

nvidiagpu: add option to format output (default=current format)

Browse source
This commit is contained in:
0xDEAD 2019-04-15 12:30:20 +02:00
parent 62d34395c7
commit eadc2b45c3
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
bumblebee/modules/nvidiagpu.py
View file

@ -2,6 +2,10 @@
"""Displays GPU name, temperature and memory usage. """Displays GPU name, temperature and memory usage.
Parameters:
* nvidiagpu.format: Format string (defaults to "{name}: {temp}°C %{usedmem}/{totalmem} MiB")
Available values are: {name} {temp} {mem_used} {mem_total}
Requires nvidia-smi Requires nvidia-smi
""" """
@ -34,13 +38,21 @@ class Module(bumblebee.engine.Module):
key, val = key.strip(), val.strip() key, val = key.strip(), val.strip()
if title == "FB Memory Usage": if title == "FB Memory Usage":
if key == "Total": if key == "Total":
totalMem = val totalMem = val.split(" ")[0]
elif key == "Used": elif key == "Used":
usedMem = val.split(" ")[0] usedMem = val.split(" ")[0]
elif key == "GPU Current Temp": elif key == "GPU Current Temp":
temp = val.split(" ")[0] temp = val.split(" ")[0]
elif key == "Product Name": elif key == "Product Name":
name = val name = val
except: except:
title = item.strip() title = item.strip()
self._utilization = u"%s: %s°C %s/%s"%(name, temp, usedMem, totalMem)
str_format = self.parameter("format", '{name}: {temp}°C {mem_used}/{mem_total} MiB')
self._utilization = str_format.format(
name = name,
temp = temp,
mem_used = usedMem,
mem_total = totalMem,
)