import subprocess, os, uuid, syslog

def execute(cmd,shell):
    proc = subprocess.Popen(cmd, shell=shell, stdout=subprocess.PIPE)
    return proc.communicate()[0].decode()

output = execute(['/sbin/vdq','-q'],False)

capacityDisk = None
cachingDisk = None
largestDisk = None
largestDiskSize = None
usbDevice = None

results = eval(output)
for disk in results:
    diskId = disk['Name']

    if disk['State'] == 'Eligible for use by VSAN':
        print(" Found Disk: " + diskId)

        diskcapacity_cmd = "localcli storage core device capacity list -d " + diskId + " | tail -1 | awk '{print $5}'"
        print(" Running disk capacity command: " + diskcapacity_cmd)
        ps = subprocess.Popen(diskcapacity_cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
        diskSize = ps.communicate()[0].decode().strip()

        if largestDisk == None:
            largestDisk = diskId
            largestDiskSize = diskSize
        else:
            if diskSize > largestDiskSize:
                capacityDisk = diskId
                cachingDisk = largestDisk
            else:
                capacityDisk = largestDisk
                cachingDisk = diskId

        print(" Largest Capacity Disk so far: " + diskId + " (" + diskSize + ")")


    if disk['State'] == 'Ineligible for use by VSAN' and disk['VSANUUID'] == "":
        usbDevice = diskId
        fo = open("/tmp/usbDevice.txt", "w")
        fo.write(usbDevice)
        fo.close()


    if cachingDisk != None and capacityDisk != None:
        # Tag Capacity Disk
        disktag_cmd = "localcli vsan storage tag add -d " + capacityDisk + " -t capacityFlash"
        print(" Running disk capacity tagging command: " + disktag_cmd)

        # Create VSAN Disk Group
        diskgroup_cmd = "localcli vsan storage add -s " + cachingDisk + " -d " + capacityDisk
        print(" Running disk group create command: " + diskgroup_cmd)