♻️ Use site name in query for racks when building devices (#29)

This commit is contained in:
Kyler 2020-05-28 17:38:15 -05:00 committed by GitHub
parent 7a061a6b12
commit 6f9feaea10
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -44,7 +44,14 @@ with file.open('r') as stream:
for assoc, details in optional_assocs.items():
if assoc in params:
model, field = details
query = { field: params.pop(assoc) }
if assoc == 'rack': # Special handling for rack query to reference rack name and site
query = {
'site': params.get('site'),
field: params.pop(assoc),
}
else:
query = { field: params.pop(assoc) }
params[assoc] = model.objects.get(**query)