From 2e1dba899ac0374a391b43868c873a0fa3423122 Mon Sep 17 00:00:00 2001 From: Ian Katz Date: Wed, 9 Jan 2019 10:02:24 -0500 Subject: [PATCH] avoid KeyError: 'VPCId' --- aws_visualizer/dot/graph_region.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/aws_visualizer/dot/graph_region.py b/aws_visualizer/dot/graph_region.py index ca920ce..d2dbc19 100644 --- a/aws_visualizer/dot/graph_region.py +++ b/aws_visualizer/dot/graph_region.py @@ -149,7 +149,9 @@ def __str__(self): return self.__key() def get_vpc_id(self): - return self['VpcId'] if 'VpcId' in self else self['VPCId'] + if 'VpcId' in self: return self['VpcId'] + if 'VPCId' in self: return self['VPCId'] + return None def get_subnets(self): return self['Subnets'] if 'Subnets' in self else list(map(lambda a: a['SubnetId'], self['AvailabilityZones']))