accepts None values for size (for downloading disks)

7292c3e8 · Guba Sándor · 94379fb3 · 7292c3e8
Commit 7292c3e8 authored Mar 17, 2014 by Guba Sándor
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletions

disk.py
+4 -1

No files found.
--- a/disk.py
+++ b/disk.py
@@ -36,7 +36,10 @@ class Disk(object):
        if type not in self.TYPES:
            raise Exception('Invalid type: %s' % format)
        self.type = type
-        self.size = int(size)
+        try:
+            self.size = int(size)
+        except:
+            self.size = None
        self.base_name = base_name

    @classmethod