Fix MBR conditional (#3192)

archlinux · Feb 22, 2025 · b83bc79 · b83bc79
1 parent c364917
commit b83bc79
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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'