vmdesc: parse_size: Support fractional units

master
Dustin 2015-08-06 16:12:07 -05:00
parent 7629a5dd97
commit 4c3ec695bf
1 changed files with 2 additions and 2 deletions

View File

@ -46,7 +46,7 @@ UNITS = {
} }
SIZE_RE = re.compile( SIZE_RE = re.compile(
r'^(?P<value>[0-9]+)\s*' r'^(?P<value>[0-9]+(?:\.[0-9]+)?)\s*'
r'(?P<unit>\w+)?\s*$' r'(?P<unit>\w+)?\s*$'
) )
@ -72,7 +72,7 @@ def parse_size(size):
factor = UNITS[parts['unit']] factor = UNITS[parts['unit']]
except KeyError: except KeyError:
raise ValueError('Invalid size : {}'.format(size)) raise ValueError('Invalid size : {}'.format(size))
return int(parts['value']) * factor return int(float(parts['value']) * factor)
class VirtualMachine(object): class VirtualMachine(object):