Make sure racks get a lookup from rack name and site

This commit is contained in:
Marco Ceppi 2019-12-18 17:35:23 -05:00
parent 635f6569f8
commit 1fb06c96e3

View File

@ -26,7 +26,6 @@ with file.open('r') as stream:
optional_assocs = { optional_assocs = {
'tenant': (Tenant, 'name'), 'tenant': (Tenant, 'name'),
'platform': (Platform, 'name'), 'platform': (Platform, 'name'),
'rack': (Rack, 'name'),
'cluster': (Cluster, 'name'), 'cluster': (Cluster, 'name'),
'primary_ip4': (IPAddress, 'address'), 'primary_ip4': (IPAddress, 'address'),
'primary_ip6': (IPAddress, 'address') 'primary_ip6': (IPAddress, 'address')
@ -49,6 +48,9 @@ with file.open('r') as stream:
params[assoc] = model.objects.get(**query) params[assoc] = model.objects.get(**query)
if 'rack' in params:
params['rack'] = Rack.objects.get(name=params.pop('rack'), site=params['site'].id)
if 'face' in params: if 'face' in params:
for rack_face in RACK_FACE_CHOICES: for rack_face in RACK_FACE_CHOICES:
if params['face'] in rack_face: if params['face'] in rack_face: