Update ModificationStatus members to uppercase

codefiles · codefiles · commit 878f404b4623 · 2025-01-12T12:30:54.000-05:00
diff --git a/archinstall/lib/disk/device_handler.py b/archinstall/lib/disk/device_handler.py
@@ -508,7 +508,7 @@ def _setup_partition(
 	) -> None:
 		# when we require a delete and the partition to be (re)created
 		# already exists then we have to delete it first
-		if requires_delete and part_mod.status in [ModificationStatus.Modify, ModificationStatus.Delete]:
+		if requires_delete and part_mod.status in [ModificationStatus.MODIFY, ModificationStatus.DELETE]:
 			info(f'Delete existing partition: {part_mod.safe_dev_path}')
 			part_info = self.find_partition(part_mod.safe_dev_path)
 
@@ -517,7 +517,7 @@ def _setup_partition(
 
 			disk.deletePartition(part_info.partition)
 
-		if part_mod.status == ModificationStatus.Delete:
+		if part_mod.status == ModificationStatus.DELETE:
 			return
 
 		start_sector = part_mod.start.convert(
diff --git a/archinstall/lib/disk/device_model.py b/archinstall/lib/disk/device_model.py
@@ -163,20 +163,20 @@ def parse_arg(cls, disk_config: _DiskLayoutConfigurationSerialization) -> DiskLa
 				continue
 
 			first = non_delete_partitions[0]
-			if first.status == ModificationStatus.Create and not first.start.is_valid_start():
+			if first.status == ModificationStatus.CREATE and not first.start.is_valid_start():
 				raise ValueError('First partition must start at no less than 1 MiB')
 
 			for i, current_partition in enumerate(non_delete_partitions[1:], start=1):
 				previous_partition = non_delete_partitions[i - 1]
 				if (
-					current_partition.status == ModificationStatus.Create
+					current_partition.status == ModificationStatus.CREATE
 					and current_partition.start < previous_partition.end
 				):
 					raise ValueError('Partitions overlap')
 
 			create_partitions = [
 				part_mod for part_mod in non_delete_partitions
-				if part_mod.status == ModificationStatus.Create
+				if part_mod.status == ModificationStatus.CREATE
 			]
 
 			if not create_partitions:
@@ -823,10 +823,10 @@ def installation_hooks(self) -> str | None:
 
 
 class ModificationStatus(Enum):
-	Exist = 'existing'
-	Modify = 'modify'
-	Delete = 'delete'
-	Create = 'create'
+	EXIST = 'existing'
+	MODIFY = 'modify'
+	DELETE = 'delete'
+	CREATE = 'create'
 
 
 class _PartitionModificationSerialization(TypedDict):
@@ -872,7 +872,7 @@ def __post_init__(self) -> None:
 		if self.is_exists_or_modify() and not self.dev_path:
 			raise ValueError('If partition marked as existing a path must be set')
 
-		if self.fs_type is None and self.status == ModificationStatus.Modify:
+		if self.fs_type is None and self.status == ModificationStatus.MODIFY:
 			raise ValueError('FS type must not be empty on modifications with status type modify')
 
 	@override
@@ -915,7 +915,7 @@ def from_existing_partition(cls, partition_info: _PartitionInfo) -> PartitionMod
 			subvol_mods = []
 
 		return PartitionModification(
-			status=ModificationStatus.Exist,
+			status=ModificationStatus.EXIST,
 			type=partition_info.type,
 			start=partition_info.start,
 			length=partition_info.length,
@@ -970,23 +970,23 @@ def is_home(self) -> bool:
 		)
 
 	def is_modify(self) -> bool:
-		return self.status == ModificationStatus.Modify
+		return self.status == ModificationStatus.MODIFY
 
 	def is_delete(self) -> bool:
-		return self.status == ModificationStatus.Delete
+		return self.status == ModificationStatus.DELETE
 
 	def exists(self) -> bool:
-		return self.status == ModificationStatus.Exist
+		return self.status == ModificationStatus.EXIST
 
 	def is_exists_or_modify(self) -> bool:
 		return self.status in [
-			ModificationStatus.Exist,
-			ModificationStatus.Delete,
-			ModificationStatus.Modify
+			ModificationStatus.EXIST,
+			ModificationStatus.DELETE,
+			ModificationStatus.MODIFY
 		]
 
 	def is_create_or_modify(self) -> bool:
-		return self.status in [ModificationStatus.Create, ModificationStatus.Modify]
+		return self.status in [ModificationStatus.CREATE, ModificationStatus.MODIFY]
 
 	@property
 	def mapper_name(self) -> str | None:
diff --git a/archinstall/lib/disk/partitioning_menu.py b/archinstall/lib/disk/partitioning_menu.py
@@ -64,7 +64,7 @@ def table_data(self) -> dict[str, str]:
 			return self.segment.table_data()
 
 		part_mod = PartitionModification(
-			status=ModificationStatus.Create,
+			status=ModificationStatus.CREATE,
 			type=PartitionType._Unknown,
 			start=self.segment.start,
 			length=self.segment.length,
@@ -179,7 +179,7 @@ def _run_actions_on_entry(self, entry: DiskSegment) -> None:
 	@override
 	def selected_action_display(self, selection: DiskSegment) -> str:
 		if isinstance(selection.segment, PartitionModification):
-			if selection.segment.status == ModificationStatus.Create:
+			if selection.segment.status == ModificationStatus.CREATE:
 				return str(_('Partition - New'))
 			elif selection.segment.is_delete() and selection.segment.dev_path:
 				title = str(_('Partition')) + '\n\n'
@@ -296,7 +296,7 @@ def _delete_partition(
 		data: list[DiskSegment]
 	) -> list[DiskSegment]:
 		if entry.is_exists_or_modify():
-			entry.status = ModificationStatus.Delete
+			entry.status = ModificationStatus.DELETE
 			part_mods = self.get_part_mods(data)
 		else:
 			part_mods = [
@@ -337,12 +337,12 @@ def _set_btrfs_subvolumes(self, partition: PartitionModification) -> None:
 		).run()
 
 	def _prompt_formatting(self, partition: PartitionModification) -> None:
-		# an existing partition can toggle between Exist or Modify
+		# an existing partition can toggle between EXIST or MODIFY
 		if partition.is_modify():
-			partition.status = ModificationStatus.Exist
+			partition.status = ModificationStatus.EXIST
 			return
 		elif partition.exists():
-			partition.status = ModificationStatus.Modify
+			partition.status = ModificationStatus.MODIFY
 
 		# If we mark a partition for formatting, but the format is CRYPTO LUKS, there's no point in formatting it really
 		# without asking the user which inner-filesystem they want to use. Since the flag 'encrypted' = True is already set,
@@ -474,7 +474,7 @@ def _create_new_partition(self, free_space: FreeSpace) -> PartitionModification:
 			mountpoint = self._prompt_mountpoint()
 
 		partition = PartitionModification(
-			status=ModificationStatus.Create,
+			status=ModificationStatus.CREATE,
 			type=PartitionType.Primary,
 			start=free_space.start,
 			length=length,
diff --git a/archinstall/lib/interactions/disk_conf.py b/archinstall/lib/interactions/disk_conf.py
@@ -224,7 +224,7 @@ def _boot_partition(sector_size: disk.SectorSize, using_gpt: bool) -> disk.Parti
 
 	# boot partition
 	return disk.PartitionModification(
-		status=disk.ModificationStatus.Create,
+		status=disk.ModificationStatus.CREATE,
 		type=disk.PartitionType.Primary,
 		start=start,
 		length=size,
@@ -393,7 +393,7 @@ def suggest_single_disk_layout(
 		root_length = available_space - root_start
 
 	root_partition = disk.PartitionModification(
-		status=disk.ModificationStatus.Create,
+		status=disk.ModificationStatus.CREATE,
 		type=disk.PartitionType.Primary,
 		start=root_start,
 		length=root_length,
@@ -428,7 +428,7 @@ def suggest_single_disk_layout(
 			flags.append(disk.PartitionFlag.LINUX_HOME)
 
 		home_partition = disk.PartitionModification(
-			status=disk.ModificationStatus.Create,
+			status=disk.ModificationStatus.CREATE,
 			type=disk.PartitionType.Primary,
 			start=home_start,
 			length=home_length,
@@ -517,7 +517,7 @@ def suggest_multi_disk_layout(
 
 	# add root partition to the root device
 	root_partition = disk.PartitionModification(
-		status=disk.ModificationStatus.Create,
+		status=disk.ModificationStatus.CREATE,
 		type=disk.PartitionType.Primary,
 		start=root_start,
 		length=root_length,
@@ -539,7 +539,7 @@ def suggest_multi_disk_layout(
 
 	# add home partition to home device
 	home_partition = disk.PartitionModification(
-		status=disk.ModificationStatus.Create,
+		status=disk.ModificationStatus.CREATE,
 		type=disk.PartitionType.Primary,
 		start=home_start,
 		length=home_length,
diff --git a/examples/full_automated_installation.py b/examples/full_automated_installation.py
@@ -18,7 +18,7 @@
 
 # create a new boot partition
 boot_partition = disk.PartitionModification(
-	status=disk.ModificationStatus.Create,
+	status=disk.ModificationStatus.CREATE,
 	type=disk.PartitionType.Primary,
 	start=disk.Size(1, disk.Unit.MiB, device.device_info.sector_size),
 	length=disk.Size(512, disk.Unit.MiB, device.device_info.sector_size),
@@ -30,7 +30,7 @@
 
 # create a root partition
 root_partition = disk.PartitionModification(
-	status=disk.ModificationStatus.Create,
+	status=disk.ModificationStatus.CREATE,
 	type=disk.PartitionType.Primary,
 	start=disk.Size(513, disk.Unit.MiB, device.device_info.sector_size),
 	length=disk.Size(20, disk.Unit.GiB, device.device_info.sector_size),
@@ -45,7 +45,7 @@
 
 # create a new home partition
 home_partition = disk.PartitionModification(
-	status=disk.ModificationStatus.Create,
+	status=disk.ModificationStatus.CREATE,
 	type=disk.PartitionType.Primary,
 	start=start_home,
 	length=length_home,
