import subprocess
import json
def get_storage_devices(removable=None, device_type=None):
"""
:param removable: if True return only removable devices, if False only non-removable. If None return all devices
:param device_type: type of the devices to list
:return: list(tuple(str, str, str))
first tuple item is device name, second identifying string(VENDOR + MODEL), third - size
"""
proc = subprocess.Popen(
['lsblk', '-dJo', 'NAME,RM,TYPE,VENDOR,MODEL,SIZE'], stdout=subprocess.PIPE, stderr=subprocess.PIPE
)
if proc.wait() == 0:
result = proc.stdout.read()
result = json.loads(result.decode('utf'))
rval = []
for device in result['blockdevices']:
if removable and device['rm'] == '0':
continue
if removable == 'False' and device['rm'] == '1':
continue
if device_type is not None and device['type'] != device_type:
continue
rval.append((
device['name'],
'{} {} {}'.format(device['vendor'].strip(), device['model'].strip(), device['size']),
device['size']
))
return rval
else:
return []