diff --git a/archinstall/lib/disk/device_handler.py b/archinstall/lib/disk/device_handler.py index c099859b5c..3f476019ce 100644 --- a/archinstall/lib/disk/device_handler.py +++ b/archinstall/lib/disk/device_handler.py @@ -706,7 +706,7 @@ def partition( if partition_table is None: raise ValueError('Modification is marked as wipe but no partitioning table was provided') - if partition_table.MBR and len(modification.partitions) > 3: + if partition_table.is_mbr() and len(modification.partitions) > 3: raise DiskError('Too many partitions on disk, MBR disks can only have 3 primary partitions') # WARNING: the entire device will be wiped and all data lost diff --git a/archinstall/lib/models/device_model.py b/archinstall/lib/models/device_model.py index c3cc6df418..71eeff6f90 100644 --- a/archinstall/lib/models/device_model.py +++ b/archinstall/lib/models/device_model.py @@ -202,6 +202,9 @@ class PartitionTable(Enum): GPT = 'gpt' MBR = 'msdos' + def is_mbr(self) -> bool: + return self == PartitionTable.MBR + class Units(Enum): BINARY = 'binary'