From e5da8c01a934ce09e76d6a59df596fe3f06e280f Mon Sep 17 00:00:00 2001 From: Jarrod Johnson Date: Tue, 15 Apr 2025 15:51:12 -0400 Subject: [PATCH] Do not attempt to print non-existent data. --- confluent_client/bin/nodeconsole | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/confluent_client/bin/nodeconsole b/confluent_client/bin/nodeconsole index 82b95583..ddfbd0e3 100755 --- a/confluent_client/bin/nodeconsole +++ b/confluent_client/bin/nodeconsole @@ -123,10 +123,11 @@ if options.screenshot: sess = client.Command() for res in sess.read('/noderange/{}/console/ikvm_screenshot'.format(args[0])): for node in res.get('databynode', {}): - imgdata = res['databynode'][node]['image']['imgdata'] - sys.stdout.write('{}: '.format(node)) - kitty_draw(imgdata.encode()) - sys.stdout.write('\n') + imgdata = res['databynode'][node].get('image', {}).get('imgdata', None) + if imgdata: + sys.stdout.write('{}: '.format(node)) + kitty_draw(imgdata.encode()) + sys.stdout.write('\n') sys.exit(0) def kill(noderange):