From a7ca037a26de53fd242f89bc6a90fd53337b4d13 Mon Sep 17 00:00:00 2001
From: Daniel Girtler <blackrabbit256@gmail.com>
Date: Tue, 7 Jun 2022 01:28:46 +1000
Subject: Update the subvolume menu - fix for #1278 (#1297)

* Update subvolume

* Add mypy compliance

Co-authored-by: Daniel Girtler <girtler.daniel@gmail.com>
Co-authored-by: Anton Hvornum <anton@hvornum.se>
---
 archinstall/lib/models/subvolume.py | 68 +++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 archinstall/lib/models/subvolume.py

(limited to 'archinstall/lib/models/subvolume.py')

diff --git a/archinstall/lib/models/subvolume.py b/archinstall/lib/models/subvolume.py
new file mode 100644
index 00000000..34a09227
--- /dev/null
+++ b/archinstall/lib/models/subvolume.py
@@ -0,0 +1,68 @@
+from dataclasses import dataclass
+from typing import List, Any, Dict
+
+
+@dataclass
+class Subvolume:
+	name: str
+	mountpoint: str
+	compress: bool = False
+	nodatacow: bool = False
+
+	def display(self) -> str:
+		options_str = ','.join(self.options)
+		return f'{_("Subvolume")}: {self.name:15} {_("Mountpoint")}: {self.mountpoint:20} {_("Options")}: {options_str}'
+
+	@property
+	def options(self) -> List[str]:
+		options = [
+			'compress' if self.compress else '',
+			'nodatacow' if self.nodatacow else ''
+		]
+		return [o for o in options if len(o)]
+
+	def json(self) -> Dict[str, Any]:
+		return {
+			'name': self.name,
+			'mountpoint': self.mountpoint,
+			'compress': self.compress,
+			'nodatacow': self.nodatacow
+		}
+
+	@classmethod
+	def _parse(cls, config_subvolumes: List[Dict[str, Any]]) -> List['Subvolume']:
+		subvolumes = []
+		for entry in config_subvolumes:
+			if not entry.get('name', None) or not entry.get('mountpoint', None):
+				continue
+
+			subvolumes.append(
+				Subvolume(
+					entry['name'],
+					entry['mountpoint'],
+					entry.get('compress', False),
+					entry.get('nodatacow', False)
+				)
+			)
+
+		return subvolumes
+
+	@classmethod
+	def _parse_backwards_compatible(cls, config_subvolumes) -> List['Subvolume']:
+		subvolumes = []
+		for name, mountpoint in config_subvolumes.items():
+			if not name or not mountpoint:
+				continue
+
+			subvolumes.append(Subvolume(name, mountpoint))
+
+		return subvolumes
+
+	@classmethod
+	def parse_arguments(cls, config_subvolumes: Any) -> List['Subvolume']:
+		if isinstance(config_subvolumes, list):
+			return cls._parse(config_subvolumes)
+		elif isinstance(config_subvolumes, dict):
+			return cls._parse_backwards_compatible(config_subvolumes)
+
+		raise ValueError('Unknown disk layout btrfs subvolume format')
-- 
cgit v1.2.3-54-g00ecf