Fix - disk size key

vmware-archive · May 20, 2020 · 975f790 · 975f790
1 parent 811c9a2
commit 975f790
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pyvcloud/vcd/utils.py b/pyvcloud/vcd/utils.py
@@ -549,8 +549,12 @@ def disk_to_dict(disk):
     result['name'] = disk.get('name')
     result['id'] = extract_id(disk.get('id'))
     result['status'] = disk.get('status')
-    result['size'] = humanfriendly.format_size(int(disk.get('size')))
-    result['size_bytes'] = disk.get('size')
+    if 'size' in disk.keys():
+        size_in_bytes = int(disk.get('size'))
+    else:
+        size_in_bytes = int(disk.get('sizeMb')) * (1024 * 1024)
+    result['size'] = humanfriendly.format_size(size_in_bytes)
+    result['size_bytes'] = size_in_bytes
     result['busType'] = disk.get('busType')
     result['busSubType'] = disk.get('busSubType')
     result['iops'] = disk.get('iops')