From fa4be63e483a0a692f26103b45ad6abe73879f37 Mon Sep 17 00:00:00 2001
From: Anton Hvornum <anton.feeds+github@gmail.com>
Date: Sun, 18 Oct 2020 20:26:09 +0200
Subject: Forgot an import, as well as made BlockDevice() have less verbose
 output on json.dumps.

---
 archinstall/lib/disk.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'archinstall/lib/disk.py')

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,
-- 
cgit v1.2.3-70-g09d2