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

[modules/nvidiagpu] quotes

Browse source
This commit is contained in:
tobi-wan-kenobi 2020-04-24 16:09:12 +02:00
parent 53afade88d
commit 97866362d5
1 changed files with 27 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
modules/contrib/nvidiagpu.py
View file

@ -3,7 +3,7 @@
"""Displays GPU name, temperature and memory usage.
Parameters:
* nvidiagpu.format: Format string (defaults to "{name}: {temp}°C %{usedmem}/{totalmem} MiB")
* nvidiagpu.format: Format string (defaults to '{name}: {temp}°C %{usedmem}/{totalmem} MiB')
Available values are: {name} {temp} {mem_used} {mem_total} {fanspeed} {clock_gpu} {clock_mem}
Requires nvidia-smi
@ -17,7 +17,7 @@ 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.utilization))
self._utilization = "Not found: 0 0/0"
self._utilization = 'Not found: 0 0/0'
def utilization(self, widget):
return self._utilization
@ -25,41 +25,41 @@ class Module(bumblebee.engine.Module):
def update(self, widgets):
sp = subprocess.Popen(['nvidia-smi', '-q'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
out_str = sp.communicate()
out_list = out_str[0].decode("utf-8").split('\n')
out_list = out_str[0].decode('utf-8').split('\n')
title = ""
usedMem = ""
totalMem = ""
temp = ""
name = "not found"
clockMem = ""
clockGpu = ""
fanspeed = ""
title = ''
usedMem = ''
totalMem = ''
temp = ''
name = 'not found'
clockMem = ''
clockGpu = ''
fanspeed = ''
for item in out_list:
try:
key, val = item.split(':')
key, val = key.strip(), val.strip()
if title == "Clocks":
if key == "Graphics":
clockGpu = val.split(" ")[0]
elif key == "Memory":
clockMem = val.split(" ")[0]
if title == "FB Memory Usage":
if key == "Total":
totalMem = val.split(" ")[0]
elif key == "Used":
usedMem = val.split(" ")[0]
elif key == "GPU Current Temp":
temp = val.split(" ")[0]
elif key == "Product Name":
if title == 'Clocks':
if key == 'Graphics':
clockGpu = val.split(' ')[0]
elif key == 'Memory':
clockMem = val.split(' ')[0]
if title == 'FB Memory Usage':
if key == 'Total':
totalMem = val.split(' ')[0]
elif key == 'Used':
usedMem = val.split(' ')[0]
elif key == 'GPU Current Temp':
temp = val.split(' ')[0]
elif key == 'Product Name':
name = val
elif key == "Fan Speed":
fanspeed = val.split(" ")[0]
elif key == 'Fan Speed':
fanspeed = val.split(' ')[0]
except:
title = item.strip()
str_format = self.parameter("format", '{name}: {temp}°C {mem_used}/{mem_total} MiB')
str_format = self.parameter('format', '{name}: {temp}°C {mem_used}/{mem_total} MiB')
self._utilization = str_format.format(
name = name,
temp = temp,