From 97866362d54d551f4d4ee2ea8fbf016c7c46aece Mon Sep 17 00:00:00 2001 From: tobi-wan-kenobi Date: Fri, 24 Apr 2020 16:09:12 +0200 Subject: [PATCH] [modules/nvidiagpu] quotes --- modules/contrib/nvidiagpu.py | 54 ++++++++++++++++++------------------ 1 file changed, 27 insertions(+), 27 deletions(-) diff --git a/modules/contrib/nvidiagpu.py b/modules/contrib/nvidiagpu.py index 2ce7d38..0041e28 100644 --- a/modules/contrib/nvidiagpu.py +++ b/modules/contrib/nvidiagpu.py @@ -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,