Fix - disk list size

vmware-archive · May 20, 2020 · aef7a4e · aef7a4e
1 parent 0d88338
commit aef7a4e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/vcd_cli/disk.py b/vcd_cli/disk.py
@@ -87,6 +87,10 @@ def list_disks(ctx):
             if hasattr(disk, 'attached_vms') and \
                hasattr(disk.attached_vms, 'VmReference'):
                 attached_vms = disk.attached_vms.VmReference.get('name')
+            if 'size' in disk.keys():
+                size_in_bytes = int(disk.get('size'))
+            else:
+                size_in_bytes = int(disk.get('sizeMb')) * (1024 * 1024)
             result.append({
                 'name':
                 disk.get('name'),
@@ -95,9 +99,8 @@ def list_disks(ctx):
                 'owner':
                 disk.Owner.User.get('name'),
                 'size':
-                humanfriendly.format_size(int(disk.get('size'))),
-                'size_bytes':
-                disk.get('size'),
+                humanfriendly.format_size(size_in_bytes),
+                'size_bytes': size_in_bytes,
                 'status':
                 VCLOUD_STATUS_MAP.get(int(disk.get('status'))),
                 'vms_attached':