#
 #    Copyright (C) 2015 vmware inc.
 #
 #    Author: Mahmoud Bassiouny <mbassiouny@vmware.com>
 
 import subprocess
 import os
 
 class Device(object):

     def __init__(self, model, path, size):

         self.model = model
         self.path = path
         self.size = size
 
     @staticmethod
     def refresh_devices():

         devices_list = subprocess.check_output(['lsblk', '-d', '-I', '8,259', '-n', '--output', 'NAME,SIZE,MODEL'], stderr=open(os.devnull, 'w'))

         return Device.wrap_devices_from_list(devices_list)

 
     @staticmethod

     def refresh_devices_bytes():

         devices_list = subprocess.check_output(['lsblk', '-d', '--bytes', '-I', '8,259', '-n', '--output', 'NAME,SIZE,MODEL'], stderr=open(os.devnull, 'w'))

         return Device.wrap_devices_from_list(devices_list)
 
     @staticmethod

     def wrap_devices_from_list(list):

         devices = []

         deviceslines = list.splitlines()
         for deviceline in deviceslines:
             cols = deviceline.split(None, 2)

             #skip Virtual NVDIMM from install list
             if(cols[0].startswith("pmem")):
                 continue

             model = "Unknown"
             if(len(cols) >= 3):
                 model = cols[2]

             devices.append(

                     Device(model #Model

                         , '/dev/' + cols[0] #Path
                         , cols[1] #size
                         ))

 
         return devices