Modifying get_item_name to handle a None object

This commit is contained in:
John Westcott IV
2020-09-08 15:28:57 -04:00
parent faa33efdd2
commit 570251dc3d

View File

@@ -52,6 +52,7 @@ class TowerAPIModule(TowerModule):
return TowerAPIModule.IDENTITY_FIELDS.get(endpoint, 'name') return TowerAPIModule.IDENTITY_FIELDS.get(endpoint, 'name')
def get_item_name(self, item, allow_unknown=False): def get_item_name(self, item, allow_unknown=False):
if item:
if 'name' in item: if 'name' in item:
return item['name'] return item['name']
@@ -65,7 +66,10 @@ class TowerAPIModule(TowerModule):
if allow_unknown: if allow_unknown:
return 'unknown' return 'unknown'
if item:
self.exit_json(msg='Cannot determine identity field for {0} object.'.format(item.get('type', 'unknown'))) self.exit_json(msg='Cannot determine identity field for {0} object.'.format(item.get('type', 'unknown')))
else:
self.exit_json(msg='Cannot determine identity field for Undefined object.')
def head_endpoint(self, endpoint, *args, **kwargs): def head_endpoint(self, endpoint, *args, **kwargs):
return self.make_request('HEAD', endpoint, **kwargs) return self.make_request('HEAD', endpoint, **kwargs)