def _ListAllDisksMac(self):
"""Lists all disks connected to the machine.
Returns:
list(MacDiskArtifact): a list of disks.
"""
disk_list = []
for mac_disk in macdisk.WholeDisks():
disk_name = mac_disk.deviceidentifier
disk_size = mac_disk.totalsize
disk = MacDiskArtifact(os.path.join('/dev', disk_name), disk_size)
disk_list.append(disk)
return disk_list
def _GetListDisksArtifact(self):
"""Generates a StringArtifact containing information about all disks.
Returns:
StringArtifact: the artifact.
"""
if self._platform == 'darwin':
#pylint: disable=protected-access
diskutil_artifact = base.StringArtifact(
'Disks/diskutil.txt',
json.dumps([md._attributes for md in macdisk.WholeDisks()]))
return diskutil_artifact
lsblk_artifact = base.StringArtifact('Disks/lsblk.txt',
json.dumps(self._GetLsblkDict()))
return lsblk_artifact