Skip to content

Commit 46b38d4

Browse files
committed
Allow total to be optional in SizeTriple
Signed-off-by: mulhern <amulhern@redhat.com>
1 parent 38635d2 commit 46b38d4

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

src/stratis_cli/_actions/_utils.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -279,11 +279,11 @@ class SizeTriple:
279279
Manage values in a size triple.
280280
"""
281281

282-
def __init__(self, total: Range, used: Optional[Range]):
282+
def __init__(self, total: Optional[Range], used: Optional[Range]):
283283
self._total = total
284284
self._used = used
285285

286-
def total(self) -> Range:
286+
def total(self) -> Optional[Range]:
287287
"""
288288
Total.
289289
"""
@@ -299,4 +299,8 @@ def free(self) -> Optional[Range]:
299299
"""
300300
Total - used.
301301
"""
302-
return None if self._used is None else self._total - self._used
302+
return (
303+
None
304+
if self._used is None or self.total is None
305+
else self._total - self._used
306+
)

0 commit comments

Comments
 (0)