Forgot an import, as well as made BlockDevice() have less verbose output on json.dumps.

2 files changed, 12 insertions, 1 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index 7cd8dd0f..b11f2318 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -25,6 +25,16 @@ class BlockDevice():
 			raise KeyError(f'{self} does not contain information: "{key}"')
 		return self.info[key]
 
+	def json(self):
+		"""
+		json() has precedence over __dump__, so this is a way
+		to give less/partial information for user readability.
+		"""
+		return {
+			'path' : self.path,
+			'size' : self.info['size'] if 'size' in self.info else '<unknown>'
+		}
+
 	def __dump__(self):
 		return {
 			'path' : self.path,
diff --git a/examples/guided.py b/examples/guided.py
index 00d2565c..958b1e5f 100644
--- a/examples/guided.py
+++ b/examples/guided.py
@@ -1,4 +1,5 @@
-import archinstall, getpass, time, json
+import archinstall
+import getpass, time, json, sys
 
 def perform_installation(device, boot_partition, language, mirrors):
 	"""