Refactor default partition table (#3194)

archlinux · Feb 23, 2025 · a51475e · a51475e
1 parent d69441a
commit a51475e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/archinstall/lib/disk/filesystem.py b/archinstall/lib/disk/filesystem.py
@@ -6,7 +6,6 @@
 
 from archinstall.tui import Tui
 
-from ..hardware import SysInfo
 from ..interactions.general_conf import ask_abort
 from ..luks import Luks2
 from ..models.device_model import (
@@ -62,9 +61,7 @@ def perform_filesystem_operations(self, show_countdown: bool = True) -> None:
 
 		# Setup the blockdevice, filesystem (and optionally encryption).
 		# Once that's done, we'll hand over to perform_installation()
-		partition_table = PartitionTable.GPT
-		if SysInfo.has_uefi() is False:
-			partition_table = PartitionTable.MBR
+		partition_table = PartitionTable.default()
 
 		# make sure all devices are unmounted
 		for mod in device_mods:

diff --git a/archinstall/lib/models/device_model.py b/archinstall/lib/models/device_model.py
@@ -205,6 +205,10 @@ class PartitionTable(Enum):
 	def is_mbr(self) -> bool:
 		return self == PartitionTable.MBR
 
+	@classmethod
+	def default(cls) -> PartitionTable:
+		return cls.GPT if SysInfo.has_uefi() else cls.MBR
+
 
 class Units(Enum):
 	BINARY = 'binary'