lib/host: Improved disk_is_available() using blkid for reliable raw disk checks.

Signed-off-by: Rushikesh Jadhav <[email protected]>

Author: rushikeshjadhav

lib/host.py

@@ -516,10 +516,12 @@ def disks(self):
         disks.sort()
         return disks
 
-    def disk_is_available(self, disk):
+    def disk_is_available(self, disk, disk_raw=False):
+        if disk_raw:
+            return self.ssh_with_result(['blkid', '/dev/' + disk]).returncode == 2
         return len(self.ssh(['lsblk', '-n', '-o', 'MOUNTPOINT', '/dev/' + disk]).strip()) == 0
 
-    def available_disks(self, blocksize=512):
+    def available_disks(self, blocksize=512, disk_raw=False):
         """
         Return a list of available disks for formatting, creating SRs or such.
 
@@ -534,7 +536,7 @@ def available_disks(self, blocksize=512):
             sec_size = line[1]
             if sec_size == str(blocksize):
                 avail_disks.append(disk)
-        return [disk for disk in avail_disks if self.disk_is_available(disk)]
+        return [disk for disk in avail_disks if self.disk_is_available(disk, disk_raw)]
 
     def file_exists(self, filepath, regular_file=True):
         option = '-f' if regular_file else '-e'