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' or device['rm'] is False):
                continue
            if removable == 'False' and (device['rm'] == '1' or device['rm'] is True):
                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 []