Fix to_mb stripping last digit

2017-03-11 01:14:50 +01:00 · 2017-03-11 01:14:50 +01:00 · a3ede682b7
commit a3ede682b7
parent 7d36e101a7
1 changed files with 3 additions and 1 deletions
--- a/vm.py
+++ b/vm.py
@ -57,8 +57,10 @@ def to_mb(value):
    v = int(value.lower()[:-1])
    if value.lower()[-1] == 'g':
        return v * 1024
-    else:
+    elif value.lower()[-1] == 'm':
        return v
+    else:
+        return int(value.lower())

 j2 = jinja2.Environment(loader=jinja2.FileSystemLoader('templates'))
 j2.filters['ipreplace'] = replace_ip