Don't add WK or loopback devices

This commit is contained in:
2Shirt 2018-12-04 16:18:45 -07:00
parent 70a742e69c
commit 6014a8fb70
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C

View file

@ -171,7 +171,22 @@ class State():
result = run_program(cmd, check=False)
json_data = json.loads(result.stdout.decode())
for dev in json_data['blockdevices']:
self.devs.append(DevObj(dev['name']))
skip_dev = False
dev_obj = DevObj(dev['name'])
# Skip loopback devices
if dev_obj.lsblk['tran'] == 'NONE':
skip_dev = True
# Skip WK devices
wk_label_regex = r'{}_(LINUX|UFD)'.format(KIT_NAME_SHORT)
for label in dev_obj.labels:
if re.search(wk_label_regex, label, re.IGNORECASE):
skip_dev = True
# Add device
if not skip_dev:
self.devs.append(DevObj(dev['name']))
# Functions
def generate_horizontal_graph(rates, oneline=False):