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

[modules/smartstatus] quotes

Browse source
This commit is contained in:
tobi-wan-kenobi 2020-04-26 10:02:05 +02:00
parent 965d2844b0
commit 8e29bf8f9e
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
modules/contrib/smartstatus.py
View file

@ -6,8 +6,8 @@
"""Displays HDD smart status of different drives or all drives """Displays HDD smart status of different drives or all drives
Parameters: Parameters:
* smartstatus.display: how to display (defaults to "combined", other choices: "seperate" or "singles") * smartstatus.display: how to display (defaults to 'combined', other choices: 'seperate' or 'singles')
* smartstauts.drives: in the case of singles which drives to display, separated comma list value, multiple accepted (defaults to "sda", example:"sda,sdc") * smartstauts.drives: in the case of singles which drives to display, separated comma list value, multiple accepted (defaults to 'sda', example:'sda,sdc')
""" """
import os import os
@ -38,7 +38,7 @@ class Module(bumblebee.engine.Module):
widgets.append(widget) widgets.append(widget)
else: else:
for device in self.devices: for device in self.devices:
if self.display == "singles" and device not in self.drives: if self.display == 'singles' and device not in self.drives:
continue continue
widget = bumblebee.output.Widget() widget = bumblebee.output.Widget()
widget.set('device', device) widget.set('device', device)
@ -60,7 +60,7 @@ class Module(bumblebee.engine.Module):
def output(self, widget): def output(self, widget):
device = widget.get('device') device = widget.get('device')
assessment = widget.get('assessment') assessment = widget.get('assessment')
widget.full_text("{}: {}".format(device, assessment)) widget.full_text('{}: {}'.format(device, assessment))
def state(self, widget): def state(self, widget):
states = [] states = []
@ -83,7 +83,7 @@ class Module(bumblebee.engine.Module):
def list_devices(self): def list_devices(self):
for (root, folders, files) in os.walk('/dev'): for (root, folders, files) in os.walk('/dev'):
if root == '/dev': if root == '/dev':
devices = {"".join(filter(lambda i: i.isdigit() == False, file)) for file in files if 'sd' in file} devices = {''.join(filter(lambda i: i.isdigit() == False, file)) for file in files if 'sd' in file}
nvme = {file for file in files if('nvme0n' in file and 'p' not in file)} nvme = {file for file in files if('nvme0n' in file and 'p' not in file)}
devices.update(nvme) devices.update(nvme)
return devices return devices
@ -114,6 +114,6 @@ class Module(bumblebee.engine.Module):
_stdout, _stderr = [i.decode('utf8') for i in cmd.communicate()] _stdout, _stderr = [i.decode('utf8') for i in cmd.communicate()]
_stdout = _stdout.split('\n') _stdout = _stdout.split('\n')
for line in _stdout: for line in _stdout:
if "Pre-fail" in line: if 'Pre-fail' in line:
assessment = "Pre-fail" assessment = 'Pre-fail'
return assessment return assessment