Skip to content

Commit bb88e64

Browse files
committed
Fix Nesting
1 parent a097f30 commit bb88e64

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

proxstar/vm.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,11 @@ def get_disk_size(self, name='virtio0'):
251251
disk_size = split.split('=')[1].rstrip('G')
252252
return disk_size
253253

254+
def check_in_gb(size):
255+
if size[-1] == 'M':
256+
size = f'{int(size.rstrip("M")) / 1000}G'
257+
return size
258+
254259
@lazy_property
255260
def disks(self):
256261
disks = []
@@ -261,9 +266,7 @@ def disks(self):
261266
disk_size = val.split(',')
262267
for split in disk_size:
263268
if 'size' in split:
264-
size = split.split('=')[1]
265-
if size[-1] == 'M':
266-
size = f'{int(size.rstrip("M")) / 1000}G'
269+
size = check_in_gb(split.split('=')[1])
267270
disk_size = size.rstrip('G')
268271
disks.append([key, disk_size])
269272
disks = sorted(disks, key=lambda x: x[0])

0 commit comments

Comments
 (0)