From 542f9cba691092ef43f700e708ef53b2b3834a1b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A1=D0=B0=D0=BC=D0=B1=D1=83=D0=BA=D0=BE=D0=B2=20=D0=95?= =?UTF-8?q?=D0=B3=D0=BE=D1=80=20=D0=90=D0=BB=D0=B5=D0=BA=D1=81=D0=B0=D0=BD?= =?UTF-8?q?=D0=B4=D1=80=D0=BE=D0=B2=D0=B8=D1=87?= Date: Tue, 19 Jul 2022 14:25:39 +0300 Subject: [PATCH] added exception for KeyError --- src/blkinfo/wrappers.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/blkinfo/wrappers.py b/src/blkinfo/wrappers.py index d6fef56..5de6d7d 100644 --- a/src/blkinfo/wrappers.py +++ b/src/blkinfo/wrappers.py @@ -153,8 +153,11 @@ def _build_disk_tree(): while parent_stack: p = parent_stack.pop() if p[1] < level: - disk_tree[p[0]]['children'].append(disk_tree[name]) - disk_tree[name]['parents'].append(p[0]) + try: + disk_tree[p[0]]['children'].append(disk_tree[name]) + disk_tree[name]['parents'].append(p[0]) + except KeyError: + continue parent_stack.append(p) break