From 2529d6a5f59eb6a16f95bf9d1117a6033c527df9 Mon Sep 17 00:00:00 2001 From: Daniel Date: Fri, 22 Apr 2022 21:23:38 +1000 Subject: Fix blockdevice key error (#1079) Co-authored-by: Daniel Girtler --- archinstall/lib/disk/blockdevice.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'archinstall/lib/disk/blockdevice.py') diff --git a/archinstall/lib/disk/blockdevice.py b/archinstall/lib/disk/blockdevice.py index ff741f18..16ba0160 100644 --- a/archinstall/lib/disk/blockdevice.py +++ b/archinstall/lib/disk/blockdevice.py @@ -7,12 +7,13 @@ from typing import Optional, Dict, Any, Iterator, Tuple, List, TYPE_CHECKING # https://stackoverflow.com/a/39757388/929999 if TYPE_CHECKING: from .partition import Partition - + from ..exceptions import DiskError, SysCallError from ..output import log from ..general import SysCommand from ..storage import storage + class BlockDevice: def __init__(self, path :str, info :Optional[Dict[str, Any]] = None): if not info: @@ -38,7 +39,9 @@ class BlockDevice: yield self.partitions[partition] def __getitem__(self, key :str, *args :str, **kwargs :str) -> Any: - if key not in self.info: + if hasattr(self, key): + return getattr(self, key) + elif key not in self.info: raise KeyError(f'{self} does not contain information: "{key}"') return self.info[key] -- cgit v1.2.3-54-g00ecf