diff --git a/menu.py b/menu.py index 1916b78..d0d9878 100644 --- a/menu.py +++ b/menu.py @@ -88,7 +88,9 @@ class ChoosePartition: self.screen.addstr(2 + i, 4, s, curses.color_pair(is_selected)) def _select_print_block_device(self, bd, i): - if 'model' not in bd or 'size' not in bd or 'name' not in bd: + if not ('model' in bd and + 'size' in bd or + 'name' in bd): raise Exception('Wrong lsblk json format. ' + 'No model, size or name in blockdevice')