Return the device ID's for all drives in a zpool.

get_zpool_drives.py

from execute import execute, Retcode
import re

def get_zpool_drives(zpool):
    """
    Returns all drives in the zpool.

    Inputs:
        zpool (str): Zpool name
    Outputs:
        drives (list): A list of zpool drives
    """
    drives = []
    
    try:
        retcode, output = execute('zpool status %s' % zpool)
    except:
        raise
    else:
        if retcode:
            raise Retcode(output)
    
    for line in output.split("\n"):
        match = re.search(r'(c[0-9]t.*d[0-9])', line)
        if match:
            drives.append(match.group(1).split()[0].strip())
            
    return drives